/dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/nn/ |
H A D | conv2d.py | 35 def conv2d(input, filter, strides, padding, dilation, layout='NCHW', out_dtype=None): argument 140 def _get_workload(data, kernel, stride, padding, out_dtype, data_layout='NCHW'): argument 168 def conv2d_nchw(Input, Filter, stride, padding, dilation, out_dtype=None): argument 234 def conv2d_hwcn(Input, Filter, stride, padding, dilation, out_dtype=None): argument 300 def conv2d_nhwc(Input, Filter, stride, padding, dilation, out_dtype='float32'): argument 504 out_dtype='int32'): argument 552 out_dtype='int32'): argument 701 layout, out_dtype, tile_size): argument 726 def conv2d_winograd_nnpack_weight_transform(kernel, convolution_algorithm, out_dtype): argument 745 input, filter, bias, strides, padding, dilation, layout, out_dtype): argument [all …]
|
H A D | conv2d_transpose.py | 29 def conv2d_transpose_nchw(Input, Filter, strides, padding, out_dtype): argument 57 def conv2d_transpose_nchw_preprocess(data, kernel, strides, padding, out_dtype): argument 82 def declaration_conv2d_transpose_impl(data, kernel, strides, padding, out_dtype): argument
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/nn/ |
H A D | conv2d.py | 35 def conv2d(input, filter, strides, padding, dilation, layout='NCHW', out_dtype=None): argument 140 def _get_workload(data, kernel, stride, padding, out_dtype, data_layout='NCHW'): argument 168 def conv2d_nchw(Input, Filter, stride, padding, dilation, out_dtype=None): argument 234 def conv2d_hwcn(Input, Filter, stride, padding, dilation, out_dtype=None): argument 300 def conv2d_nhwc(Input, Filter, stride, padding, dilation, out_dtype='float32'): argument 504 out_dtype='int32'): argument 552 out_dtype='int32'): argument 701 layout, out_dtype, tile_size): argument 726 def conv2d_winograd_nnpack_weight_transform(kernel, convolution_algorithm, out_dtype): argument 745 input, filter, bias, strides, padding, dilation, layout, out_dtype): argument [all …]
|
H A D | conv2d_transpose.py | 29 def conv2d_transpose_nchw(Input, Filter, strides, padding, out_dtype): argument 57 def conv2d_transpose_nchw_preprocess(data, kernel, strides, padding, out_dtype): argument 82 def declaration_conv2d_transpose_impl(data, kernel, strides, padding, out_dtype): argument
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/nn/ |
H A D | conv2d.py | 51 def conv2d(input, filter, strides, padding, dilation, layout="NCHW", out_dtype=None): argument 157 def _get_workload(data, kernel, stride, padding, out_dtype, data_layout="NCHW"): argument 188 def conv2d_nchw(Input, Filter, stride, padding, dilation, out_dtype=None): argument 260 def conv2d_hwcn(Input, Filter, stride, padding, dilation, out_dtype=None): argument 334 def conv2d_nhwc(Input, Filter, stride, padding, dilation, out_dtype="float32"): argument 406 def conv2d_NCHWc(data, kernel, stride, padding, dilation, layout, out_layout, out_dtype="float32"): argument 508 data, kernel, stride, padding, dilation, layout, out_layout, out_dtype="int32" argument 722 def conv2d_winograd_nnpack_weight_transform(kernel, convolution_algorithm, out_dtype): argument 745 def group_conv2d_nchw(Input, Filter, stride, padding, dilation, groups, out_dtype=None): argument 830 def group_conv2d_nhwc(Input, Filter, stride, padding, dilation, groups, out_dtype=None): argument [all …]
|
H A D | conv1d.py | 25 def conv1d(data, kernel, strides=1, padding="VALID", dilation=1, layout="NCW", out_dtype=None): argument 67 def conv1d_ncw(data, kernel, strides=1, padding="VALID", dilation=1, out_dtype=None): argument 127 def conv1d_nwc(data, kernel, strides=1, padding="VALID", dilation=1, out_dtype=None): argument
|
H A D | conv2d_transpose.py | 28 def conv2d_transpose_nchw(Input, Filter, strides, padding, out_dtype, output_padding): argument 61 def conv2d_transpose_nchw_preprocess(data, kernel, strides, padding, out_dtype, output_padding): argument 89 def declaration_conv2d_transpose_impl(data, kernel, strides, padding, out_dtype, output_padding): argument
|
H A D | conv3d_transpose.py | 28 def conv3d_transpose_ncdhw(Input, Filter, strides, padding, out_dtype, output_padding): argument 61 def conv3d_transpose_ncdhw_preprocess(data, kernel, strides, padding, out_dtype, output_padding): argument 96 def declaration_conv3d_transpose_impl(data, kernel, strides, padding, out_dtype, output_padding): argument
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/relay/op/annotation/ |
H A D | annotation.cc | 80 const Type& out_dtype) -> Array<te::Tensor> { in __anon86cb13ea0302() 106 const Type& out_dtype) -> Array<te::Tensor> { in __anon86cb13ea0402() 122 const Type& out_dtype) -> Array<te::Tensor> { in __anon86cb13ea0502() 138 const Type& out_dtype) -> Array<te::Tensor> { in __anon86cb13ea0602() 159 const Type& out_dtype) -> Array<te::Tensor> { in __anon86cb13ea0802() 181 const Type& out_dtype) -> Array<te::Tensor> { in __anon86cb13ea0902() 205 const Type& out_dtype) -> Array<te::Tensor> { in __anon86cb13ea0b02()
|
/dports/misc/tvm/incubator-tvm-0.6.1/src/relay/op/annotation/ |
H A D | annotation.cc | 82 const Type& out_dtype, const Target& target) -> Array<Tensor> { in __anon5d58c44f0302() 108 const Type& out_dtype, const Target& target) -> Array<Tensor> { in __anon5d58c44f0402() 126 const Type& out_dtype, const Target& target) -> Array<Tensor> { in __anon5d58c44f0502() 143 const Type& out_dtype, const Target& target) -> Array<Tensor> { in __anon5d58c44f0602() 166 const Type& out_dtype, const Target& target) -> Array<Tensor> { in __anon5d58c44f0802()
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/src/relay/op/annotation/ |
H A D | annotation.cc | 82 const Type& out_dtype, const Target& target) -> Array<Tensor> { in __anonb93c1a250302() 108 const Type& out_dtype, const Target& target) -> Array<Tensor> { in __anonb93c1a250402() 126 const Type& out_dtype, const Target& target) -> Array<Tensor> { in __anonb93c1a250502() 143 const Type& out_dtype, const Target& target) -> Array<Tensor> { in __anonb93c1a250602() 166 const Type& out_dtype, const Target& target) -> Array<Tensor> { in __anonb93c1a250802()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/relay/op/nn/ |
H A D | _nn.py | 459 def compute_lrn(attrs, inputs, out_dtype): argument 471 def compute_upsampling(attrs, inputs, out_dtype): argument 485 def compute_upsampling3d(attrs, inputs, out_dtype): argument 508 def compute_mirror_pad(attrs, inputs, out_dtype): argument 552 def compute_contrib_conv2d_gemm_weight_transform(attrs, inputs, out_dtype): argument 611 def compute_bitpack(attrs, inputs, out_dtype): argument 659 def compute_cross_entropy(attrs, inputs, out_dtype): argument 670 def compute_dilate(attrs, inputs, out_dtype): argument 680 def compute_cross_entropy_with_logits(attrs, inputs, out_dtype): argument 691 def compute_depth_to_space(attrs, inputs, out_dtype): argument [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/relay/op/image/ |
H A D | resize.cc | 56 DataType out_dtype = param->out_dtype; in ResizeRel() local 69 String coordinate_transformation_mode, DataType out_dtype) { in MakeResize() 126 DataType out_dtype = param->out_dtype; in Resize3dRel() local 139 String coordinate_transformation_mode, DataType out_dtype) { in MakeResize3d() 188 DataType out_dtype = param->out_dtype; in CropAndResizeRel() local 209 DataType out_dtype) { in MakeCropAndResize()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/relay/op/nn/ |
H A D | convolution.cc | 61 String out_layout, DataType out_dtype) { in __anon9765ad300102() 95 String out_layout, DataType out_dtype) { in __anon9765ad300202() 129 String out_layout, DataType out_dtype) { in __anon9765ad300302() 164 String out_layout, Array<IndexExpr> output_padding, DataType out_dtype) { in __anon9765ad300402() 209 String out_layout, Array<IndexExpr> output_padding, DataType out_dtype) { in __anon9765ad300502() 294 String kernel_layout, String out_layout, DataType out_dtype) { in __anon9765ad300702() 352 String kernel_layout, String out_layout, DataType out_dtype) { in __anon9765ad300902() 405 DataType out_dtype) { in MakeConv2DWinogradNNPACKWeightTransform() 490 String out_layout, DataType out_dtype) { in __anon9765ad300d02() 517 String out_layout, DataType out_dtype) { in __anon9765ad300e02() [all …]
|
/dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/arm_cpu/ |
H A D | conv2d.py | 42 def conv2d_arm_cpu(cfg, data, kernel, strides, padding, dilation, layout, out_dtype): argument 140 def conv2d_arm_cpu_winograd(cfg, data, kernel, strides, padding, dilation, layout, out_dtype): argument 146 def _decl_winograd(cfg, data, kernel, strides, padding, dilation, layout, out_dtype, tile_size): argument 327 cfg, data, kernel, strides, padding, dilation, layout, out_dtype): argument 336 cfg, data, kernel, strides, padding, dilation, layout, out_dtype): argument 344 cfg, data, kernel, strides, padding, dilation, layout, out_dtype, convolution_algorithm): argument 399 def conv2d_winograd_ww(cfg, data, kernel, strides, padding, dilation, layout, out_dtype, tile_size): argument 425 padding, dilation, layout, out_dtype): argument
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/arm_cpu/ |
H A D | conv2d.py | 40 def conv2d_nchw_spatial_pack(cfg, data, kernel, strides, padding, dilation, out_dtype): argument 77 def conv2d_nhwc_spatial_pack(cfg, data, kernel, strides, padding, dilation, out_dtype): argument 96 def conv2d_nchw_winograd(cfg, data, kernel, strides, padding, dilation, out_dtype): argument 116 def _decl_winograd(cfg, data, kernel, strides, padding, dilation, out_dtype, tile_size): argument 338 def conv2d_nchw_winograd_nnpack(cfg, data, kernel, strides, padding, dilation, out_dtype): argument 382 cfg, data, kernel, strides, padding, dilation, out_dtype, convolution_algorithm argument 450 cfg, data, transformed_kernel, bias, strides, padding, dilation, out_dtype argument 509 def conv2d_direct_simd(cfg, data, kernel, strides, padding, dilation, out_dtype): argument
|
H A D | conv2d_int8.py | 30 def _get_default_config(cfg, data, kernel, strides, padding, out_dtype): argument 45 def conv2d_NCHWc_int8(cfg, data, kernel, strides, padding, dilation, layout, out_layout, out_dtype): argument 119 def compute_conv2d_NHWC_quantized(cfg, data, kernel, strides, padding, dilation, out_dtype): argument 132 cfg, data, B, strides, padding, dilation, out_dtype, kernel_size=None, output_channels=None argument
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/arm_cpu/ |
H A D | conv2d.py | 42 def conv2d_arm_cpu(cfg, data, kernel, strides, padding, dilation, layout, out_dtype): argument 140 def conv2d_arm_cpu_winograd(cfg, data, kernel, strides, padding, dilation, layout, out_dtype): argument 146 def _decl_winograd(cfg, data, kernel, strides, padding, dilation, layout, out_dtype, tile_size): argument 327 cfg, data, kernel, strides, padding, dilation, layout, out_dtype): argument 336 cfg, data, kernel, strides, padding, dilation, layout, out_dtype): argument 344 cfg, data, kernel, strides, padding, dilation, layout, out_dtype, convolution_algorithm): argument 399 def conv2d_winograd_ww(cfg, data, kernel, strides, padding, dilation, layout, out_dtype, tile_size): argument 425 padding, dilation, layout, out_dtype): argument
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/x86/ |
H A D | dense.py | 142 def dense_nopack(cfg, data, weight, bias=None, out_dtype=None): argument 186 def dense_pack(cfg, data, weight, bias=None, out_dtype=None): argument 235 def dense_blas_common(cfg, data, weight, bias, out_dtype, lib): argument 259 def dense_cblas(cfg, data, weight, bias=None, out_dtype=None): argument 271 def dense_mkl(cfg, data, weight, bias=None, out_dtype=None): argument 283 def dense_mkldnn(cfg, data, weight, bias=None, out_dtype=None): argument
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/contrib/ |
H A D | test_cublas.py | 25 def verify_matmul_add(in_dtype, out_dtype, rtol=1e-5): argument 55 def verify_matmul_add_igemm(in_dtype, out_dtype, rtol=1e-5): argument 115 def verify_batch_matmul(in_dtype, out_dtype, rtol=1e-5): argument
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/image/ |
H A D | resize.py | 74 out_dtype=None, argument 132 def _cast_output(value, data_dtype="float32", out_dtype=None): argument 219 out_dtype=None, argument 277 def _cast_output(value, data_dtype="float32", out_dtype=None): argument 418 out_dtype=None, argument 482 def _cast_output(value, data_dtype="float32", out_dtype=None): argument 614 out_dtype=None, argument 740 out_dtype=None, argument 852 out_dtype=None, argument
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/relay/op/image/ |
H A D | image.py | 29 out_dtype=None, argument 82 out_dtype=None, argument 132 out_dtype=None, argument 185 out_dtype="", argument
|
/dports/misc/tvm/incubator-tvm-0.6.1/python/tvm/relay/op/nn/ |
H A D | _nn.py | 253 def compute_conv2d_transpose(attrs, inputs, out_dtype, target): argument 391 def compute_lrn(attrs, inputs, out_dtype, target): argument 410 def compute_l2_normalize(attrs, inputs, out_dtype, target): argument 434 def compute_upsampling(attrs, inputs, out_dtype, target): argument 449 def compute_mirror_pad(attrs, inputs, out_dtype, target): argument 511 attrs, inputs, out_dtype, target): argument 564 def compute_contrib_conv2d_NCHWc(attrs, inputs, out_dtype, target): argument 648 def compute_deformable_conv2d(attrs, inputs, out_dtype, target): argument 674 def compute_bitpack(attrs, inputs, out_dtype, target): argument 695 def compute_bitserial_conv2d(attrs, inputs, out_dtype, target): argument [all …]
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/python/tvm/relay/op/nn/ |
H A D | _nn.py | 253 def compute_conv2d_transpose(attrs, inputs, out_dtype, target): argument 391 def compute_lrn(attrs, inputs, out_dtype, target): argument 410 def compute_l2_normalize(attrs, inputs, out_dtype, target): argument 434 def compute_upsampling(attrs, inputs, out_dtype, target): argument 449 def compute_mirror_pad(attrs, inputs, out_dtype, target): argument 511 attrs, inputs, out_dtype, target): argument 564 def compute_contrib_conv2d_NCHWc(attrs, inputs, out_dtype, target): argument 648 def compute_deformable_conv2d(attrs, inputs, out_dtype, target): argument 674 def compute_bitpack(attrs, inputs, out_dtype, target): argument 695 def compute_bitserial_conv2d(attrs, inputs, out_dtype, target): argument [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/cuda/ |
H A D | conv2d_hwnc_tensorcore.py | 34 def unpack_HWNCnc_to_hwnc(packed_out, out_dtype): argument 67 def conv2d_hwnc_tensorcore(data, kernel, strides, padding, dilation, in_dtype, out_dtype="int32"): argument 76 def hwnc_tensorcore_cuda(cfg, Input, Filter, stride, padding, dilation, out_dtype="int32"): argument
|