| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | transpose_builder_test.cpp | 55 std::vector<uint32_t> inputsIndex = { 0, 1 }; variable
58 InitTensor(inputsIndex, outputsIndex);
71 std::vector<uint32_t> inputsIndex = { 0, 1 }; variable
74 InitTensor(inputsIndex, outputsIndex);
91 InitTensor(inputsIndex, outputsIndex);
104 std::vector<uint32_t> inputsIndex = { 0, 1 }; variable
107 InitTensor(inputsIndex, outputsIndex);
120 std::vector<uint32_t> inputsIndex = { 0, 1 }; variable
138 InitTensor(inputsIndex, outputsIndex);
156 InitTensor(inputsIndex, outputsIndex);
[all …]
|
| H A D | stack_builder_test.cpp | 75 InitTensor(inputsIndex, outputsIndex);
93 InitTensor(inputsIndex, outputsIndex);
112 InitTensor(inputsIndex, outputsIndex);
130 InitTensor(inputsIndex, outputsIndex);
163 InitTensor(inputsIndex, outputsIndex);
181 InitTensor(inputsIndex, outputsIndex);
199 InitTensor(inputsIndex, outputsIndex);
217 InitTensor(inputsIndex, outputsIndex);
238 InitTensor(inputsIndex, outputsIndex);
256 InitTensor(inputsIndex, outputsIndex);
[all …]
|
| H A D | unsqueeze_builder_test.cpp | 74 InitTensor(inputsIndex, outputsIndex);
92 InitTensor(inputsIndex, outputsIndex);
111 InitTensor(inputsIndex, outputsIndex);
129 InitTensor(inputsIndex, outputsIndex);
162 InitTensor(inputsIndex, outputsIndex);
180 InitTensor(inputsIndex, outputsIndex);
198 InitTensor(inputsIndex, outputsIndex);
216 InitTensor(inputsIndex, outputsIndex);
237 InitTensor(inputsIndex, outputsIndex);
255 InitTensor(inputsIndex, outputsIndex);
[all …]
|
| H A D | sub_builder_test.cpp | 74 InitTensor(inputsIndex, outputsIndex);
92 InitTensor(inputsIndex, outputsIndex);
111 InitTensor(inputsIndex, outputsIndex);
129 InitTensor(inputsIndex, outputsIndex);
162 InitTensor(inputsIndex, outputsIndex);
180 InitTensor(inputsIndex, outputsIndex);
198 InitTensor(inputsIndex, outputsIndex);
216 InitTensor(inputsIndex, outputsIndex);
238 InitTensor(inputsIndex, outputsIndex);
256 InitTensor(inputsIndex, outputsIndex);
[all …]
|
| H A D | tanh_builder_test.cpp | 55 std::vector<uint32_t> inputsIndex = { 0 }; variable
58 InitTensor(inputsIndex, outputsIndex);
71 std::vector<uint32_t> inputsIndex = { 0 }; variable
74 InitTensor(inputsIndex, outputsIndex);
91 InitTensor(inputsIndex, outputsIndex);
104 std::vector<uint32_t> inputsIndex = { 0 }; variable
107 InitTensor(inputsIndex, outputsIndex);
120 std::vector<uint32_t> inputsIndex = { 0 }; variable
135 std::vector<uint32_t> inputsIndex = { 0 }; variable
157 InitTensor(inputsIndex, outputsIndex);
[all …]
|
| H A D | squared_difference_builder_test.cpp | 31 void InitTensor(const std::vector<uint32_t>& inputsIndex,
63 std::vector<uint32_t> inputsIndex = { 0, 1 }; variable
81 std::vector<uint32_t> inputsIndex = { 0, 1 }; variable
100 std::vector<uint32_t> inputsIndex = { 0, 1, 2 }; variable
118 std::vector<uint32_t> inputsIndex = { 0, 1 }; variable
136 std::vector<uint32_t> inputsIndex = { 0, 1 }; variable
150 std::vector<uint32_t> inputsIndex = { 0, 1 }; variable
167 std::vector<uint32_t> inputsIndex = { 0, 1 }; variable
172 InitTensor(inputsIndex, outputsIndex);
196 std::vector<uint32_t> inputsIndex = { 0, 1 }; variable
[all …]
|
| H A D | softmax_builder_test.cpp | 71 std::vector<uint32_t> inputsIndex = { 0 }; variable
75 InitTensor(inputsIndex, outputsIndex);
93 InitTensor(inputsIndex, outputsIndex);
112 InitTensor(inputsIndex, outputsIndex);
130 InitTensor(inputsIndex, outputsIndex);
162 InitTensor(inputsIndex, outputsIndex);
201 InitTensor(inputsIndex, outputsIndex);
219 InitTensor(inputsIndex, outputsIndex);
237 InitTensor(inputsIndex, outputsIndex);
257 InitTensor(inputsIndex, outputsIndex);
[all …]
|
| H A D | spacetobatchnd_builder_test.cpp | 123 InitTensor(inputsIndex, outputsIndex);
142 InitTensor(inputsIndex, outputsIndex);
164 InitTensor(inputsIndex, outputsIndex);
184 InitTensor(inputsIndex, outputsIndex);
199 std::vector<uint32_t> inputsIndex = {}; variable
204 InitTensor(inputsIndex, outputsIndex);
224 InitTensor(inputsIndex, outputsIndex);
245 InitTensor(inputsIndex, outputsIndex);
314 InitTensor(inputsIndex, outputsIndex);
338 InitTensor(inputsIndex, outputsIndex);
[all …]
|
| H A D | squeeze_builder_test.cpp | 71 std::vector<uint32_t> inputsIndex = { 0 }; variable
75 InitTensor(inputsIndex, outputsIndex);
89 std::vector<uint32_t> inputsIndex = { 0 }; variable
93 InitTensor(inputsIndex, outputsIndex);
112 InitTensor(inputsIndex, outputsIndex);
130 InitTensor(inputsIndex, outputsIndex);
163 InitTensor(inputsIndex, outputsIndex);
181 InitTensor(inputsIndex, outputsIndex);
199 InitTensor(inputsIndex, outputsIndex);
220 InitTensor(inputsIndex, outputsIndex);
[all …]
|
| H A D | sqrt_builder_test.cpp | 31 void InitTensor(const std::vector<uint32_t>& inputsIndex,
63 std::vector<uint32_t> inputsIndex = { 0 }; variable
81 std::vector<uint32_t> inputsIndex = { 0 }; variable
100 std::vector<uint32_t> inputsIndex = { 0, 1 }; variable
118 std::vector<uint32_t> inputsIndex = { 0 }; variable
136 std::vector<uint32_t> inputsIndex = { 0 }; variable
150 std::vector<uint32_t> inputsIndex = { 0 }; variable
168 std::vector<uint32_t> inputsIndex = { 0 }; variable
173 InitTensor(inputsIndex, outputsIndex);
197 std::vector<uint32_t> inputsIndex = { 0 }; variable
[all …]
|
| H A D | split_builder_test.cpp | 103 InitTensor(inputsIndex, outputsIndex);
123 InitTensor(inputsIndex, outputsIndex);
144 InitTensor(inputsIndex, outputsIndex);
164 InitTensor(inputsIndex, outputsIndex);
199 InitTensor(inputsIndex, outputsIndex);
219 InitTensor(inputsIndex, outputsIndex);
239 InitTensor(inputsIndex, outputsIndex);
259 InitTensor(inputsIndex, outputsIndex);
279 InitTensor(inputsIndex, outputsIndex);
299 InitTensor(inputsIndex, outputsIndex);
[all …]
|
| H A D | topk_builder_test.cpp | 85 InitTensor(inputsIndex, outputsIndex);
105 InitTensor(inputsIndex, outputsIndex);
126 InitTensor(inputsIndex, outputsIndex);
146 InitTensor(inputsIndex, outputsIndex);
181 InitTensor(inputsIndex, outputsIndex);
201 InitTensor(inputsIndex, outputsIndex);
221 InitTensor(inputsIndex, outputsIndex);
244 InitTensor(inputsIndex, outputsIndex);
267 InitTensor(inputsIndex, outputsIndex);
287 InitTensor(inputsIndex, outputsIndex);
[all …]
|
| H A D | strided_slice_builder_test.cpp | 141 InitTensor(inputsIndex, outputsIndex);
158 InitTensor(inputsIndex, outputsIndex);
285 InitTensor(inputsIndex, outputsIndex);
306 InitTensor(inputsIndex, outputsIndex);
327 InitTensor(inputsIndex, outputsIndex);
348 InitTensor(inputsIndex, outputsIndex);
370 InitTensor(inputsIndex, outputsIndex);
396 InitTensor(inputsIndex, outputsIndex);
423 InitTensor(inputsIndex, outputsIndex);
450 InitTensor(inputsIndex, outputsIndex);
[all …]
|
| H A D | slice_builder_test.cpp | 72 InitTensor(inputsIndex, outputsIndex);
86 InitTensor(inputsIndex, outputsIndex);
101 inputsIndex = { 0, 1, 2, 3 };
105 InitTensor(inputsIndex, outputsIndex);
119 inputsIndex = { 0, 1, 2 };
123 InitTensor(inputsIndex, outputsIndex);
137 inputsIndex = {};
141 InitTensor(inputsIndex, outputsIndex);
155 inputsIndex = { 0, 1, 2 };
159 InitTensor(inputsIndex, outputsIndex);
[all …]
|
| H A D | tile_builder_test.cpp | 31 void InitTensor(const std::vector<uint32_t>& inputsIndex,
38 std::vector<uint32_t> inputsIndex = { 0, 1 };
51 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
72 InitTensor(inputsIndex, outputsIndex);
86 InitTensor(inputsIndex, outputsIndex);
101 inputsIndex = { 0, 1, 2 };
105 InitTensor(inputsIndex, outputsIndex);
122 InitTensor(inputsIndex, outputsIndex);
149 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr);
166 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr);
[all …]
|
| /ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/ |
| H A D | layernorm_builder.cpp | 106 const std::vector<uint32_t>& inputsIndex, in Build() argument
115 …OH_NN_ReturnCode returnCode = CheckIOIndex(inputsIndex, outputsIndex, allTensors, INPUT_NUM, OUTPU… in Build()
121 m_inputsIndex = inputsIndex; in Build()
145 auto inputShape = allTensors[inputsIndex[INPUT_X]]->GetDimensions(); in Build()
153 returnCode = ValidateGammaAndBetaShape(inputsIndex, m_beginNormAxis, allTensors); in Build()
179 …NN_ReturnCode LayerNormBuilder::ValidateGammaAndBetaShape(const std::vector<uint32_t>& inputsIndex, in ValidateGammaAndBetaShape() argument
182 auto inputShape = allTensors[inputsIndex[INPUT_X]]->GetDimensions(); in ValidateGammaAndBetaShape()
183 auto gammaShape = allTensors[inputsIndex[INPUT_GAMMA]]->GetDimensions(); in ValidateGammaAndBetaShape()
184 auto betaShape = allTensors[inputsIndex[INPUT_BETA]]->GetDimensions(); in ValidateGammaAndBetaShape()
|
| H A D | conv2d_builder.cpp | 43 OH_NN_ReturnCode Conv2DBuilder::SetInputAndOutput(const std::vector<uint32_t>& inputsIndex, in SetInputAndOutput() argument
47 …OH_NN_ReturnCode returnCode = CheckIOIndex(inputsIndex, outputsIndex, allTensors, INPUT_NUM, OUTPU… in SetInputAndOutput()
53 m_inputsIndex = inputsIndex; in SetInputAndOutput()
59 OH_NN_ReturnCode Conv2DBuilder::SetChannel(const std::vector<uint32_t>& inputsIndex, in SetChannel() argument
63 auto weightShape = allTensors[inputsIndex[CONV2D_INPUT_WEIGHT]]->GetDimensions(); in SetChannel()
75 void Conv2DBuilder::SetKernelSize(const std::vector<uint32_t>& inputsIndex, in SetKernelSize() argument
79 auto weightShape = allTensors[inputsIndex[CONV2D_INPUT_WEIGHT]]->GetDimensions(); in SetKernelSize()
227 const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex, in Build() argument
235 OH_NN_ReturnCode returnCode = SetInputAndOutput(inputsIndex, outputsIndex, allTensors); in Build()
246 returnCode = SetChannel(inputsIndex, allTensors); in Build()
[all …]
|
| H A D | concat_builder.cpp | 56 const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex, in Build() argument
64 if (inputsIndex.size() < MINIMUM_INTPUT) { in Build()
74 OH_NN_ReturnCode returnCode = SetInputsAndOutputs(inputsIndex, outputsIndex, allTensors); in Build()
109 OH_NN_ReturnCode ConcatBuilder::SetInputsAndOutputs(const std::vector<uint32_t>& inputsIndex, in SetInputsAndOutputs() argument
114 …bool isOverTensorSize = std::any_of(inputsIndex.begin(), inputsIndex.end(), [allTensorsSize](uint3… in SetInputsAndOutputs()
131 m_inputsIndex = inputsIndex; in SetInputsAndOutputs()
|
| H A D | split_builder.cpp | 29 OH_NN_ReturnCode SplitBuilder::SetInputAndOutput(const std::vector<uint32_t> &inputsIndex, in SetInputAndOutput() argument
32 auto inputSize = inputsIndex.size(); in SetInputAndOutput()
39 …bool isOverTensorSize = std::any_of(inputsIndex.begin(), inputsIndex.end(), [allTensorSize](uint32… in SetInputAndOutput()
55 m_inputsIndex = inputsIndex; in SetInputAndOutput()
125 const std::vector<uint32_t> &inputsIndex, in Build() argument
134 OH_NN_ReturnCode returnCode = SetInputAndOutput(inputsIndex, outputsIndex, allTensors); in Build()
|
| H A D | depthwise_conv2d_native_builder.cpp | 87 …NN_ReturnCode DepthwiseConv2DNativeBuilder::SetKernelSize(const std::vector<uint32_t>& inputsIndex, in SetKernelSize() argument
91 auto weightShape = allTensors[inputsIndex[INPUT_WEIGHT]]->GetDimensions(); in SetKernelSize()
185 const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex, in SetInputAndOutput() argument
188 …OH_NN_ReturnCode returnCode = CheckIOIndex(inputsIndex, outputsIndex, allTensors, INPUT_NUM, OUTPU… in SetInputAndOutput()
194 m_inputsIndex = inputsIndex; in SetInputAndOutput()
201 const std::vector<uint32_t>& inputsIndex, const std::vector<uint32_t>& outputsIndex, in Build() argument
209 OH_NN_ReturnCode ret = SetInputAndOutput(inputsIndex, outputsIndex, allTensors); in Build()
220 auto inputShape = allTensors[inputsIndex[INPUT_X]]->GetDimensions(); in Build()
227 ret = SetKernelSize(inputsIndex, allTensors); in Build()
|
| H A D | conv2d_transpose_builder.cpp | 43 OH_NN_ReturnCode Conv2DTransposeBuilder::SetInput(const std::vector<uint32_t>& inputsIndex, in SetInput() argument
47 …OH_NN_ReturnCode returnCode = CheckIOIndex(inputsIndex, outputsIndex, allTensors, INPUT_NUM, OUTPU… in SetInput()
53 m_inputsIndex = inputsIndex; in SetInput()
57 auto weightShape = allTensors[inputsIndex[INPUT_WEIGHT]]->GetDimensions(); in SetInput()
69 void Conv2DTransposeBuilder::SetKernelSize(const std::vector<uint32_t>& inputsIndex, in SetKernelSize() argument
72 auto weightShape = allTensors[inputsIndex[INPUT_WEIGHT]]->GetDimensions(); in SetKernelSize()
241 const std::vector<uint32_t>& inputsIndex, in Build() argument
250 OH_NN_ReturnCode returnCode = SetInput(inputsIndex, outputsIndex, allTensors); in Build()
261 SetKernelSize(inputsIndex, allTensors); in Build()
|
| H A D | unstack_builder.cpp | 54 const std::vector<uint32_t>& inputsIndex, in Build() argument
63 if (inputsIndex.size() != INPUT_NUM) { in Build()
74 …bool isOverTensorSize = std::any_of(inputsIndex.begin(), inputsIndex.end(), [allTensorsSize](uint3… in Build()
90 m_inputsIndex = inputsIndex; in Build()
|
| H A D | fullconnection_builder.cpp | 33 …_ReturnCode FullConnectionBuilder::SetFullConnectionInput(const std::vector<uint32_t>& inputsIndex, in SetFullConnectionInput() argument
43 …bool isOverTensorSize = std::any_of(inputsIndex.begin(), inputsIndex.end(), [allTensorsSize](uint3… in SetFullConnectionInput()
51 m_inputsIndex = inputsIndex; in SetFullConnectionInput()
171 const std::vector<uint32_t>& inputsIndex, in Build() argument
180 OH_NN_ReturnCode returnCode = SetFullConnectionInput(inputsIndex, outputsIndex, allTensors); in Build()
|
| H A D | cast_builder.cpp | 35 const std::vector<uint32_t>& inputsIndex, in Build() argument
43 auto ret = CheckIOIndex(inputsIndex, outputsIndex, allTensors, INPUT_NUM, OUTPUT_NUM); in Build()
48 m_inputsIndex = inputsIndex; in Build()
57 auto castType = allTensors[inputsIndex[INPUT_TYPE]]->GetBuffer(); in Build()
|
| /ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ |
| H A D | ops_builder.cpp | 28 void OpsBuilder::GetInputIndex(std::vector<uint32_t>& inputsIndex, in GetInputIndex() argument
32 std::transform(m_inputsIndex.begin(), m_inputsIndex.end(), std::back_inserter(inputsIndex), in GetInputIndex()
58 OH_NN_ReturnCode OpsBuilder::CheckIOIndex(const std::vector<uint32_t>& inputsIndex, in CheckIOIndex() argument
64 size_t inputsIndexSize = inputsIndex.size(); in CheckIOIndex()
76 …bool isOverTensorSize = std::any_of(inputsIndex.begin(), inputsIndex.end(), [allTensorsSize](uint3… in CheckIOIndex()
|