| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | elementary_function_info.h | 33 …ExpInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const Pr… in ExpInfo() argument
34 : ActivationOther(name, inputs_shape, outputs_shape, attrs, std::make_shared<ExpCost>()) {} in ExpInfo()
40 …LogInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const Pr… in LogInfo() argument
41 : ActivationOther(name, inputs_shape, outputs_shape, attrs, std::make_shared<LogCost>()) {} in LogInfo()
47 …CosInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const Pr… in CosInfo() argument
48 : ActivationOther(name, inputs_shape, outputs_shape, attrs, std::make_shared<CosCost>()) {} in CosInfo()
54 ACosInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ACosInfo() argument
56 : ActivationOther(name, inputs_shape, outputs_shape, attrs, std::make_shared<ACosCost>()) {} in ACosInfo()
62 LogicalNotInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in LogicalNotInfo() argument
64 … : ActivationOther(name, inputs_shape, outputs_shape, attrs, std::make_shared<LogicalNotCost>()) {} in LogicalNotInfo()
[all …]
|
| D | arithmetic_info.h | 34 …ArithmeticBase(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs… in ArithmeticBase() argument
36 : OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, cost) {} in ArithmeticBase()
55 …SubInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const Pr… in SubInfo() argument
56 : ArithmeticBase(name, inputs_shape, outputs_shape, attrs, std::make_shared<SubCost>()) {} in SubInfo()
62 …AddInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const Pr… in AddInfo() argument
63 … : ArithmeticBase(name, inputs_shape, outputs_shape, attrs, std::make_shared<TensorAddCost>()) {} in AddInfo()
69 …MulInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const Pr… in MulInfo() argument
70 : ArithmeticBase(name, inputs_shape, outputs_shape, attrs, std::make_shared<MulCost>()) {} in MulInfo()
76 …DivInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const Pr… in DivInfo() argument
77 : ArithmeticBase(name, inputs_shape, outputs_shape, attrs, std::make_shared<DivCost>()) {} in DivInfo()
[all …]
|
| D | activation_info.h | 34 …ActivationBase(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs… in ActivationBase() argument
36 : OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, cost) {} in ActivationBase()
51 Activation(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in Activation() argument
53 : ActivationBase(name, inputs_shape, outputs_shape, attrs, cost) {} in Activation()
64 ActivationInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ActivationInfo() argument
66 … : Activation(name, inputs_shape, outputs_shape, attrs, std::make_shared<ActivationInfoCost>()) {} in ActivationInfo()
75 ActivationOther(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ActivationOther() argument
77 : Activation(name, inputs_shape, outputs_shape, attrs, cost) {} in ActivationOther()
86 GeLUInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in GeLUInfo() argument
88 : ActivationOther(name, inputs_shape, outputs_shape, attrs, std::make_shared<GeLUCost>()) {} in GeLUInfo()
[all …]
|
| D | comparison_function_info.h | 33 EqualInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in EqualInfo() argument
35 : ArithmeticBase(name, inputs_shape, outputs_shape, attrs, std::make_shared<EqualCost>()) {} in EqualInfo()
41 …ApproximateEqualInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_sh… in ApproximateEqualInfo() argument
43 …: ArithmeticBase(name, inputs_shape, outputs_shape, attrs, std::make_shared<ApproximateEqualCost>(… in ApproximateEqualInfo()
49 NotEqualInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in NotEqualInfo() argument
51 … : ArithmeticBase(name, inputs_shape, outputs_shape, attrs, std::make_shared<NotEqualCost>()) {} in NotEqualInfo()
57 MaximumInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in MaximumInfo() argument
59 : ArithmeticBase(name, inputs_shape, outputs_shape, attrs, std::make_shared<MaximumCost>()) {} in MaximumInfo()
65 MinimumInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in MinimumInfo() argument
67 : ArithmeticBase(name, inputs_shape, outputs_shape, attrs, std::make_shared<MinimumCost>()) {} in MinimumInfo()
[all …]
|
| D | reduce_method_info.h | 35 ReduceMethod(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ReduceMethod() argument
37 : OperatorInfo(name, inputs_shape, outputs_shape, attrs, cost) {} in ReduceMethod()
63 ReduceMaxInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ReduceMaxInfo() argument
65 : ReduceMethod(name, inputs_shape, outputs_shape, attrs, std::make_shared<ReduceMaxCost>()) { in ReduceMaxInfo()
74 …ArgMaxWithValueInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_sha… in ArgMaxWithValueInfo() argument
76 …: ReduceMethod(name, inputs_shape, outputs_shape, attrs, std::make_shared<ArgMaxWithValueCost>()) { in ArgMaxWithValueInfo()
95 …ArgMinWithValueInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_sha… in ArgMinWithValueInfo() argument
97 : ArgMaxWithValueInfo(name, inputs_shape, outputs_shape, attrs) { in ArgMinWithValueInfo()
106 ReduceMeanInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ReduceMeanInfo() argument
108 … : ReduceMethod(name, inputs_shape, outputs_shape, attrs, std::make_shared<ReduceMeanCost>()) {} in ReduceMeanInfo()
[all …]
|
| D | matmul_info.h | 35 MatMulBase(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in MatMulBase() argument
37 : OperatorInfo(name, inputs_shape, outputs_shape, attrs, std::make_shared<MatMulCost>()) {} in MatMulBase()
73 …MatMul(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const Pri… in MatMul() argument
74 : MatMulBase(name, inputs_shape, outputs_shape, attrs) {} in MatMul()
83 MatMulInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in MatMulInfo() argument
85 : MatMul(name, inputs_shape, outputs_shape, attrs) {} in MatMulInfo()
91 BatchMatMulInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in BatchMatMulInfo() argument
93 : MatMul(name, inputs_shape, outputs_shape, attrs) {} in BatchMatMulInfo()
|
| D | unsorted_segment_op_info.h | 43 …UnsortedSegmentOpInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_s… in UnsortedSegmentOpInfo() argument
45 : OperatorInfo(name, inputs_shape, outputs_shape, attrs, cost) {} in UnsortedSegmentOpInfo()
68 …UnsortedSegmentSumInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_… in UnsortedSegmentSumInfo() argument
70 …: UnsortedSegmentOpInfo(name, inputs_shape, outputs_shape, attrs, std::make_shared<UnsortedSegment… in UnsortedSegmentSumInfo()
76 …UnsortedSegmentMinInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_… in UnsortedSegmentMinInfo() argument
78 …: UnsortedSegmentOpInfo(name, inputs_shape, outputs_shape, attrs, std::make_shared<UnsortedSegment… in UnsortedSegmentMinInfo()
89 …UnsortedSegmentMaxInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_… in UnsortedSegmentMaxInfo() argument
91 …: UnsortedSegmentOpInfo(name, inputs_shape, outputs_shape, attrs, std::make_shared<UnsortedSegment… in UnsortedSegmentMaxInfo()
|
| D | batch_parallel_info.h | 32 …BatchParallelInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in BatchParallelInfo() argument
34 : OperatorInfo(name, inputs_shape, outputs_shape, attrs, cost), dev_num_(1) {} in BatchParallelInfo()
35 …BatchParallelInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in BatchParallelInfo() argument
37 …: OperatorInfo(name, inputs_shape, outputs_shape, attrs, std::make_shared<BatchParallelCost>()), d… in BatchParallelInfo()
62 SparseSoftmaxCrossEntropyWithLogitsInfo(const std::string &name, const Shapes &inputs_shape, in SparseSoftmaxCrossEntropyWithLogitsInfo() argument
64 : BatchParallelInfo(name, inputs_shape, outputs_shape, attrs, in SparseSoftmaxCrossEntropyWithLogitsInfo()
|
| D | operator_info.cc | 59 Status OperatorInfo::CheckStrategyValue(const StrategyPtr &strategy, const Shapes &inputs_shape) { in CheckStrategyValue() argument
66 size_t inputs_shape_size = inputs_shape.size(); in CheckStrategyValue()
81 Shape sub_input_shape = inputs_shape.at(i); in CheckStrategyValue()
1061 Status GenerateStrategiesForTwoEqualInputs(int64_t stage_id, const Shapes &inputs_shape, in GenerateStrategiesForTwoEqualInputs() argument
1068 if ((inputs_shape.size() != 2) || (splittable_inputs.size() != 2)) { in GenerateStrategiesForTwoEqualInputs()
1073 if ((inputs_shape[0].size() != inputs_shape[1].size()) || in GenerateStrategiesForTwoEqualInputs()
1079 Shapes input0_shape = {inputs_shape[0]}; in GenerateStrategiesForTwoEqualInputs()
1094 Status GenerateStrategiesForBroadcastLeft(int64_t stage_id, const Shapes &inputs_shape, const Shape… in GenerateStrategiesForBroadcastLeft() argument
1101 if (inputs_shape[0].size() >= inputs_shape[1].size()) { in GenerateStrategiesForBroadcastLeft()
1107 Shapes tmp_inputs_shape = {inputs_shape[1], inputs_shape[1]}; in GenerateStrategiesForBroadcastLeft()
[all …]
|
| D | gather_v2_p_info.h | 34 GatherPInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape,
36 : OperatorInfo(name, inputs_shape, outputs_shape, attrs, std::make_shared<GatherV2PCost>()), in OperatorInfo() argument
93 …SparseGatherV2Info(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shap…
95 : GatherPInfo(name, inputs_shape, outputs_shape, attrs, replace_op_name) {} in GatherPInfo() argument
101 …EmbeddingLookupInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_sha… in EmbeddingLookupInfo() argument
103 : GatherPInfo(name, inputs_shape, outputs_shape, attrs) {} in EmbeddingLookupInfo()
|
| D | resizebilinear_info.h | 34 …ResizeBilinearInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &out… in ResizeBilinearInfo() argument
36 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<ResizeBilinearC… in ResizeBilinearInfo()
59 …ResizeNearestNeighborInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outpu… in ResizeNearestNeighborInfo() argument
61 : ResizeBilinearInfo(name, inputs_shape, outputs_shape, attrs) {} in ResizeNearestNeighborInfo()
|
| D | maxpool_info.h | 34 …MaxPoolInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_sh… in MaxPoolInfo() argument
36 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<MaxPoolCost>())… in MaxPoolInfo()
61 AvgPoolInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in AvgPoolInfo() argument
63 : MaxPoolInfo(name, inputs_shape, outputs_shape, attrs) {} in AvgPoolInfo()
|
| D | conv2d_info.h | 35 …Conv2DInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_sha… in Conv2DInfo() argument
37 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<BatchParallelCo… in Conv2DInfo()
126 …Conv2DBackpropInputInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs… in Conv2DBackpropInputInfo() argument
128 : Conv2DInfo(name, inputs_shape, outputs_shape, attrs) {} in Conv2DBackpropInputInfo()
153 …Conv2DTransposeInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_sha… in Conv2DTransposeInfo() argument
155 : Conv2DBackpropInputInfo(name, inputs_shape, outputs_shape, attrs) {} in Conv2DTransposeInfo()
|
| D | virtual_output_info.h | 34 …VirtualOutputInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in VirtualOutputInfo() argument
36 : VirtualDatasetInfo(name, inputs_shape, outputs_shape, attrs) {} in VirtualOutputInfo()
|
| D | l2_normalize_info.h | 34 L2NormalizeInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in L2NormalizeInfo() argument
36 : Activation(name, inputs_shape, outputs_shape, attrs, std::make_shared<L2NormalizeCost>()) {} in L2NormalizeInfo()
|
| /third_party/mindspore/tests/ut/cpp/parallel/ops_info/ |
| D | generate_strategy_test.cc | 56 Shapes inputs_shape = {{64, 32}, {64, 32}}; in TEST_F() local
58 GenerateStrategiesWithBroadcast(0, inputs_shape, splittable_inputs, &sp_vector); in TEST_F()
69 Shapes inputs_shape = {{32}, {64, 32}}; in TEST_F() local
71 GenerateStrategiesWithBroadcast(0, inputs_shape, splittable_inputs, &sp_vector); in TEST_F()
84 Shapes inputs_shape = {{64, 32}, {32}}; in TEST_F() local
86 GenerateStrategiesWithBroadcast(0, inputs_shape, splittable_inputs, &sp_vector); in TEST_F()
99 Shapes inputs_shape = {{64, 1}, {1, 32}}; in TEST_F() local
101 GenerateStrategiesWithBroadcast(0, inputs_shape, splittable_inputs, &sp_vector); in TEST_F()
113 Shapes inputs_shape = {{64, 8, 1}, {1, 8, 32}}; in TEST_F() local
115 GenerateStrategiesWithBroadcast(0, inputs_shape, splittable_inputs, &sp_vector); in TEST_F()
[all …]
|
| D | prelu_test.cc | 55 Shapes inputs_shape = {{64, 4, 8, 16}, {4}}; in SetUp() local
58 prelu = std::make_shared<PReLUInfo>("prelu_info", inputs_shape, outputs_shape, attr); in SetUp()
161 Shapes inputs_shape = {{64, 4, 8, 16}, {4}}; in TEST_F() local
162 GenerateStrategiesForIndependentInputs(0, inputs_shape, splittable_inputs, &sp_vector); in TEST_F()
266 Shapes inputs_shape = {{1024, 4}, {4}}; in TEST_F() local
267 GenerateStrategiesForIndependentInputs(0, inputs_shape, splittable_inputs, &sp_vector); in TEST_F()
|
| D | transpose_test.cc | 59 Shapes inputs_shape = {{128, 64}}; in SetUp() local
66 transpose = std::make_shared<TransposeInfo>("transpose_info", inputs_shape, outputs_shape, attr); in SetUp()
187 Shapes inputs_shape = {{128, 64}}; in TEST_F() local
188 GenerateStrategiesForIndependentInputs(0, inputs_shape, splittable_inputs, &sp_vector); in TEST_F()
|
| D | tmpidentity_test.cc | 58 Shapes inputs_shape = {{2, 4, 8, 16}}; in SetUp() local
60 identity_ptr = std::make_shared<TmpIdentityInfo>(inputs_shape, outputs_shape, attr); in SetUp()
161 Shapes inputs_shape = {{2, 4, 8, 16}}; in TEST_F() local
162 GenerateStrategiesForIndependentInputs(0, inputs_shape, splittable_inputs, &sp_vector); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | optimizer_info_builder.cc | 28 … const Lengths &lens, const InputsShapePtr &inputs_shape, size_t worker_num, in Build() argument
32 MS_EXCEPTION_IF_NULL(inputs_shape); in Build()
34 BuildInputs(weight, keys, values, lens, inputs_shape, worker_num, pserver_kernel, sharded); in Build()
58 const InputsShapePtr &inputs_shape) { in GenInputAddrPtr() argument
79 if (inputs_shape != nullptr) { in GenInputAddrPtr()
82 EXC_IF_VEC_IDX_OOB((*inputs_shape), origin_index); in GenInputAddrPtr()
83 MS_EXCEPTION_IF_NULL((*inputs_shape)[origin_index]); in GenInputAddrPtr()
84 auto shape = *((*inputs_shape)[origin_index]); in GenInputAddrPtr()
147 … const Lengths &lens, const InputsShapePtr &inputs_shape, size_t, in BuildInputs() argument
196 …enInputAddrPtr<float>(kSparseAdam, "grad", const_cast<float *>(values.data()), lens, inputs_shape); in BuildInputs()
[all …]
|
| D | optimizer_info_builder.h | 37 … const Values &values, const Lengths &lens, const InputsShapePtr &inputs_shape, size_t worker_num,
41 … const Lengths &lens, const InputsShapePtr &inputs_shape, size_t worker_num,
50 const Lengths &lens, const InputsShapePtr &inputs_shape = nullptr);
60 const InputsShapePtr &inputs_shape, size_t worker_num,
69 const InputsShapePtr &inputs_shape, size_t worker_num,
78 const InputsShapePtr &inputs_shape, size_t worker_num,
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | timedistributed.py | 27 def _check_reshape_pos(reshape_pos, inputs_shape, outputs_shape, prim_name=None): argument
29 if reshape_pos >= len(outputs_shape) or inputs_shape[reshape_pos] != outputs_shape[reshape_pos]:
125 inputs_shape = inputs.shape
126 time_axis_new = len(inputs_shape) - 2 if reshape_with_axis == len(inputs_shape) - 1 \
130 perm = _generate_perm(time_axis_new, time_axis, len(inputs_shape))
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | reluv2.cc | 73 abstract::ShapePtr inputs_shape; in InferShape() local
76 inputs_shape = std::make_shared<abstract::Shape>(input_shape); in InferShape()
78 …return std::make_shared<abstract::TupleShape>(std::vector<abstract::BaseShapePtr>{inputs_shape, ma… in InferShape()
82 inputs_shape = std::make_shared<abstract::Shape>(input_shape, min_shape, max_shape); in InferShape()
84 …return std::make_shared<abstract::TupleShape>(std::vector<abstract::BaseShapePtr>{inputs_shape, ma… in InferShape()
|
| D | ctcloss.cc | 41 auto inputs_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(inputs->BuildShape())[kShape]; in CheckCTCLossInputs() local
49 …(void)CheckAndConvertUtils::CheckInteger("inputs rank", SizeToLong(inputs_shape.size()), kEqual, i… in CheckCTCLossInputs()
63 if (inputs_shape[1] != sequence_length_shape[0]) { in CheckCTCLossInputs()
65 << inputs_shape[1] << " and " << sequence_length_shape[0]; in CheckCTCLossInputs()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | debug_ops.py | 367 def infer_shape(self, *inputs_shape): argument
368 if len(inputs_shape) == 1:
369 return inputs_shape[0]
370 return inputs_shape
|