| /dports/misc/tvm/incubator-tvm-0.6.1/topi/recipe/conv/ |
| H A D | depthwise_conv2d_test.py | 25 …om topi.cuda.depthwise_conv2d import schedule_depthwise_conv2d_nchw, schedule_depthwise_conv2d_nhwc
165 s1 = schedule_depthwise_conv2d_nhwc(DepthwiseConv2d)
166 s2 = schedule_depthwise_conv2d_nhwc(ScaleShift)
167 s3 = schedule_depthwise_conv2d_nhwc(Relu)
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/apps/topi_recipe/conv/ |
| H A D | depthwise_conv2d_test.py | 28 schedule_depthwise_conv2d_nhwc,
193 s1 = schedule_depthwise_conv2d_nhwc(DepthwiseConv2d)
194 s2 = schedule_depthwise_conv2d_nhwc(ScaleShift)
195 s3 = schedule_depthwise_conv2d_nhwc(Relu)
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/topi/recipe/conv/ |
| H A D | depthwise_conv2d_test.py | 25 …om topi.cuda.depthwise_conv2d import schedule_depthwise_conv2d_nchw, schedule_depthwise_conv2d_nhwc
165 s1 = schedule_depthwise_conv2d_nhwc(DepthwiseConv2d)
166 s2 = schedule_depthwise_conv2d_nhwc(ScaleShift)
167 s3 = schedule_depthwise_conv2d_nhwc(Relu)
|
| /dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/hls/ |
| H A D | nn.py | 145 @generic.schedule_depthwise_conv2d_nhwc.register(["hls"])
146 def schedule_depthwise_conv2d_nhwc(outs): function
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/hls/ |
| H A D | nn.py | 145 @generic.schedule_depthwise_conv2d_nhwc.register(["hls"])
146 def schedule_depthwise_conv2d_nhwc(outs): function
|
| /dports/misc/tvm/incubator-tvm-0.6.1/topi/tests/python/ |
| H A D | test_topi_depthwise_conv2d.py | 160 s1 = topi.generic.schedule_depthwise_conv2d_nhwc(DepthwiseConv2d)
161 s2 = topi.generic.schedule_depthwise_conv2d_nhwc(ScaleShift)
162 s3 = topi.generic.schedule_depthwise_conv2d_nhwc(Relu)
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/topi/tests/python/ |
| H A D | test_topi_depthwise_conv2d.py | 160 s1 = topi.generic.schedule_depthwise_conv2d_nhwc(DepthwiseConv2d)
161 s2 = topi.generic.schedule_depthwise_conv2d_nhwc(ScaleShift)
162 s3 = topi.generic.schedule_depthwise_conv2d_nhwc(Relu)
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/topi/python/ |
| H A D | test_topi_depthwise_conv2d.py | 50 "generic": (topi.nn.depthwise_conv2d_nhwc, topi.generic.schedule_depthwise_conv2d_nhwc),
53 topi.arm_cpu.schedule_depthwise_conv2d_nhwc,
55 "gpu": (topi.nn.depthwise_conv2d_nhwc, topi.cuda.schedule_depthwise_conv2d_nhwc),
|
| /dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/cuda/ |
| H A D | depthwise_conv2d.py | 134 @generic.schedule_depthwise_conv2d_nhwc.register(["cuda", "gpu"])
135 def schedule_depthwise_conv2d_nhwc(outs): function
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/cuda/ |
| H A D | depthwise_conv2d.py | 134 @generic.schedule_depthwise_conv2d_nhwc.register(["cuda", "gpu"])
135 def schedule_depthwise_conv2d_nhwc(outs): function
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/relay/op/strategy/ |
| H A D | rocm.py | 118 wrap_topi_schedule(topi.cuda.schedule_depthwise_conv2d_nhwc),
|
| H A D | hls.py | 120 wrap_topi_schedule(topi.hls.schedule_depthwise_conv2d_nhwc),
|
| H A D | arm_cpu.py | 193 wrap_topi_schedule(topi.arm_cpu.schedule_depthwise_conv2d_nhwc),
|
| H A D | x86.py | 163 wrap_topi_schedule(topi.generic.schedule_depthwise_conv2d_nhwc),
|
| /dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/intel_graphics/ |
| H A D | depthwise_conv2d.py | 135 @generic.schedule_depthwise_conv2d_nhwc.register(["intel_graphics"])
136 def schedule_depthwise_conv2d_nhwc(outs): function
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/intel_graphics/ |
| H A D | depthwise_conv2d.py | 135 @generic.schedule_depthwise_conv2d_nhwc.register(["intel_graphics"])
136 def schedule_depthwise_conv2d_nhwc(outs): function
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/generic/ |
| H A D | nn.py | 382 def schedule_depthwise_conv2d_nhwc(outs): function
|
| /dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/generic/ |
| H A D | nn.py | 265 def schedule_depthwise_conv2d_nhwc(outs): function
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/generic/ |
| H A D | nn.py | 265 def schedule_depthwise_conv2d_nhwc(outs): function
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/hls/ |
| H A D | nn.py | 140 def schedule_depthwise_conv2d_nhwc(outs): function
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/cuda/ |
| H A D | depthwise_conv2d.py | 140 def schedule_depthwise_conv2d_nhwc(outs): function
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/intel_graphics/ |
| H A D | depthwise_conv2d.py | 139 def schedule_depthwise_conv2d_nhwc(outs): function
|
| /dports/misc/tvm/incubator-tvm-0.6.1/python/tvm/autotvm/task/ |
| H A D | topi_integration.py | 101 topi.generic.schedule_depthwise_conv2d_nhwc],
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/python/tvm/autotvm/task/ |
| H A D | topi_integration.py | 101 topi.generic.schedule_depthwise_conv2d_nhwc],
|
| /dports/misc/tvm/incubator-tvm-0.6.1/nnvm/python/nnvm/top/ |
| H A D | nn.py | 164 return topi.generic.schedule_depthwise_conv2d_nhwc(outs)
|