Home
last modified time | relevance | path

Searched defs:out_dtype (Results 1 – 25 of 371) sorted by relevance

12345678910>>...15

/dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/nn/
H A Dconv2d.py35 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 Dconv2d_transpose.py29 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 Dconv2d.py35 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 Dconv2d_transpose.py29 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 Dconv2d.py51 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 Dconv1d.py25 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 Dconv2d_transpose.py28 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 Dconv3d_transpose.py28 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 Dannotation.cc80 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 Dannotation.cc82 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 Dannotation.cc82 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.py459 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 Dresize.cc56 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 Dconvolution.cc61 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 Dconv2d.py42 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 Dconv2d.py40 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 Dconv2d_int8.py30 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 Dconv2d.py42 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 Ddense.py142 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 Dtest_cublas.py25 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 Dresize.py74 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 Dimage.py29 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.py253 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.py253 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 Dconv2d_hwnc_tensorcore.py34 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

12345678910>>...15