Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dsqrt_builder_test.cpp42 std::vector<int32_t> inputDim = {3}; in InitTensor() local
65 std::vector<int32_t> inputDim = {3}; variable
83 std::vector<int32_t> inputDim = {3}; variable
102 std::vector<int32_t> inputDim = {3}; variable
120 std::vector<int32_t> inputDim = {3}; variable
152 std::vector<int32_t> inputDim = {3}; variable
H A Dsquared_difference_builder_test.cpp42 std::vector<int32_t> inputDim = {3}; in InitTensor() local
65 std::vector<int32_t> inputDim = {3}; variable
83 std::vector<int32_t> inputDim = {3}; variable
102 std::vector<int32_t> inputDim = {3}; variable
120 std::vector<int32_t> inputDim = {3}; variable
152 std::vector<int32_t> inputDim = {3}; variable
H A Dtile_builder_test.cpp47 std::vector<int32_t> inputDim = {2, 2}; in InitTensor() local
147 std::vector<int32_t> inputDim = {2, 2}; variable
162 std::vector<int32_t> inputDim = {2, 2}; variable
187 std::vector<int32_t> inputDim = {2, 2}; variable
205 std::vector<int32_t> inputDim = {2, 2}; variable
H A Dslice_builder_test.cpp47 std::vector<int32_t> inputDim = {3, 2, 3}; in InitTensor() local
173 std::vector<int32_t> inputDim = { 3, 2, 3 }; variable
199 std::vector<int32_t> inputDim = { 3, 2, 3 }; variable
219 std::vector<int32_t> inputDim = { 3, 2, 3 }; variable
H A Dtanh_builder_test.cpp41 std::vector<int32_t> inputDim = {1, 5, 1, 1}; in InitTensor() local
137 std::vector<int32_t> inputDim = {1, 5, 1, 1}; variable
H A Dstrided_slice_builder_test.cpp113 std::vector<int32_t> inputDim = {3, 2, 3}; in InitTensor() local
176 std::vector<int32_t> inputDim = {3, 2, 3}; variable
198 std::vector<int32_t> inputDim = {3, 2, 3}; variable
235 std::vector<int32_t> inputDim = {3, 2, 3}; variable
257 std::vector<int32_t> inputDim = {3, 2, 3}; variable
H A Dsoftmax_builder_test.cpp56 std::vector<int32_t> inputDim = {1, 5, 1, 1}; in InitTensor() local
179 std::vector<int32_t> inputDim = {1, 5, 1, 1}; variable
H A Dtranspose_builder_test.cpp41 std::vector<int32_t> inputDim = {2, 3}; in InitTensor() local
H A Dsplit_builder_test.cpp50 std::vector<int32_t> inputDim = {2, 4}; in InitTensor() local
380 std::vector<int32_t> inputDim = {2, 4}; variable
H A Dspacetobatchnd_builder_test.cpp50 std::vector<int32_t> inputDim = {3, 2, 3}; in InitTensor() local
61 std::vector<int32_t> inputDim = {1, 2, 2, 1}; in InitTensor() local
H A Dsqueeze_builder_test.cpp56 std::vector<int32_t> inputDim = {3, 2, 1}; in InitTensor() local
H A Dstack_builder_test.cpp45 std::vector<int32_t> inputDim = {2}; in InitTensor() local
H A Dunsqueeze_builder_test.cpp55 std::vector<int32_t> inputDim = {1, 5, 1}; in InitTensor() local
H A Dsub_builder_test.cpp55 std::vector<int32_t> inputDim = {3}; in InitTensor() local
H A Dtopk_builder_test.cpp45 std::vector<int32_t> inputDim = {9}; in InitTensor() local
/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