Home
last modified time | relevance | path

Searched refs:inputDim (Results 1 – 16 of 16) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dsquared_difference_builder_test.cpp42 std::vector<int32_t> inputDim = {3}; in InitTensor() local
45 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
65 std::vector<int32_t> inputDim = {3}; variable
68 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr);
83 std::vector<int32_t> inputDim = {3}; variable
86 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr);
102 std::vector<int32_t> inputDim = {3}; variable
105 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr);
120 std::vector<int32_t> inputDim = {3}; variable
123 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr);
[all …]
H A Dsqrt_builder_test.cpp42 std::vector<int32_t> inputDim = {3}; in InitTensor() local
45 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
65 std::vector<int32_t> inputDim = {3}; variable
68 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr);
83 std::vector<int32_t> inputDim = {3}; variable
86 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr);
102 std::vector<int32_t> inputDim = {3}; variable
105 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr);
120 std::vector<int32_t> inputDim = {3}; variable
123 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr);
[all …]
H A Dtile_builder_test.cpp47 std::vector<int32_t> inputDim = {2, 2}; in InitTensor() local
51 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
147 std::vector<int32_t> inputDim = {2, 2}; variable
149 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr);
162 std::vector<int32_t> inputDim = {2, 2}; variable
166 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr);
187 std::vector<int32_t> inputDim = {2, 2}; variable
190 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr);
205 std::vector<int32_t> inputDim = {2, 2}; variable
209 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr);
H A Dslice_builder_test.cpp47 std::vector<int32_t> inputDim = {3, 2, 3}; in InitTensor() local
51 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
173 std::vector<int32_t> inputDim = { 3, 2, 3 }; variable
178 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr);
199 std::vector<int32_t> inputDim = { 3, 2, 3 }; variable
204 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr);
219 std::vector<int32_t> inputDim = { 3, 2, 3 }; variable
224 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr);
H A Dtanh_builder_test.cpp41 std::vector<int32_t> inputDim = {1, 5, 1, 1}; in InitTensor() local
44 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
137 std::vector<int32_t> inputDim = {1, 5, 1, 1}; variable
139 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr);
H A Dstrided_slice_builder_test.cpp113 std::vector<int32_t> inputDim = {3, 2, 3}; in InitTensor() local
117 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
176 std::vector<int32_t> inputDim = {3, 2, 3}; variable
180 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr);
198 std::vector<int32_t> inputDim = {3, 2, 3}; variable
202 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr);
235 std::vector<int32_t> inputDim = {3, 2, 3}; variable
239 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr);
257 std::vector<int32_t> inputDim = {3, 2, 3}; variable
262 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr);
H A Dsoftmax_builder_test.cpp56 std::vector<int32_t> inputDim = {1, 5, 1, 1}; in InitTensor() local
60 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
179 std::vector<int32_t> inputDim = {1, 5, 1, 1}; variable
183 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr);
H A Dtranspose_builder_test.cpp41 std::vector<int32_t> inputDim = {2, 3}; in InitTensor() local
44 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
H A Dsplit_builder_test.cpp50 std::vector<int32_t> inputDim = {2, 4}; in InitTensor() local
54 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
380 std::vector<int32_t> inputDim = {2, 4}; variable
383 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr);
H A Dspacetobatchnd_builder_test.cpp50 std::vector<int32_t> inputDim = {3, 2, 3}; in InitTensor() local
54 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
61 std::vector<int32_t> inputDim = {1, 2, 2, 1}; in InitTensor() local
67 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
H A Dsqueeze_builder_test.cpp56 std::vector<int32_t> inputDim = {3, 2, 1}; in InitTensor() local
60 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
H A Dstack_builder_test.cpp45 std::vector<int32_t> inputDim = {2}; in InitTensor() local
49 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
H A Dunsqueeze_builder_test.cpp55 std::vector<int32_t> inputDim = {1, 5, 1}; in InitTensor() local
59 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
H A Dsub_builder_test.cpp55 std::vector<int32_t> inputDim = {3}; in InitTensor() local
59 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
H A Dtopk_builder_test.cpp45 std::vector<int32_t> inputDim = {9}; in InitTensor() local
48 SaveInputTensor(inputsIndex, OH_NN_FLOAT32, inputDim, nullptr); in InitTensor()
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dneural_network_runtime.cpp328 std::vector<int32_t> inputDim; in ParseInputDimsFromExtensions() local
331 inputDim.clear(); in ParseInputDimsFromExtensions()
349 inputDim.emplace_back(dimsValue[dataIndex]); in ParseInputDimsFromExtensions()
355 extensionConfig.inputDims.emplace_back(inputDim); in ParseInputDimsFromExtensions()