Home
last modified time | relevance | path

Searched defs:OutputDim (Results 1 – 15 of 15) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dsqrt_builder_test.cpp43 std::vector<int32_t> OutputDim = {3}; in InitTensor() local
66 std::vector<int32_t> OutputDim = {3}; variable
84 std::vector<int32_t> OutputDim = {3}; variable
103 std::vector<int32_t> OutputDim = {3}; variable
121 std::vector<int32_t> OutputDim = {3}; variable
153 std::vector<int32_t> OutputDim = {3}; variable
H A Dsquared_difference_builder_test.cpp43 std::vector<int32_t> OutputDim = {3}; in InitTensor() local
66 std::vector<int32_t> OutputDim = {3}; variable
84 std::vector<int32_t> OutputDim = {3}; variable
103 std::vector<int32_t> OutputDim = {3}; variable
121 std::vector<int32_t> OutputDim = {3}; variable
H A Dslice_builder_test.cpp48 std::vector<int32_t> OutputDim = {1, 1, 3}; in InitTensor() local
174 std::vector<int32_t> OutputDim = { 1, 1, 3 }; variable
200 std::vector<int32_t> OutputDim = { 1, 1, 3 }; variable
220 std::vector<int32_t> OutputDim = { 1, 1, 3 }; variable
H A Dtile_builder_test.cpp48 std::vector<int32_t> OutputDim = {4, 4}; in InitTensor() local
163 std::vector<int32_t> OutputDim = {4, 4}; variable
188 std::vector<int32_t> OutputDim = {4, 4}; variable
206 std::vector<int32_t> OutputDim = {4, 4}; variable
H A Dstrided_slice_builder_test.cpp114 std::vector<int32_t> OutputDim = {1, 2, 2}; in InitTensor() local
177 std::vector<int32_t> OutputDim = {1, 2, 2}; variable
199 std::vector<int32_t> OutputDim = {1, 2, 2}; variable
236 std::vector<int32_t> OutputDim = {1, 2, 2}; variable
258 std::vector<int32_t> OutputDim = {1, 2, 2}; variable
H A Dsoftmax_builder_test.cpp57 std::vector<int32_t> OutputDim = {1, 5, 1, 1}; in InitTensor() local
180 std::vector<int32_t> OutputDim = {1, 5, 1, 1}; variable
H A Dtranspose_builder_test.cpp42 std::vector<int32_t> OutputDim = {3, 2}; in InitTensor() local
H A Dtanh_builder_test.cpp42 std::vector<int32_t> OutputDim = {1, 5, 1, 1}; in InitTensor() local
H A Dspacetobatchnd_builder_test.cpp51 std::vector<int32_t> OutputDim = {1, 1, 3}; in InitTensor() local
62 std::vector<int32_t> OutputDim = {4, 1, 1, 1}; in InitTensor() local
H A Dsqueeze_builder_test.cpp57 std::vector<int32_t> OutputDim = {3, 2}; in InitTensor() local
H A Dstack_builder_test.cpp46 std::vector<int32_t> OutputDim = {2, 2}; in InitTensor() local
H A Dunsqueeze_builder_test.cpp56 std::vector<int32_t> OutputDim = {1, 1, 5, 1}; in InitTensor() local
H A Dsub_builder_test.cpp56 std::vector<int32_t> OutputDim = {3}; in InitTensor() local
H A Dtopk_builder_test.cpp46 std::vector<int32_t> OutputDim = {3}; in InitTensor() local
H A Dsplit_builder_test.cpp51 std::vector<int32_t> OutputDim = {1, 4, 0, 0}; in InitTensor() local