Home
last modified time | relevance | path

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

/foundation/ai/neural_network_runtime/test/unittest/ops/
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
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
Dslice_builder_test.cpp45 std::vector<int32_t> OutputDim = {1, 1, 3}; in InitTensor() local
170 std::vector<int32_t> OutputDim = {1, 1, 3}; variable
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
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
Dtranspose_builder_test.cpp42 std::vector<int32_t> OutputDim = {3, 2}; in InitTensor() local
Dtile_builder_test.cpp43 std::vector<int32_t> OutputDim = {4, 4}; in InitTensor() local
Dtanh_builder_test.cpp42 std::vector<int32_t> OutputDim = {1, 5, 1, 1}; in InitTensor() local
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
Dsqueeze_builder_test.cpp57 std::vector<int32_t> OutputDim = {3, 2}; in InitTensor() local
Dtopk_builder_test.cpp45 std::vector<int32_t> OutputDim = {3}; in InitTensor() local
Dstack_builder_test.cpp46 std::vector<int32_t> OutputDim = {2, 2}; in InitTensor() local
Dunsqueeze_builder_test.cpp56 std::vector<int32_t> OutputDim = {1, 1, 5, 1}; in InitTensor() local
Dsub_builder_test.cpp56 std::vector<int32_t> OutputDim = {3}; in InitTensor() local
Dsplit_builder_test.cpp51 std::vector<int32_t> OutputDim = {1, 4, 0, 0}; in InitTensor() local