| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | arithmetic_tests.cc | 44 OpParameter *CreateParameter(schema::PrimitiveType type, const std::vector<int> &input0_shape, in CreateParameter() argument
48 …int input0_size = std::accumulate(input0_shape.begin(), input0_shape.end(), 1, std::multiplies<>()… in CreateParameter()
59 std::vector<int> input0_shape = {1, 2, 2, 3}; in TEST_F() local
67 auto *param = CreateParameter(schema::PrimitiveType_AddFusion, input0_shape, input1_shape); in TEST_F()
68 …TestMain({{input0_shape, input0_data, VAR}, {input1_shape, input1_data, CONST_TENSOR}}, {output_sh… in TEST_F()
74 std::vector<int> input0_shape = {1, 2, 2, 3}; in TEST_F() local
81 auto *param = CreateParameter(schema::PrimitiveType_MulFusion, input0_shape, input1_shape); in TEST_F()
82 …TestMain({{input0_shape, input0_data, VAR}, {input1_shape, input1_data, CONST_TENSOR}}, {output_sh… in TEST_F()
88 std::vector<int> input0_shape = {1, 2, 2, 3}; in TEST_F() local
96 …CreateParameter(schema::PrimitiveType_SubFusion, input0_shape, input1_shape, schema::ActivationTyp… in TEST_F()
[all …]
|
| D | concat_tests.cc | 33 std::vector<int> input0_shape = {1, 1, 1, 8}; in TEST_F() local
42 …TestMain({{input0_shape, input0_data, VAR}, {input1_shape, input1_data, VAR}}, {output_shape, outp… in TEST_F()
48 std::vector<int> input0_shape = {1}; in TEST_F() local
57 …TestMain({{input0_shape, input0_data, VAR}, {input1_shape, input1_data, VAR}}, {output_shape, outp… in TEST_F()
63 std::vector<int> input0_shape = {2, 2, 2, 8}; in TEST_F() local
85 …TestMain({{input0_shape, input0_data, VAR}, {input1_shape, input1_data, VAR}}, {output_shape, outp… in TEST_F()
91 std::vector<int> input0_shape = {2, 3, 2, 8}; in TEST_F() local
175 TestMain({{input0_shape, input0_data, VAR}, in TEST_F()
186 std::vector<int> input0_shape = {1, 1, 8}; in TEST_F() local
206 TestMain({{input0_shape, input0_data, VAR}, in TEST_F()
[all …]
|
| D | power_tests.cc | 39 std::vector<int> input0_shape = {1, 2, 8}; in TEST_F() local
51 …TestMain({{input0_shape, input0_data, VAR}, {input1_shape, input1_data, VAR}}, {output_shape, outp… in TEST_F()
57 std::vector<int> input0_shape = {2, 8}; in TEST_F() local
71 …TestMain({{input0_shape, input0_data, VAR}, {input1_shape, input1_data, VAR}}, {output_shape, outp… in TEST_F()
77 std::vector<int> input0_shape = {2, 7}; in TEST_F() local
90 …TestMain({{input0_shape, input0_data, VAR}, {input1_shape, input1_data, VAR}}, {output_shape, outp… in TEST_F()
96 std::vector<int> input0_shape = {1, 2, 8}; in TEST_F() local
105 TestMain({{input0_shape, input0_data, VAR}}, {output_shape, output_data}, param, fp16_enable, in TEST_F()
111 std::vector<int> input0_shape = {2, 8}; in TEST_F() local
124 …TestMain({{input0_shape, input0_data, VAR}, {input1_shape, input1_data, VAR}}, {output_shape, outp… in TEST_F()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | where.cc | 38 …auto input0_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->Buil… in WhereInfer() local
47 for (size_t j = 0; j < input0_shape.size(); j++) { in WhereInfer()
48 if (input0_shape[j] == input1_shape[j] && input0_shape[j] != input2_shape[j]) { in WhereInfer()
52 if (input0_shape[j] == input2_shape[j] && input0_shape[j] != input1_shape[j]) { in WhereInfer()
56 if (input1_shape[j] != input2_shape[j] && input0_shape[j] == input1_shape[j]) { in WhereInfer()
61 if (temp == input0_shape.size()) { in WhereInfer()
62 return std::make_shared<abstract::AbstractTensor>(input0_type, input0_shape); in WhereInfer()
65 input0_shape[axisout] = (int64_t)nummax; in WhereInfer()
66 return std::make_shared<abstract::AbstractTensor>(input0_type, input0_shape); in WhereInfer()
|
| D | tensor_list_from_tensor.cc | 31 …auto input0_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[k… in TensorListFromTensorInferShape() local
32 if (input0_shape.size() < 1) { in TensorListFromTensorInferShape()
33 MS_LOG(ERROR) << "input0_shape.size():" << input0_shape.size() << " must be greater than 0!"; in TensorListFromTensorInferShape()
35 int64_t dim0 = input0_shape[0]; in TensorListFromTensorInferShape()
|
| D | tensor_list_stack.cc | 55 …auto input0_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[k… in TensorListStackInfer() local
56 …int64_t num = std::accumulate(input0_shape.begin(), input0_shape.end(), 1LL, std::multiplies<int64… in TensorListStackInfer()
|
| /third_party/mindspore/mindspore/core/ops/fusion/ |
| D | full_connection.cc | 69 auto input0_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input0->BuildShape())[kShape]; in FullConnectionInfer() local
83 if (use_axis && (prim_axis < 1 || prim_axis > (int64_t)input0_shape.size())) { in FullConnectionInfer()
88 for (size_t t = LongToSize(prim_axis); t < input0_shape.size(); t++) { in FullConnectionInfer()
89 new_k *= input0_shape[t]; in FullConnectionInfer()
103 std::vector<int64_t> out_shape = {(int64_t)input0_shape.size()}; in FullConnectionInfer()
109 for (size_t i = 0; i < input0_shape.size(); i++) { in FullConnectionInfer()
110 total *= input0_shape[i]; in FullConnectionInfer()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | stack_base.cc | 60 auto input0_shape = in_tensors_.front()->shape(); in ReSize() local
61 axis_ = param->axis_ < 0 ? param->axis_ + input0_shape.size() + 1 : param->axis_; in ReSize()
67 CHECK_LESS_RETURN(input0_shape.size(), static_cast<size_t>(axis_)); in ReSize()
68 copy_size_ = GetCopyNum(input0_shape, axis_, input0_shape.size()) * data_type_size_; in ReSize()
69 outer_size_ = GetOuterSize(input0_shape, axis_); in ReSize()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | concat_infer.c | 46 const int *input0_shape = inputs[0]->shape_; in ConcatInferShape() local
59 …ShapeSet(input0_shape_without_axis, &input0_shape_without_axis_size, input0_shape, input0_shape_si… in ConcatInferShape()
64 int output_axis_dim = input0_shape[axis]; in ConcatInferShape()
94 output_shape[i] = input0_shape[i]; in ConcatInferShape()
|
| /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
64 TensorInfo input0(input0_layout, input0_shape, input0_slice_shape), in TEST_F()
119 Shape input0_shape{200, 300}, output0_shape{200, 300}; in TEST_F() local
121 TensorInfo input0_info(input0_layout, input0_shape, input0_slice_shape), in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | operator_costmodel.cc | 89 Shape input0_shape = input0.shape(); in GetForwardCommCost() local
91 if (input0_shape[input0_shape.size() - 1] == input0_slice_shape[input0_slice_shape.size() - 1]) { in GetForwardCommCost()
135 Shape input0_shape = inputs[0].shape(); in GetForwardComputationCost() local
136 if (input0_shape[input0_shape.size() - 1] != input0_slice_shape[input0_slice_shape.size() - 1]) { in GetForwardComputationCost()
1099 Shape input0_shape = input0.shape(); in GetForwardCommCost() local
1101 if (cross_batch_ && IsDataParallel(input0_shape, input0_slice_shape, stage_id)) { in GetForwardCommCost()
1105 …auto pos = std::find_if(dim_list.begin(), dim_list.end(), [input0_shape, input0_slice_shape](int64… in GetForwardCommCost()
1106 return input0_shape[LongToSize(index)] != input0_slice_shape[LongToSize(index)]; in GetForwardCommCost()
1145 Shape input0_shape = input0.shape(); in GetForwardComputationCost() local
1146 if (!cross_batch_ || !IsDataParallel(input0_shape, input0_slice_shape, stage_id)) { in GetForwardComputationCost()
[all …]
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/ |
| D | batchnorm.cc | 77 auto input0_shape = in_tensors_.at(0)->shape(); in SetConstArgs() local
78 cl_int4 input_shape_ = {input0_shape.at(0), input0_shape.at(1), input0_shape.at(2), in SetConstArgs()
79 UP_DIV(input0_shape.at(3), C4NUM)}; in SetConstArgs()
88 if (ocl_runtime_->SetKernelArg(kernel_, arg_cn++, input0_shape.at(3)) != CL_SUCCESS) { in SetConstArgs()
|
| /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
412 if (SwapLastTwoElements(&input0_shape) == FAILED) { in GenerateStrategies()
426 size_t input1_shape_size = input1_shape.size(), input0_shape_size = input0_shape.size(); in GenerateStrategies()
430 if (input0_shape.size() >= input1_shape.size()) { in GenerateStrategies()
431 combined_shape = input0_shape; in GenerateStrategies()
435 combined_shape.push_back(input0_shape[input0_shape.size() - 2]); in GenerateStrategies()
|
| D | operator_info.cc | 1079 Shapes input0_shape = {inputs_shape[0]}; in GenerateStrategiesForTwoEqualInputs() local
1081 …if (GenerateStrategiesForIndependentInputs(stage_id, input0_shape, input0_splittable, sp_vector) !… in GenerateStrategiesForTwoEqualInputs()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _inner_ops.py | 591 def infer_shape(self, input0_shape, input1_shape, input2_shape): argument
592 outshape0 = input0_shape
|