MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / TypeCvt

Class TypeCvt

imperative/python/megengine/utils/network_node.py:374–383  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

372
373
374class TypeCvt(OpNode):
375 type = "TypeCvt"
376 opdef = builtin.TypeCvt
377
378 @classmethod
379 def load(cls, opr):
380 obj = super(TypeCvt, cls).load(opr)
381 t_dtype = opr.outputs[0].dtype
382 obj.params["dtype"] = t_dtype
383 return obj
384
385
386class MatrixInverse(OpNode):

Callers 2

syncbn_stage0Function · 0.50
batch_norm_ndFunction · 0.50

Calls

no outgoing calls

Tested by 1

batch_norm_ndFunction · 0.40