Home
last modified time | relevance | path

Searched defs:input_shape2 (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Dsparse_to_dense_tests.cc34 std::vector<int> input_shape2 = {6}; in TEST_F() local
57 std::vector<int> input_shape2 = {1}; in TEST_F() local
80 std::vector<int> input_shape2 = {6}; in TEST_F() local
103 std::vector<int> input_shape2 = {6}; in TEST_F() local
125 std::vector<int> input_shape2 = {1}; in TEST_F() local
147 std::vector<int> input_shape2 = {1}; in TEST_F() local
169 std::vector<int> input_shape2 = {6}; in TEST_F() local
Dprelu_tests.cc34 std::vector<int> input_shape2 = {1}; in TEST_F() local
53 std::vector<int> input_shape2 = {1, 1, 1, 6}; in TEST_F() local
/third_party/mindspore/tests/ut/cpp/dataset/
Dpad_end_op_test.cc92 TensorShape input_shape2({2}); in TEST_F() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/
Dsquared_difference_kernel.h60 auto input_shape2 = AnfAlgo::GetInputRealDeviceShapeIfExist(kernel_node, 1); in Init() local