Home
last modified time | relevance | path

Searched refs:TVM_ATTR_FIELD (Results 1 – 25 of 61) sorted by relevance

123

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/include/tvm/relay/attrs/
H A Dnn.h64 TVM_ATTR_FIELD(strides)
69 TVM_ATTR_FIELD(padding)
74 TVM_ATTR_FIELD(dilation)
81 TVM_ATTR_FIELD(channels)
124 TVM_ATTR_FIELD(strides)
127 TVM_ATTR_FIELD(padding)
135 TVM_ATTR_FIELD(dilation)
218 TVM_ATTR_FIELD(strides)
221 TVM_ATTR_FIELD(padding)
1045 TVM_ATTR_FIELD(mode)
[all …]
H A Dtransform.h56 TVM_ATTR_FIELD(num_newaxis)
67 TVM_ATTR_FIELD(axis)
90 TVM_ATTR_FIELD(reverse)
116 TVM_ATTR_FIELD(axis)
127 TVM_ATTR_FIELD(axis)
169 TVM_ATTR_FIELD(indexing)
192 TVM_ATTR_FIELD(axis)
212 TVM_ATTR_FIELD(axis)
226 TVM_ATTR_FIELD(batch_axis)
238 TVM_ATTR_FIELD(axis)
[all …]
H A Dvision.h44 TVM_ATTR_FIELD(sizes)
47 TVM_ATTR_FIELD(ratios)
50 TVM_ATTR_FIELD(steps)
53 TVM_ATTR_FIELD(offsets)
68 TVM_ATTR_FIELD(variances)
103 TVM_ATTR_FIELD(iou_threshold)
111 TVM_ATTR_FIELD(coord_start)
133 TVM_ATTR_FIELD(spatial_scale)
138 TVM_ATTR_FIELD(sample_ratio)
190 TVM_ATTR_FIELD(scales)
[all …]
H A Dbitserial.h49 TVM_ATTR_FIELD(pack_type)
71 TVM_ATTR_FIELD(strides)
74 TVM_ATTR_FIELD(padding)
79 TVM_ATTR_FIELD(kernel_size)
82 TVM_ATTR_FIELD(channels)
88 TVM_ATTR_FIELD(weight_bits)
91 TVM_ATTR_FIELD(data_layout)
94 TVM_ATTR_FIELD(kernel_layout)
97 TVM_ATTR_FIELD(pack_dtype)
119 TVM_ATTR_FIELD(weight_bits)
[all …]
H A Dimage.h50 TVM_ATTR_FIELD(method)
83 TVM_ATTR_FIELD(method)
114 TVM_ATTR_FIELD(method)
120 TVM_ATTR_FIELD(extrapolation_value)
137 TVM_ATTR_FIELD(strides)
140 TVM_ATTR_FIELD(padding)
148 TVM_ATTR_FIELD(dilations)
151 TVM_ATTR_FIELD(data_layout)
158 TVM_ATTR_FIELD(kernel_layout)
164 TVM_ATTR_FIELD(out_dtype)
[all …]
H A Dalgorithm.h43 TVM_ATTR_FIELD(axis).set_default(-1).describe(
46 TVM_ATTR_FIELD(is_ascend).set_default(true).describe(
49 TVM_ATTR_FIELD(dtype)
63 TVM_ATTR_FIELD(k).describe("Number of top elements to select");
64 TVM_ATTR_FIELD(axis).set_default(-1).describe("Axis along which to sort the input tensor.");
65 TVM_ATTR_FIELD(ret_type).set_default("both").describe(
70 TVM_ATTR_FIELD(is_ascend).set_default(false).describe(
73 TVM_ATTR_FIELD(dtype)
H A Dreduce.h41 TVM_ATTR_FIELD(axis)
56 TVM_ATTR_FIELD(keepdims).set_default(false).describe(
59 TVM_ATTR_FIELD(exclude).set_default(false).describe(
71 TVM_ATTR_FIELD(axis)
86 TVM_ATTR_FIELD(keepdims).set_default(false).describe(
89 TVM_ATTR_FIELD(exclude).set_default(false).describe(
91 TVM_ATTR_FIELD(unbiased).set_default(false).describe("Whether to use the unbiased estimation.");
H A Dmemory.h49 TVM_ATTR_FIELD(dtype)
52 TVM_ATTR_FIELD(device_id).describe("The device id on which to allocate memory.");
53 TVM_ATTR_FIELD(device_type).describe("The device type on which to allocate memory.");
66 TVM_ATTR_FIELD(dtype)
69 TVM_ATTR_FIELD(const_shape).describe("The shape of constant used to aid in type inference.");
70 TVM_ATTR_FIELD(assert_shape)
/dports/misc/tvm/incubator-tvm-0.6.1/include/tvm/relay/attrs/
H A Dnn.h45 TVM_ATTR_FIELD(axis)
78 TVM_ATTR_FIELD(channels)
149 TVM_ATTR_FIELD(channels)
359 TVM_ATTR_FIELD(units)
475 TVM_ATTR_FIELD(rate)
489 TVM_ATTR_FIELD(axis)
495 TVM_ATTR_FIELD(center)
498 TVM_ATTR_FIELD(scale)
515 TVM_ATTR_FIELD(axis)
580 TVM_ATTR_FIELD(eps)
[all …]
H A Dvision.h43 TVM_ATTR_FIELD(sizes)
46 TVM_ATTR_FIELD(ratios)
49 TVM_ATTR_FIELD(steps)
53 TVM_ATTR_FIELD(offsets)
74 TVM_ATTR_FIELD(variances)
139 TVM_ATTR_FIELD(spatial_scale)
144 TVM_ATTR_FIELD(sample_ratio)
180 TVM_ATTR_FIELD(stride)
198 TVM_ATTR_FIELD(scales)
201 TVM_ATTR_FIELD(ratios)
[all …]
H A Dtransform.h39 TVM_ATTR_FIELD(dtype)
50 TVM_ATTR_FIELD(axis)
66 TVM_ATTR_FIELD(axis)
77 TVM_ATTR_FIELD(axes)
116 TVM_ATTR_FIELD(shape)
118 TVM_ATTR_FIELD(dtype)
134 TVM_ATTR_FIELD(stop)
136 TVM_ATTR_FIELD(step)
168 TVM_ATTR_FIELD(reps)
189 TVM_ATTR_FIELD(axis)
[all …]
H A Dbitserial.h48 TVM_ATTR_FIELD(pack_type)
70 TVM_ATTR_FIELD(strides)
73 TVM_ATTR_FIELD(padding)
78 TVM_ATTR_FIELD(kernel_size)
81 TVM_ATTR_FIELD(channels)
96 TVM_ATTR_FIELD(pack_dtype)
115 TVM_ATTR_FIELD(units)
117 TVM_ATTR_FIELD(data_bits)
123 TVM_ATTR_FIELD(pack_dtype)
126 TVM_ATTR_FIELD(out_dtype)
[all …]
H A Dalgorithm.h41 TVM_ATTR_FIELD(axis).set_default(-1)
44 TVM_ATTR_FIELD(is_ascend).set_default(true)
47 TVM_ATTR_FIELD(dtype).set_default(NullValue<DataType>())
60 TVM_ATTR_FIELD(k).set_default(1)
62 TVM_ATTR_FIELD(axis).set_default(-1)
64 TVM_ATTR_FIELD(ret_type).set_default("both")
69 TVM_ATTR_FIELD(is_ascend).set_default(false)
72 TVM_ATTR_FIELD(dtype).set_default(NullValue<DataType>())
H A Dimage.h43 TVM_ATTR_FIELD(size).set_default(NullValue<Array<IndexExpr> >())
45 TVM_ATTR_FIELD(layout).set_default("NCHW")
50 TVM_ATTR_FIELD(method).set_default("bilinear")
55 TVM_ATTR_FIELD(align_corners).set_default(true)
57 TVM_ATTR_FIELD(out_dtype)
H A Dmemory.h43 TVM_ATTR_FIELD(dtype)
47 TVM_ATTR_FIELD(const_shape)
50 TVM_ATTR_FIELD(assert_shape)
64 TVM_ATTR_FIELD(is_input)
/dports/misc/py-tvm/incubator-tvm-0.6.1/include/tvm/relay/attrs/
H A Dnn.h45 TVM_ATTR_FIELD(axis)
78 TVM_ATTR_FIELD(channels)
149 TVM_ATTR_FIELD(channels)
359 TVM_ATTR_FIELD(units)
475 TVM_ATTR_FIELD(rate)
489 TVM_ATTR_FIELD(axis)
495 TVM_ATTR_FIELD(center)
498 TVM_ATTR_FIELD(scale)
515 TVM_ATTR_FIELD(axis)
580 TVM_ATTR_FIELD(eps)
[all …]
H A Dvision.h43 TVM_ATTR_FIELD(sizes)
46 TVM_ATTR_FIELD(ratios)
49 TVM_ATTR_FIELD(steps)
53 TVM_ATTR_FIELD(offsets)
74 TVM_ATTR_FIELD(variances)
139 TVM_ATTR_FIELD(spatial_scale)
144 TVM_ATTR_FIELD(sample_ratio)
180 TVM_ATTR_FIELD(stride)
198 TVM_ATTR_FIELD(scales)
201 TVM_ATTR_FIELD(ratios)
[all …]
H A Dtransform.h39 TVM_ATTR_FIELD(dtype)
50 TVM_ATTR_FIELD(axis)
66 TVM_ATTR_FIELD(axis)
77 TVM_ATTR_FIELD(axes)
116 TVM_ATTR_FIELD(shape)
118 TVM_ATTR_FIELD(dtype)
134 TVM_ATTR_FIELD(stop)
136 TVM_ATTR_FIELD(step)
168 TVM_ATTR_FIELD(reps)
189 TVM_ATTR_FIELD(axis)
[all …]
H A Dbitserial.h48 TVM_ATTR_FIELD(pack_type)
70 TVM_ATTR_FIELD(strides)
73 TVM_ATTR_FIELD(padding)
78 TVM_ATTR_FIELD(kernel_size)
81 TVM_ATTR_FIELD(channels)
96 TVM_ATTR_FIELD(pack_dtype)
115 TVM_ATTR_FIELD(units)
117 TVM_ATTR_FIELD(data_bits)
123 TVM_ATTR_FIELD(pack_dtype)
126 TVM_ATTR_FIELD(out_dtype)
[all …]
H A Dalgorithm.h41 TVM_ATTR_FIELD(axis).set_default(-1)
44 TVM_ATTR_FIELD(is_ascend).set_default(true)
47 TVM_ATTR_FIELD(dtype).set_default(NullValue<DataType>())
60 TVM_ATTR_FIELD(k).set_default(1)
62 TVM_ATTR_FIELD(axis).set_default(-1)
64 TVM_ATTR_FIELD(ret_type).set_default("both")
69 TVM_ATTR_FIELD(is_ascend).set_default(false)
72 TVM_ATTR_FIELD(dtype).set_default(NullValue<DataType>())
H A Dimage.h43 TVM_ATTR_FIELD(size).set_default(NullValue<Array<IndexExpr> >())
45 TVM_ATTR_FIELD(layout).set_default("NCHW")
50 TVM_ATTR_FIELD(method).set_default("bilinear")
55 TVM_ATTR_FIELD(align_corners).set_default(true)
57 TVM_ATTR_FIELD(out_dtype)
/dports/misc/tvm/incubator-tvm-0.6.1/include/tvm/relay/qnn/
H A Dattrs.h44 TVM_ATTR_FIELD(input_scale)
48 TVM_ATTR_FIELD(output_scale)
62 TVM_ATTR_FIELD(out_dtype)
75 TVM_ATTR_FIELD(out_dtype)
114 TVM_ATTR_FIELD(axis)
157 TVM_ATTR_FIELD(channels)
177 TVM_ATTR_FIELD(out_dtype)
203 TVM_ATTR_FIELD(lhs_scale)
207 TVM_ATTR_FIELD(rhs_scale)
227 TVM_ATTR_FIELD(units)
[all …]
/dports/misc/py-tvm/incubator-tvm-0.6.1/include/tvm/relay/qnn/
H A Dattrs.h44 TVM_ATTR_FIELD(input_scale)
48 TVM_ATTR_FIELD(output_scale)
62 TVM_ATTR_FIELD(out_dtype)
75 TVM_ATTR_FIELD(out_dtype)
114 TVM_ATTR_FIELD(axis)
157 TVM_ATTR_FIELD(channels)
177 TVM_ATTR_FIELD(out_dtype)
203 TVM_ATTR_FIELD(lhs_scale)
207 TVM_ATTR_FIELD(rhs_scale)
227 TVM_ATTR_FIELD(units)
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/relay/backend/contrib/ethosn/
H A Dcodegen_ethosn.h248 TVM_ATTR_FIELD(variant)
251 TVM_ATTR_FIELD(strategy0).set_default(true);
252 TVM_ATTR_FIELD(strategy1).set_default(true);
253 TVM_ATTR_FIELD(strategy3).set_default(true);
254 TVM_ATTR_FIELD(strategy4).set_default(true);
255 TVM_ATTR_FIELD(strategy6).set_default(true);
256 TVM_ATTR_FIELD(strategy7).set_default(true);
257 TVM_ATTR_FIELD(dump_ram).set_default(false);
260 TVM_ATTR_FIELD(block_config_32x8).set_default(true);
262 TVM_ATTR_FIELD(block_config_8x8).set_default(true);
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/include/tvm/relay/qnn/
H A Dattrs.h42 TVM_ATTR_FIELD(axis)
47 TVM_ATTR_FIELD(rounding).set_default("UPWARD").describe(
57 TVM_ATTR_FIELD(out_dtype)
69 TVM_ATTR_FIELD(out_dtype).describe("Output data type, can be one of [int8 or uint8].");
70 TVM_ATTR_FIELD(axis)
83 TVM_ATTR_FIELD(axis)

123