Searched defs:input_shape1 (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | sparse_to_dense_tests.cc | 33 std::vector<int> input_shape1 = {3}; in TEST_F() local
56 std::vector<int> input_shape1 = {2}; in TEST_F() local
79 std::vector<int> input_shape1 = {2}; in TEST_F() local
102 std::vector<int> input_shape1 = {1}; in TEST_F() local
124 std::vector<int> input_shape1 = {1}; in TEST_F() local
146 std::vector<int> input_shape1 = {1}; in TEST_F() local
168 std::vector<int> input_shape1 = {1}; in TEST_F() local
|
| D | prelu_tests.cc | 33 std::vector<int> input_shape1 = {1, 4, 5, 6}; in TEST_F() local
52 std::vector<int> input_shape1 = {1, 4, 5, 6}; in TEST_F() local
|
| D | fill_tests.cc | 42 std::vector<int> input_shape1 = {2}; in TEST_F() local
113 std::vector<int> input_shape1 = {2, 4}; in TEST_F() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | arithmetic_infer.c | 39 … const int *input_shape1, int *ndim, int *in_shape0, int *in_shape1, int *out_shape, in BroadCastInferShape()
69 const int *input_shape1 = input1->shape_; in ArithmeticInferShape() local
|
| D | broadcast_to_infer.c | 67 const int *input_shape1, int *ndim, int *in_shape0, int *in_shape1) { in MakeUpInputShapes()
120 const int *input_shape1, int *ndim, int *out_shape, bool *has_broad_cast) { in BroadCastToShape()
|
| /third_party/mindspore/tests/ut/cpp/dataset/ |
| D | pad_end_op_test.cc | 39 TensorShape input_shape1({2, 2}); in TEST_F() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | squared_difference_kernel.h | 59 auto input_shape1 = AnfAlgo::GetInputRealDeviceShapeIfExist(kernel_node, 0); in Init() local
|