Searched defs:input_shape1 (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore-src/source/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
112 std::vector<int> input_shape1 = {2, 4}; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/ |
| D | where_infer.c | 23 … const int *input_shape1, int *ndim, int *in_shape0, int *in_shape1, int *out_shape, in WhereBroadCastInferShape()
71 const int *input_shape1 = input1->shape_; in WhereInferShape() local
|
| 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 | 68 const int *input_shape1, int *ndim, int *in_shape0, int *in_shape1) { in MakeUpInputShapes()
121 const int *input_shape1, int *ndim, int *out_shape, bool *has_broad_cast) { in BroadCastToShape()
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | crop_and_resize_grad_image.cc | 60 …pes(const std::string &prim_name, const ShapeVector &input_shape0, const ShapeVector &input_shape1, in CheckShapes()
132 …auto input_shape1 = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[ImagekBoxes]->GetSh… in CropAndResizeGradImageInferShape() local
|
| D | crop_and_resize_grad_boxes.cc | 78 …auto input_shape1 = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kImages]->GetShape(… in CropAndResizeGradBoxesInferShape() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/dataset/ |
| D | pad_end_op_test.cc | 39 TensorShape input_shape1({2, 2}); in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/ |
| D | sparse_matrix_mat_mul_cpu_kernel.cc | 97 auto input_shape1 = inputs.at(kIndex0)->GetShapeVector(); in Resize() local
|