Searched defs:_get_workload (Results 1 – 6 of 6) sorted by relevance
/dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/nn/ |
H A D | depthwise_conv2d.py | 33 def _get_workload(data, kernel, stride, padding, out_dtype): function
|
H A D | conv2d.py | 140 def _get_workload(data, kernel, stride, padding, out_dtype, data_layout='NCHW'): function
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/nn/ |
H A D | depthwise_conv2d.py | 49 def _get_workload(data, kernel, stride, padding, out_dtype): function
|
H A D | conv2d.py | 157 def _get_workload(data, kernel, stride, padding, out_dtype, data_layout="NCHW"): function
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/nn/ |
H A D | depthwise_conv2d.py | 33 def _get_workload(data, kernel, stride, padding, out_dtype): function
|
H A D | conv2d.py | 140 def _get_workload(data, kernel, stride, padding, out_dtype, data_layout='NCHW'): function
|