Package org.apache.wayang.basic.model.op

Class Op

java.lang.Object

org.apache.wayang.basic.model.op.Op

All Implemented Interfaces:

Serializable

Direct Known Subclasses:

ArgMax, BatchNorm, Cast, Conv, ConvLSTM2D, CrossEntropyLoss, Eq, Get, Input, Linear, Mean, MSELoss, ReLU, Reshape, Sigmoid, Slice, Softmax, Transpose, ZeroLike

public abstract class Op
extends Object
implements Serializable

See Also:

• Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static enum	Op.DType

Field Summary

Fields

Modifier and Type	Field	Description
protected final Op.DType	dType
protected final List<Op>	fromList
protected final int	id
protected final String	name

Constructor Summary

Constructors

Constructor	Description
Op(String name, Op.DType dType)
Op(Op.DType dType)

Method Summary

Modifier and Type	Method	Description
Op.DType	getDType()
List<Op>	getFromList()
int	getId()
String	getName()
abstract int	inputsRequired()
Op	with(Op... ops)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Details

id

protected final int id

name

protected final String name

fromList

protected final List<Op> fromList

dType

protected final Op.DType dType

Constructor Details

Op

public Op(Op.DType dType)

Op

public Op(String name,
 Op.DType dType)

Method Details

getId

public int getId()

getName

public String getName()

getDType

public Op.DType getDType()

getFromList

public List<Op> getFromList()

with

public Op with(Op... ops)

inputsRequired

public abstract int inputsRequired()