| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | elementary_function_info.h | 33 …string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const PrimitiveAttrs &attrs) in ExpInfo()
40 …string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const PrimitiveAttrs &attrs) in LogInfo()
47 …string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const PrimitiveAttrs &attrs) in CosInfo()
55 const PrimitiveAttrs &attrs) in ACosInfo()
63 const PrimitiveAttrs &attrs) in LogicalNotInfo()
70 …string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const PrimitiveAttrs &attrs) in AbsInfo()
78 const PrimitiveAttrs &attrs) in SignInfo()
86 const PrimitiveAttrs &attrs) in FloorInfo()
94 const PrimitiveAttrs &attrs) in RoundInfo()
102 const PrimitiveAttrs &attrs) in ReciprocalInfo()
[all …]
|
| D | arithmetic_info.h | 35 const PrimitiveAttrs &attrs, OperatorCostPtr cost) in ArithmeticBase()
55 …string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const PrimitiveAttrs &attrs) in SubInfo()
62 …string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const PrimitiveAttrs &attrs) in AddInfo()
69 …string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const PrimitiveAttrs &attrs) in MulInfo()
76 …string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const PrimitiveAttrs &attrs) in DivInfo()
83 …string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const PrimitiveAttrs &attrs) in ModInfo()
91 const PrimitiveAttrs &attrs) in RealDivInfo()
99 const PrimitiveAttrs &attrs) in FloorDivInfo()
107 const PrimitiveAttrs &attrs) in FloorModInfo()
114 …string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const PrimitiveAttrs &attrs) in PowInfo()
[all …]
|
| D | activation_info.h | 35 const PrimitiveAttrs &attrs, OperatorCostPtr cost) in ActivationBase()
52 const PrimitiveAttrs &attrs, OperatorCostPtr cost) in Activation()
65 const PrimitiveAttrs &attrs) in ActivationInfo()
76 const PrimitiveAttrs &attrs, OperatorCostPtr cost) in ActivationOther()
87 const PrimitiveAttrs &attrs) in GeLUInfo()
95 const PrimitiveAttrs &attrs) in FastGeLUInfo()
103 const PrimitiveAttrs &attrs) in TanhInfo()
111 const PrimitiveAttrs &attrs) in Softmax()
128 const PrimitiveAttrs &attrs) in SoftmaxInfo()
136 const PrimitiveAttrs &attrs) in LogSoftmaxInfo()
[all …]
|
| D | comparison_function_info.h | 34 const PrimitiveAttrs &attrs) in EqualInfo()
42 const PrimitiveAttrs &attrs) in ApproximateEqualInfo()
50 const PrimitiveAttrs &attrs) in NotEqualInfo()
58 const PrimitiveAttrs &attrs) in MaximumInfo()
66 const PrimitiveAttrs &attrs) in MinimumInfo()
74 const PrimitiveAttrs &attrs) in GreaterInfo()
82 const PrimitiveAttrs &attrs) in GreaterEqualInfo()
90 const PrimitiveAttrs &attrs) in LessInfo()
98 const PrimitiveAttrs &attrs) in LessEqualInfo()
|
| D | reduce_method_info.h | 36 const PrimitiveAttrs &attrs, OperatorCostPtr cost) in ReduceMethod()
64 const PrimitiveAttrs &attrs) in ReduceMaxInfo()
75 const PrimitiveAttrs &attrs) in ArgMaxWithValueInfo()
96 const PrimitiveAttrs &attrs) in ArgMinWithValueInfo()
107 const PrimitiveAttrs &attrs) in ReduceMeanInfo()
119 const PrimitiveAttrs &attrs) in ReduceSumInfo()
130 const PrimitiveAttrs &attrs) in ReduceAnyInfo()
143 const PrimitiveAttrs &attrs) in ReduceMinInfo()
|
| D | matmul_info.h | 36 const PrimitiveAttrs &attrs) in MatMulBase()
73 …string &name, const Shapes &inputs_shape, const Shapes &outputs_shape, const PrimitiveAttrs &attrs) in MatMul()
84 const PrimitiveAttrs &attrs) in MatMulInfo()
92 const PrimitiveAttrs &attrs) in BatchMatMulInfo()
|
| D | unsorted_segment_op_info.h | 44 const PrimitiveAttrs &attrs, OperatorCostPtr cost) in UnsortedSegmentOpInfo()
69 const PrimitiveAttrs &attrs) in UnsortedSegmentSumInfo()
77 const PrimitiveAttrs &attrs) in UnsortedSegmentMinInfo()
90 const PrimitiveAttrs &attrs) in UnsortedSegmentMaxInfo()
|
| D | batch_parallel_info.h | 33 const PrimitiveAttrs &attrs, OperatorCostPtr cost) in BatchParallelInfo()
36 const PrimitiveAttrs &attrs) in BatchParallelInfo()
63 const Shapes &outputs_shape, const PrimitiveAttrs &attrs) in SparseSoftmaxCrossEntropyWithLogitsInfo()
|
| D | gather_v2_p_info.h | 35 const PrimitiveAttrs &attrs, const std::string &replace_op_name = GATHERV2)
94 … const PrimitiveAttrs &attrs, const std::string &replace_op_name = SPARSE_GATHERV2)
102 const PrimitiveAttrs &attrs) in EmbeddingLookupInfo()
|
| D | resizebilinear_info.h | 35 const PrimitiveAttrs &attrs) in ResizeBilinearInfo()
60 const PrimitiveAttrs &attrs) in ResizeNearestNeighborInfo()
|
| D | maxpool_info.h | 35 const PrimitiveAttrs &attrs) in MaxPoolInfo()
62 const PrimitiveAttrs &attrs) in AvgPoolInfo()
|
| D | conv2d_info.h | 36 const PrimitiveAttrs &attrs) in Conv2DInfo()
127 const PrimitiveAttrs &attrs) in Conv2DBackpropInputInfo()
154 const PrimitiveAttrs &attrs) in Conv2DTransposeInfo()
|
| D | virtual_output_info.h | 35 const PrimitiveAttrs &attrs) in VirtualOutputInfo()
|
| D | l2_normalize_info.h | 35 const PrimitiveAttrs &attrs) in L2NormalizeInfo()
|
| D | bias_add_info.h | 36 const PrimitiveAttrs &attrs) in BiasAddInfo()
|
| D | reluv2_info.h | 39 const PrimitiveAttrs &attrs) in ReLUV2Info()
|
| D | prelu_info.h | 37 const PrimitiveAttrs &attrs) in PReLUInfo()
|
| D | gathernd_info.h | 35 const PrimitiveAttrs &attrs) in GatherNdInfo()
|
| D | select_info.h | 35 const PrimitiveAttrs &attrs) in SelectInfo()
|
| D | tmp_identity_info.h | 35 …TmpIdentityInfo(const Shapes &inputs_shape, const Shapes &outputs_shape, const PrimitiveAttrs &att…
|
| D | transpose_info.h | 37 const PrimitiveAttrs &attrs) in TransposeInfo()
|
| D | pack_info.h | 35 const PrimitiveAttrs &attrs) in StackInfo()
|
| D | split_info.h | 34 const PrimitiveAttrs &attrs) in SplitInfo()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | dynamic_creator.h | 31 …OperatorInfoPtr objectCreator##className(std::string name, Shapes in, Shapes out, PrimitiveAttrs &…
37 const PrimitiveAttrs &attrs);
52 const PrimitiveAttrs &attrs, size_t count) { in Create()
|
| D | step_parallel.h | 91 OperatorInfoPtr OperatorInstance(const PrimitivePtr &prim, const PrimitiveAttrs &attrs,
95 OperatorInfoPtr NewOperatorInstance(const PrimitivePtr &prim, const PrimitiveAttrs &attrs,
|