| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | arithmetic_tests.cc | 45 const std::vector<int> &input1_shape, in CreateParameter()
60 std::vector<int> input1_shape = {1, 2, 2, 3}; in TEST_F() local
75 std::vector<int> input1_shape = {1}; in TEST_F() local
89 std::vector<int> input1_shape = {3}; in TEST_F() local
104 std::vector<int> input1_shape = {1, 2, 2, 3}; in TEST_F() local
118 std::vector<int> input1_shape = {2, 2, 2, 3}; in TEST_F() local
132 std::vector<int> input1_shape = {1, 1, 1, 4}; in TEST_F() local
147 std::vector<int> input1_shape = {1, 1, 3, 4}; in TEST_F() local
162 std::vector<int> input1_shape = {1, 2, 2, 3}; in TEST_F() local
|
| D | concat_tests.cc | 34 std::vector<int> input1_shape = {1, 1, 1, 8}; in TEST_F() local
49 std::vector<int> input1_shape = {1}; in TEST_F() local
64 std::vector<int> input1_shape = {2, 2, 2, 8}; in TEST_F() local
92 std::vector<int> input1_shape = {2, 3, 2, 8}; in TEST_F() local
187 std::vector<int> input1_shape = {1, 1, 8}; in TEST_F() local
218 std::vector<int> input1_shape = {2, 2, 2, 9}; in TEST_F() local
249 std::vector<int> input1_shape = {1, 7}; in TEST_F() local
267 std::vector<int> input1_shape = {1, 1, 1, 7}; in TEST_F() local
290 std::vector<int> input1_shape = {1, 1, 1, 7}; in TEST_F() local
317 std::vector<int> input1_shape = {1, 1, 1, 7}; in TEST_F() local
|
| D | power_tests.cc | 40 std::vector<int> input1_shape = {1, 2, 8}; in TEST_F() local
58 std::vector<int> input1_shape = {2, 8}; in TEST_F() local
78 std::vector<int> input1_shape = {2, 7}; in TEST_F() local
112 std::vector<int> input1_shape = {2, 8}; in TEST_F() local
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | tensor_list_stack.cc | 63 …auto input1_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->BuildShape())[k… in TensorListStackInfer() local
|
| D | where.cc | 40 …auto input1_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex1]->Buil… in WhereInfer() local
|
| /third_party/mindspore/mindspore/core/ops/fusion/ |
| D | full_connection.cc | 70 auto input1_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input1->BuildShape())[kShape]; in FullConnectionInfer() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | matmul_gpu_kernel.h | 112 auto input1_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0); in Init() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | scale_int8.cc | 130 auto input1_shape = in_tensors_.at(1)->shape(); in InitParameter() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | operator_costmodel.cc | 112 Shape input1_shape = input1.shape(); in GetBackwardCommCost() local
157 Shape input1_shape = input1.shape(); in GetBackwardComputationCost() local
213 Shape input1_shape = input1.shape(); in GetBackwardCommCost() local
279 Shape input1_shape = input1.shape(); in GetBackwardCommCost() local
516 Shape input1_shape = input1.shape(); in GetBackwardCommCost() local
554 Shape input1_shape = input1.shape(); in GetBackwardComputationCost() local
690 Shape input1_shape = input1.shape(); in GetBackwardCommCost() local
|
| /third_party/mindspore/tests/ut/cpp/parallel/auto_parallel/ |
| D | operator_costmodel_test.cc | 62 Shape input0_shape{200, 300}, input1_shape{300, 500}, output0_shape{200, 500}; in TEST_F() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | matmul_info.cc | 410 Shape input0_shape = inputs_shape_[0], input1_shape = inputs_shape_[1]; in GenerateStrategies() local
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _inner_ops.py | 591 def infer_shape(self, input0_shape, input1_shape, input2_shape): argument
|
| /third_party/mindspore/tests/ut/cpp/parallel/ops_info/ |
| D | matmul_info_test.cc | 594 Shape input1_shape = input1_info.shape(); in TEST_F() local
|