Searched defs:input_shape0 (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | sparse_to_dense_tests.cc | 32 std::vector<int> input_shape0 = {6, 3}; in TEST_F() local
55 std::vector<int> input_shape0 = {6, 2}; in TEST_F() local
78 std::vector<int> input_shape0 = {6, 2}; in TEST_F() local
101 std::vector<int> input_shape0 = {6, 1}; in TEST_F() local
123 std::vector<int> input_shape0 = {7, 1}; in TEST_F() local
145 std::vector<int> input_shape0 = {6}; in TEST_F() local
167 std::vector<int> input_shape0 = {6}; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/ |
| D | where_infer.c | 22 …adCastInferShape(const int input_shape0_size, const int input_shape1_size, const int *input_shape0, in WhereBroadCastInferShape()
70 const int *input_shape0 = input0->shape_; in WhereInferShape() local
|
| D | arithmetic_infer.c | 38 …adCastInferShape(const int input_shape0_size, const int input_shape1_size, const int *input_shape0, in BroadCastInferShape()
67 const int *input_shape0 = input0->shape_; in ArithmeticInferShape() local
|
| D | broadcast_to_infer.c | 67 …akeUpInputShapes(const int input_shape0_size, const int input_shape1_size, const int *input_shape0, in MakeUpInputShapes()
120 …BroadCastToShape(const int input_shape0_size, const int input_shape1_size, const int *input_shape0, in BroadCastToShape()
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | crop_and_resize_grad_image.cc | 60 void CheckShapes(const std::string &prim_name, const ShapeVector &input_shape0, const ShapeVector &… in CheckShapes()
130 …auto input_shape0 = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[ImagekGrads]->GetSh… in CropAndResizeGradImageInferShape() local
|
| D | crop_and_resize_grad_boxes.cc | 76 …auto input_shape0 = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kGrads]->GetShape()… in CropAndResizeGradBoxesInferShape() local
|