| /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()
40 …LogInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const Pr… in LogInfo()
47 …CosInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const Pr… in CosInfo()
54 ACosInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ACosInfo()
62 LogicalNotInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in LogicalNotInfo()
70 …AbsInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const Pr… in AbsInfo()
77 SignInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in SignInfo()
85 FloorInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in FloorInfo()
93 RoundInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in RoundInfo()
101 ReciprocalInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ReciprocalInfo()
[all …]
|
| D | arithmetic_info.h | 34 …eticBase(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ArithmeticBase()
55 …SubInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const Pr… in SubInfo()
62 …AddInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const Pr… in AddInfo()
69 …MulInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const Pr… in MulInfo()
76 …DivInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const Pr… in DivInfo()
83 …ModInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const Pr… in ModInfo()
90 RealDivInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in RealDivInfo()
98 FloorDivInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in FloorDivInfo()
106 FloorModInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in FloorModInfo()
114 …PowInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const Pr… in PowInfo()
[all …]
|
| D | activation_info.h | 34 …tionBase(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ActivationBase()
51 Activation(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in Activation()
64 ActivationInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ActivationInfo()
75 ActivationOther(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ActivationOther()
86 GeLUInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in GeLUInfo()
94 FastGeLUInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in FastGeLUInfo()
102 TanhInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in TanhInfo()
110 explicit Softmax(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in Softmax()
127 SoftmaxInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in SoftmaxInfo()
135 LogSoftmaxInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in LogSoftmaxInfo()
[all …]
|
| D | comparison_function_info.h | 33 EqualInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in EqualInfo()
41 …roximateEqualInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ApproximateEqualInfo()
49 NotEqualInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in NotEqualInfo()
57 MaximumInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in MaximumInfo()
65 MinimumInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in MinimumInfo()
73 GreaterInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in GreaterInfo()
81 GreaterEqualInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in GreaterEqualInfo()
89 LessInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in LessInfo()
97 LessEqualInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in LessEqualInfo()
|
| D | reduce_method_info.h | 35 ReduceMethod(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ReduceMethod()
63 ReduceMaxInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ReduceMaxInfo()
74 …gMaxWithValueInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ArgMaxWithValueInfo()
95 …gMinWithValueInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ArgMinWithValueInfo()
106 ReduceMeanInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ReduceMeanInfo()
118 ReduceSumInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ReduceSumInfo()
129 ReduceAnyInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ReduceAnyInfo()
142 ReduceMinInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ReduceMinInfo()
|
| D | unsorted_segment_op_info.h | 43 …rtedSegmentOpInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in UnsortedSegmentOpInfo()
68 …tedSegmentSumInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in UnsortedSegmentSumInfo()
76 …tedSegmentMinInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in UnsortedSegmentMinInfo()
89 …tedSegmentMaxInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in UnsortedSegmentMaxInfo()
|
| D | matmul_info.h | 35 MatMulBase(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in MatMulBase()
73 …MatMul(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const Pri… in MatMul()
83 MatMulInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in MatMulInfo()
91 BatchMatMulInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in BatchMatMulInfo()
|
| D | batch_parallel_info.h | 32 …BatchParallelInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in BatchParallelInfo()
35 …BatchParallelInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in BatchParallelInfo()
63 const Shapes &outputs_shape, const PrimitiveAttrs &attrs) in SparseSoftmaxCrossEntropyWithLogitsInfo()
|
| D | resizebilinear_info.h | 34 …nearInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ResizeBilinearInfo()
59 …arestNeighborInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ResizeNearestNeighborInfo()
|
| D | gather_v2_p_info.h | 36 : OperatorInfo(name, inputs_shape, outputs_shape, attrs, std::make_shared<GatherV2PCost>()), in OperatorInfo() argument
95 : GatherPInfo(name, inputs_shape, outputs_shape, attrs, replace_op_name) {} in GatherPInfo() argument
101 …beddingLookupInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in EmbeddingLookupInfo()
|
| D | maxpool_info.h | 34 …PoolInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_shape, in MaxPoolInfo()
61 AvgPoolInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in AvgPoolInfo()
|
| D | conv2d_info.h | 35 …nv2DInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_shape, in Conv2DInfo()
126 …BackpropInputInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in Conv2DBackpropInputInfo()
153 …nv2DTransposeInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in Conv2DTransposeInfo()
|
| D | l2_normalize_info.h | 34 L2NormalizeInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in L2NormalizeInfo()
|
| D | reluv2_info.h | 38 …LUV2Info(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ReLUV2Info()
|
| D | virtual_output_info.h | 34 …VirtualOutputInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in VirtualOutputInfo()
|
| D | prelu_info.h | 36 PReLUInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in PReLUInfo()
|
| D | loss_info.h | 37 …opyWithLogitsInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in SoftmaxCrossEntropyWithLogitsInfo()
|
| D | broadcast_to_info.h | 37 …stToInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_shape, in BroadcastToInfo()
|
| D | gatherd_info.h | 34 …herDInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_shape, in GatherDInfo()
|
| D | split_info.h | 33 …plitInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_shape, in SplitInfo()
|
| D | concat_info.h | 34 …ncatInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_shape, in ConcatInfo()
|
| D | virtual_dataset_info.h | 33 …irtualDatasetInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in VirtualDatasetInfo()
|
| D | transpose_info.h | 36 TransposeInfo(const std::string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, in TransposeInfo()
|
| D | uniform_real_info.h | 34 …RealInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_shape, in UniformRealInfo()
|
| D | bias_add_info.h | 35 …sAddInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_shape, in BiasAddInfo()
|