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

Function GenerateDeconvOperations

dnn/scripts/cutlass_generator/generator.py:1815–1823  ·  view source on GitHub ↗
(args)

Source from the content-addressed store, hash-verified

1813
1814
1815def GenerateDeconvOperations(args):
1816 if args.type == "simt":
1817 return GenerateDeconv_Simt(args)
1818 else:
1819 assert args.type == "tensorop8816", (
1820 "operation deconv only support"
1821 "simt and tensorop8816. (got:{})".format(args.type)
1822 )
1823 return GenerateDeconv_TensorOp_8816(args)
1824
1825
1826def GenerateDwconv2dFpropOperations(args):

Callers 2

write_op_listFunction · 0.90
generator.pyFile · 0.85

Calls 3

GenerateDeconv_SimtFunction · 0.85
formatMethod · 0.45

Tested by

no test coverage detected