megengine.utils.network_node

Functions

flops_batchmatmul(opnode, inputs, outputs)

flops_conv(opnode, inputs, outputs)

flops_elemwise(opnode, inputs, outputs)

flops_matmul(opnode, inputs, outputs)

receptive_field(opnode, inputs, outputs)

str_to_mge_class(classname)

Classes

AdaptivePoolingForward()

Argmax()

Argmin()

ArgsortForward()

AssertEqual()

AxisAddRemove()

BatchConvBiasForward()

BatchNormForward()

BatchedIncrMeshIndexing()

BatchedMatrixMul()

BatchedMeshIndexing()

BatchedSetMeshIndexing()

Broadcast()

Concat()

CondTake()

ConstOpBase([data, name, device, graph])

ConvBiasForward()

ConvolutionBackwardData()

ConvolutionForward()

Copy()

CvtColorForward()

DeformableConvForward()

DeformablePSROIPoolingForward()

Dimshuffle()

Dot()

Elemwise()

ElemwiseMultiType()

Eye()

GetVarShape()

GroupLocalForward()

Host2DeviceCopy([shape, dtype, name, device])

Identity()

ImmutableTensor([data, name, device, graph])

IncrMeshIndexing()

IncrSubtensor()

IndexingBase()

IndexingIncrMultiAxisVec()

IndexingMultiAxisVec()

IndexingOneHot()

IndexingSetMultiAxisVec()

IndexingSetOneHot()

Linspace()

MatrixInverse()

MatrixMul()

MeshIndexing()

NMSKeep()

NetworkNode()

NvOf()

OpNode()

PoolingForward()

RNGOpr()

ROIAlignForward()

ROIPoolingForward()

ReadOnlyOpNode()

Reduce()

RemapForward()

Reshape()

ResizeForward()

SVD()

SetMeshIndexing()

SetSubtensor()

SharedDeviceTensor([data, name, device, graph])

Subtensor()

TopK()

TypeCvt()

VarNode([var, owner_opr, name])

VarNodeMeta(name, bases, namespace, **kwargs)

WarpAffineForward()

WarpPerspectiveForward()