| /frameworks/ml/nn/runtime/test/generated/models/ |
| D | l2_normalization_axis.model.cpp | 8 auto axis = model->addOperand(&type1); in CreateModel_dim4_axis0() local
31 auto axis = model->addOperand(&type1); in CreateModel_dim4_axis0_relaxed() local
56 auto axis = model->addOperand(&type1); in CreateModel_dim4_axis0_float16() local
80 auto axis = model->addOperand(&type1); in CreateModel_dim4_axis0_quant8() local
103 auto axis = model->addOperand(&type1); in CreateModel_dim4_axis0_neg() local
126 auto axis = model->addOperand(&type1); in CreateModel_dim4_axis0_neg_relaxed() local
151 auto axis = model->addOperand(&type1); in CreateModel_dim4_axis0_neg_float16() local
175 auto axis = model->addOperand(&type1); in CreateModel_dim4_axis0_neg_quant8() local
198 auto axis = model->addOperand(&type1); in CreateModel_dim4_axis1() local
221 auto axis = model->addOperand(&type1); in CreateModel_dim4_axis1_relaxed() local
[all …]
|
| D | channel_shuffle.model.cpp | 9 auto axis = model->addOperand(&type1); in CreateModel_dim4_axis0() local
35 auto axis = model->addOperand(&type1); in CreateModel_dim4_axis0_neg() local
61 auto axis = model->addOperand(&type1); in CreateModel_dim4_axis1() local
87 auto axis = model->addOperand(&type1); in CreateModel_dim4_axis1_neg() local
113 auto axis = model->addOperand(&type1); in CreateModel_dim4_axis2() local
139 auto axis = model->addOperand(&type1); in CreateModel_dim4_axis2_neg() local
165 auto axis = model->addOperand(&type1); in CreateModel_dim4_axis3() local
191 auto axis = model->addOperand(&type1); in CreateModel_dim4_axis3_neg() local
217 auto axis = model->addOperand(&type1); in CreateModel_dim3_axis0() local
243 auto axis = model->addOperand(&type1); in CreateModel_dim3_axis0_neg() local
[all …]
|
| D | argmin_1.model.cpp | 9 auto axis = model->addOperand(&type1); in CreateModel() local
33 auto axis = model->addOperand(&type1); in CreateModel_relaxed() local
59 auto axis = model->addOperand(&type1); in CreateModel_float16() local
83 auto axis = model->addOperand(&type1); in CreateModel_int32() local
107 auto axis = model->addOperand(&type1); in CreateModel_quant8() local
131 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
155 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_relaxed() local
181 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_float16() local
205 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_int32() local
229 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_quant8() local
|
| D | argmax_3.model.cpp | 9 auto axis = model->addOperand(&type1); in CreateModel() local
33 auto axis = model->addOperand(&type1); in CreateModel_relaxed() local
59 auto axis = model->addOperand(&type1); in CreateModel_float16() local
83 auto axis = model->addOperand(&type1); in CreateModel_int32() local
107 auto axis = model->addOperand(&type1); in CreateModel_quant8() local
131 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
155 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_relaxed() local
181 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_float16() local
205 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_int32() local
229 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_quant8() local
|
| D | argmin_3.model.cpp | 9 auto axis = model->addOperand(&type1); in CreateModel() local
33 auto axis = model->addOperand(&type1); in CreateModel_relaxed() local
59 auto axis = model->addOperand(&type1); in CreateModel_float16() local
83 auto axis = model->addOperand(&type1); in CreateModel_int32() local
107 auto axis = model->addOperand(&type1); in CreateModel_quant8() local
131 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
155 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_relaxed() local
181 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_float16() local
205 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_int32() local
229 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_quant8() local
|
| D | argmax_1.model.cpp | 9 auto axis = model->addOperand(&type1); in CreateModel() local
33 auto axis = model->addOperand(&type1); in CreateModel_relaxed() local
59 auto axis = model->addOperand(&type1); in CreateModel_float16() local
83 auto axis = model->addOperand(&type1); in CreateModel_int32() local
107 auto axis = model->addOperand(&type1); in CreateModel_quant8() local
131 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
155 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_relaxed() local
181 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_float16() local
205 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_int32() local
229 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_quant8() local
|
| D | argmin_2.model.cpp | 9 auto axis = model->addOperand(&type1); in CreateModel() local
33 auto axis = model->addOperand(&type1); in CreateModel_relaxed() local
59 auto axis = model->addOperand(&type1); in CreateModel_float16() local
83 auto axis = model->addOperand(&type1); in CreateModel_int32() local
107 auto axis = model->addOperand(&type1); in CreateModel_quant8() local
131 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
155 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_relaxed() local
181 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_float16() local
205 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_int32() local
229 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_quant8() local
|
| D | argmax_2.model.cpp | 9 auto axis = model->addOperand(&type1); in CreateModel() local
33 auto axis = model->addOperand(&type1); in CreateModel_relaxed() local
59 auto axis = model->addOperand(&type1); in CreateModel_float16() local
83 auto axis = model->addOperand(&type1); in CreateModel_int32() local
107 auto axis = model->addOperand(&type1); in CreateModel_quant8() local
131 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
155 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_relaxed() local
181 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_float16() local
205 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_int32() local
229 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_quant8() local
|
| D | softmax_v1_2.model.cpp | 1168 auto axis = model->addOperand(&type1); in CreateModel_axis_dim4_axis0() local
1195 auto axis = model->addOperand(&type1); in CreateModel_axis_dim4_axis0_neg() local
1222 auto axis = model->addOperand(&type1); in CreateModel_axis_dim4_axis1() local
1249 auto axis = model->addOperand(&type1); in CreateModel_axis_dim4_axis1_neg() local
1276 auto axis = model->addOperand(&type1); in CreateModel_axis_dim4_axis2() local
1303 auto axis = model->addOperand(&type1); in CreateModel_axis_dim4_axis2_neg() local
1330 auto axis = model->addOperand(&type1); in CreateModel_axis_dim4_axis3() local
1357 auto axis = model->addOperand(&type1); in CreateModel_axis_dim4_axis3_neg() local
1384 auto axis = model->addOperand(&type1); in CreateModel_axis_dim3_axis0() local
1411 auto axis = model->addOperand(&type1); in CreateModel_axis_dim3_axis0_neg() local
[all …]
|
| D | split_float_4.model.cpp | 9 auto axis = model->addOperand(&type1); in CreateModel() local
37 auto axis = model->addOperand(&type1); in CreateModel_relaxed() local
67 auto axis = model->addOperand(&type1); in CreateModel_float16() local
95 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
123 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_relaxed() local
153 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_float16() local
|
| D | split_float_2.model.cpp | 9 auto axis = model->addOperand(&type1); in CreateModel() local
37 auto axis = model->addOperand(&type1); in CreateModel_relaxed() local
67 auto axis = model->addOperand(&type1); in CreateModel_float16() local
95 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
123 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_relaxed() local
153 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_float16() local
|
| D | split_float_5.model.cpp | 9 auto axis = model->addOperand(&type1); in CreateModel() local
37 auto axis = model->addOperand(&type1); in CreateModel_relaxed() local
67 auto axis = model->addOperand(&type1); in CreateModel_float16() local
95 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
123 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_relaxed() local
153 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_float16() local
|
| D | split_float_3.model.cpp | 9 auto axis = model->addOperand(&type1); in CreateModel() local
38 auto axis = model->addOperand(&type1); in CreateModel_relaxed() local
69 auto axis = model->addOperand(&type1); in CreateModel_float16() local
98 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
127 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_relaxed() local
158 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_float16() local
|
| D | split_float_1.model.cpp | 9 auto axis = model->addOperand(&type1); in CreateModel() local
38 auto axis = model->addOperand(&type1); in CreateModel_relaxed() local
69 auto axis = model->addOperand(&type1); in CreateModel_float16() local
98 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
127 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_relaxed() local
158 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_float16() local
|
| D | split_int32_2.model.cpp | 9 auto axis = model->addOperand(&type1); in CreateModel() local
37 auto axis = model->addOperand(&type1); in CreateModel_relaxed() local
67 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
95 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_relaxed() local
|
| D | split_int32_4.model.cpp | 9 auto axis = model->addOperand(&type1); in CreateModel() local
37 auto axis = model->addOperand(&type1); in CreateModel_relaxed() local
67 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
95 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_relaxed() local
|
| D | split_quant8_2.model.cpp | 9 auto axis = model->addOperand(&type1); in CreateModel() local
37 auto axis = model->addOperand(&type1); in CreateModel_relaxed() local
67 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
95 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_relaxed() local
|
| D | local_response_normalization_v1_2.model.cpp | 13 auto axis = model->addOperand(&type1); in CreateModel_axis_dim4_axis0() local
49 auto axis = model->addOperand(&type1); in CreateModel_axis_dim4_axis0_float16() local
85 auto axis = model->addOperand(&type1); in CreateModel_axis_dim4_axis0_neg() local
121 auto axis = model->addOperand(&type1); in CreateModel_axis_dim4_axis0_neg_float16() local
157 auto axis = model->addOperand(&type1); in CreateModel_axis_dim4_axis1() local
193 auto axis = model->addOperand(&type1); in CreateModel_axis_dim4_axis1_float16() local
229 auto axis = model->addOperand(&type1); in CreateModel_axis_dim4_axis1_neg() local
265 auto axis = model->addOperand(&type1); in CreateModel_axis_dim4_axis1_neg_float16() local
301 auto axis = model->addOperand(&type1); in CreateModel_axis_dim4_axis2() local
337 auto axis = model->addOperand(&type1); in CreateModel_axis_dim4_axis2_float16() local
[all …]
|
| D | split_int32_3.model.cpp | 9 auto axis = model->addOperand(&type1); in CreateModel() local
38 auto axis = model->addOperand(&type1); in CreateModel_relaxed() local
69 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
98 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_relaxed() local
|
| D | split_int32_1.model.cpp | 9 auto axis = model->addOperand(&type1); in CreateModel() local
38 auto axis = model->addOperand(&type1); in CreateModel_relaxed() local
69 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape() local
98 auto axis = model->addOperand(&type1); in CreateModel_dynamic_output_shape_relaxed() local
|
| /frameworks/ml/nn/common/operations/ |
| D | Split.cpp | 28 bool splitGeneric(const Scalar* inputData, const Shape& inputShape, int32_t axis, in splitGeneric()
54 bool splitFloat16(const _Float16* inputData, const Shape& inputShape, int32_t axis, in splitFloat16()
61 bool splitFloat32(const float* inputData, const Shape& inputShape, int32_t axis, in splitFloat32()
68 bool splitQuant8(const uint8_t* inputData, const Shape& inputShape, int32_t axis, in splitQuant8()
75 bool splitInt32(const int32_t* inputData, const Shape& inputShape, int32_t axis, in splitInt32()
|
| D | L2Normalization.cpp | 39 inline bool l2normFloat32Impl(const float* inputData, const Shape& inputShape, int32_t axis, in l2normFloat32Impl()
66 inline bool l2normQuant8Impl(const uint8_t* inputData, const Shape& inputShape, int32_t axis, in l2normQuant8Impl()
98 bool l2normFloat32(const float* inputData, const Shape& inputShape, int32_t axis, float* outputData, in l2normFloat32()
114 bool l2normFloat16(const _Float16* inputData, const Shape& inputShape, int32_t axis, in l2normFloat16()
127 bool l2normQuant8(const uint8_t* inputData, const Shape& inputShape, int32_t axis, in l2normQuant8()
172 int32_t axis = context->getNumInputs() == kNumInputs in prepare() local
191 int32_t axis = context->getNumInputs() == kNumInputs in execute() local
|
| D | ExpandDims.cpp | 27 bool prepare(const Shape& input, int32_t axis, Shape* output) { in prepare()
40 bool eval(const uint8_t* inputData, const Shape& inputShape, int32_t axis, uint8_t* outputData, in eval()
|
| D | Gather.cpp | 41 inline bool eval(const T* inputData, const Shape& inputShape, int32_t axis, in eval()
79 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in prepare() local
97 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in execute() local
|
| D | ChannelShuffle.cpp | 39 inline bool eval(const T* inputData, const Shape& inputShape, int32_t numGroups, int32_t axis, in eval()
77 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in prepare() local
86 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in execute() local
|