| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | bias_add_info.h | 35 …BiasAddInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_sh… in BiasAddInfo() argument
37 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<BiasAddCost>())… in BiasAddInfo()
|
| D | reluv2_info.h | 38 …ReLUV2Info(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_sha… in ReLUV2Info() argument
40 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<ReLUV2Cost>()) … in ReLUV2Info()
|
| D | gathernd_info.h | 34 …GatherNdInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_s… in GatherNdInfo() argument
36 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<GatherNdCost>()… in GatherNdInfo()
|
| D | select_info.h | 34 …SelectInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_sha… in SelectInfo() argument
36 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<SelectCost>()) … in SelectInfo()
|
| D | pack_info.h | 34 …StackInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_shap… in StackInfo() argument
36 … : OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<PackCost>()) {} in StackInfo()
|
| D | split_info.h | 33 …SplitInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_shap… in SplitInfo() argument
35 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<SplitCost>()) {} in SplitInfo()
|
| D | uniform_real_info.h | 34 …UniformRealInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &output… in UniformRealInfo() argument
36 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<MaxPoolCost>())… in UniformRealInfo()
|
| D | concat_info.h | 34 …ConcatInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_sha… in ConcatInfo() argument
36 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<ConcatCost>()) … in ConcatInfo()
|
| D | broadcast_to_info.h | 37 …BroadcastToInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &output… in BroadcastToInfo() argument
39 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<BroadcastToCost… in BroadcastToInfo()
|
| D | topk_info.h | 35 …TopKInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_shape, in TopKInfo() argument
37 … : OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<TopKCost>()) {} in TopKInfo()
|
| D | gatherd_info.h | 34 …GatherDInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_sh… in GatherDInfo() argument
36 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<GatherDCost>())… in GatherDInfo()
|
| D | scatter_update_info.h | 34 …ScatterUpdateInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outp… in ScatterUpdateInfo() argument
36 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<ScatterUpdateCo… in ScatterUpdateInfo()
|
| D | batchnorm_info.h | 34 …BatchNormInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_… in BatchNormInfo() argument
36 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<BatchParallelCo… in BatchNormInfo()
|
| D | tile_info.h | 35 …TileInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_shape, in TileInfo() argument
37 … : OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<TileCost>()) {} in TileInfo()
|
| D | unique_info.h | 33 …UniqueInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_sha… in UniqueInfo() argument
35 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<UniqueCost>()) … in UniqueInfo()
|
| D | layer_norm_info.h | 41 …LayerNormInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_… in LayerNormInfo() argument
43 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<LayerNormCost>(… in LayerNormInfo()
|
| 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()
|
| D | strided_slice_info.h | 35 …StridedSliceInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outpu… in StridedSliceInfo() argument
37 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<StridedSliceCos… in StridedSliceInfo()
|
| 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()
|
| D | slice_info.h | 35 …SliceInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_shap… in SliceInfo() argument
37 … : OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<SliceCost>()), in SliceInfo()
|
| D | get_next_info.h | 33 …GetNextInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_sh… in GetNextInfo() argument
35 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<GetNextCost>())… in GetNextInfo()
|
| D | uniform_candidate_sampler_info.h | 34 …UniformCandidateSamplerInfo(const std::string &operator_name, const Shapes &inputs_shape, const Sh… in UniformCandidateSamplerInfo() argument
36 : OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, in UniformCandidateSamplerInfo()
|
| D | operator_info.cc | 297 OperatorName operator_name = VIRTUAL_DIV; in CreateVirtualDivOp() local
306 Operator op = std::make_pair(operator_name, operator_arg); in CreateVirtualDivOp()
325 OperatorName operator_name = ALL_REDUCE; in CreateAllReduceOp() local
328 Operator op = std::make_pair(operator_name, operator_arg); in CreateAllReduceOp()
334 OperatorName operator_name = REDUCE_SCATTER; in CreateReduceScatterOp() local
337 Operator op = std::make_pair(operator_name, operator_arg); in CreateReduceScatterOp()
400 OperatorName operator_name = ALL_GATHER; in CreateAllGatherOp() local
409 Operator op = std::make_pair(operator_name, operator_arg); in CreateAllGatherOp()
418 OperatorName operator_name = MINI_STEP_ALL_GATHER; in CreateMiniStepAllGatherOp() local
433 Operator op = std::make_pair(operator_name, operator_arg); in CreateMiniStepAllGatherOp()
[all …]
|
| 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()
|
| /third_party/mindspore/mindspore/profiler/parser/ |
| D | hccl_parser.py | 234 operator_name = os.path.basename(operator_dir)
235 …op_mapping_info = self._communication_operator_name_mapping_info.get(operator_name.split('_')[0], …
237 op_mapping_name = [item[1] for item in op_mapping_info if item[0] == operator_name]
242 operator_name = op_mapping_name[0]
243 operators_cost_info[operator_name] = operator_cost
|