| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/ |
| D | rec_generate_strategy.h | 30 …trategy(const std::shared_ptr<Graph> &graph, const std::vector<std::shared_ptr<OperatorInfo>> &ops,
34 …eMatMul(const std::shared_ptr<Graph> &graph, const std::vector<std::shared_ptr<OperatorInfo>> &ops,
37 Strategys PrepareStridedSlice(const std::vector<std::shared_ptr<OperatorInfo>> &ops, const size_t i…
39 …eOneHot(const std::shared_ptr<Graph> &graph, const std::vector<std::shared_ptr<OperatorInfo>> &ops,
42 … const std::vector<std::shared_ptr<OperatorInfo>> &ops, const size_t iter_graph,
44 Strategys PrepareGatherV2(const std::vector<std::shared_ptr<OperatorInfo>> &ops, const size_t iter_…
45 Strategys PrepareGatherV2P(const std::vector<std::shared_ptr<OperatorInfo>> &ops, const size_t iter…
46 Dimensions PrepareGatherV2POutputStrategy(const std::vector<std::shared_ptr<OperatorInfo>> &ops,
48 Strategys PrepareL2Normalize(const std::vector<std::shared_ptr<OperatorInfo>> &ops, const size_t it…
51 … const std::vector<std::shared_ptr<OperatorInfo>> &ops, const size_t iter_graph,
[all …]
|
| D | rec_generate_strategy.cc | 32 …trategy(const std::shared_ptr<Graph> &graph, const std::vector<std::shared_ptr<OperatorInfo>> &ops, in GenerateStrategy()
65 …eMatMul(const std::shared_ptr<Graph> &graph, const std::vector<std::shared_ptr<OperatorInfo>> &ops, in PrepareMatMul()
156 Strategys PrepareStridedSlice(const std::vector<std::shared_ptr<OperatorInfo>> &ops, const size_t i… in PrepareStridedSlice()
181 …eOneHot(const std::shared_ptr<Graph> &graph, const std::vector<std::shared_ptr<OperatorInfo>> &ops, in PrepareOneHot()
208 Strategys PrepareGatherV2(const std::vector<std::shared_ptr<OperatorInfo>> &ops, const size_t iter_… in PrepareGatherV2()
225 Strategys PrepareGatherV2P(const std::vector<std::shared_ptr<OperatorInfo>> &ops, const size_t iter… in PrepareGatherV2P()
291 Dimensions PrepareGatherV2POutputStrategy(const std::vector<std::shared_ptr<OperatorInfo>> &ops, in PrepareGatherV2POutputStrategy()
321 Strategys PrepareL2Normalize(const std::vector<std::shared_ptr<OperatorInfo>> &ops, const size_t it… in PrepareL2Normalize()
348 … const std::vector<std::shared_ptr<OperatorInfo>> &ops, const size_t iter_graph, in PrepareAxisRelatedStrategy()
400 … const std::vector<std::shared_ptr<OperatorInfo>> &ops, const size_t iter_graph, in MakeRecSearchStrategy()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | edge_costmodel.h | 34 using OperatorInfoPtr = std::shared_ptr<mindspore::parallel::OperatorInfo>;
40 Edge(const std::string &edge_name, const std::shared_ptr<OperatorInfo> &prev_op, in Edge()
41 …const std::shared_ptr<OperatorInfo> &next_op, const size_t &output_index_, const size_t &input_ind… in Edge()
52 Edge(const std::string &edge_name, const std::shared_ptr<OperatorInfo> &prev_op, in Edge()
53 …const std::shared_ptr<OperatorInfo> &next_op, const size_t &output_index_, const size_t &input_ind… in Edge()
63 Edge(const std::string &edge_name, const std::shared_ptr<OperatorInfo> &prev_op, in Edge()
64 const std::shared_ptr<OperatorInfo> &next_op, const std::vector<size_t> &output_indexs_, in Edge()
78 std::shared_ptr<OperatorInfo> prev_operator() const { return prev_op_; } in prev_operator()
79 std::shared_ptr<OperatorInfo> next_operator() const { return next_op_; } in next_operator()
122 …void EdgeEliminationSetNewCost(std::shared_ptr<OperatorInfo> u, const std::vector<std::shared_ptr<…
[all …]
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/parallel/ |
| D | operator_info.h | 45 class OperatorInfo; variable
46 using OperatorInfoPtr = std::shared_ptr<OperatorInfo>;
47 class OperatorInfo {
49 OperatorInfo(const std::string &name, const SplitStrategy &strategy) in OperatorInfo() function
55 virtual ~OperatorInfo() = default;
98 std::unique_ptr<OperatorInfo> OperatorInfoCreator(const std::string &name, const SplitStrategy &str… in OperatorInfoCreator()
|
| D | operator_info.cc | 38 std::shared_ptr<abstract::AbstractTensor> OperatorInfo::CreateFakeAbstractTensor() const { in CreateFakeAbstractTensor()
44 int OperatorInfo::CheckSplitResult(const AnfNodePtr &result_anf_node, const std::vector<AnfNodePtr>… in CheckSplitResult()
53 void OperatorInfo::Init(const FuncGraphPtr &func_graph, const CNodePtr &cnode, int32_t fmk_type) { in Init()
60 int OperatorInfo::SetCNodeBackend() { in SetCNodeBackend()
87 int OperatorInfo::CheckStrategyValue() { in CheckStrategyValue()
106 int OperatorInfo::CreateMultipleOutputsOfAnfNode(const AnfNodePtr &node, size_t output_num, in CreateMultipleOutputsOfAnfNode()
137 AnfNodePtr OperatorInfo::CreateConcateNode(const CNodePtr &orig_node, const std::vector<AnfNodePtr>… in CreateConcateNode()
164 AnfNodePtr OperatorInfo::CreateReduceNode(const CNodePtr &orig_node, const std::vector<AnfNodePtr> … in CreateReduceNode()
189 int OperatorInfo::DoSplit() { in DoSplit()
|
| D | conv2d_info.h | 31 class Conv2DInfo : public OperatorInfo {
33 …Conv2DInfo(const std::string &name, const SplitStrategy &strategy) : OperatorInfo(name, strategy) … in Conv2DInfo()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | operator_info.cc | 59 Status OperatorInfo::CheckStrategyValue(const StrategyPtr &strategy, const Shapes &inputs_shape) { in CheckStrategyValue()
136 void OperatorInfo::ResetQueueMember() { in ResetQueueMember()
150 Status OperatorInfo::InferAttrs() { in InferAttrs()
162 Status OperatorInfo::InferMirrorOps() { in InferMirrorOps()
202 Status OperatorInfo::InferTensorInfo() { in InferTensorInfo()
231 Status OperatorInfo::InferRepeatedCalcInfo() { in InferRepeatedCalcInfo()
257 void OperatorInfo::SetRepeatedCalcDevMatrix() { in SetRepeatedCalcDevMatrix()
270 void OperatorInfo::ResetTensorMapIfRepeatedCalc() { in ResetTensorMapIfRepeatedCalc()
524 Status OperatorInfo::CreateGroupByTensorMap(const Shape &tensor_map, std::vector<Group> *group) { in CreateGroupByTensorMap()
547 Status OperatorInfo::CreateGroupForOptShard(TensorLayout *const tensor_layout, std::vector<Group> *… in CreateGroupForOptShard()
[all …]
|
| D | batch_parallel_info.h | 30 class BatchParallelInfo : public OperatorInfo {
34 : OperatorInfo(name, inputs_shape, outputs_shape, attrs, cost), dev_num_(1) {} in BatchParallelInfo()
37 …: OperatorInfo(name, inputs_shape, outputs_shape, attrs, std::make_shared<BatchParallelCost>()), d… in BatchParallelInfo()
|
| D | prelu_info.h | 34 class PReLUInfo : public OperatorInfo {
38 : OperatorInfo(name, inputs_shape, outputs_shape, attrs, std::make_shared<PReLUCost>()) {} in PReLUInfo()
|
| D | reluv2_info.h | 36 class ReLUV2Info : public OperatorInfo {
40 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<ReLUV2Cost>()) … in ReLUV2Info()
|
| D | bias_add_info.h | 33 class BiasAddInfo : public OperatorInfo {
37 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<BiasAddCost>())… in BiasAddInfo()
|
| D | operator_info.h | 55 class OperatorInfo {
57 …OperatorInfo(std::string name, Shapes inputs_shape, Shapes outputs_shape, PrimitiveAttrs attrs, Op… in OperatorInfo() function
73 virtual ~OperatorInfo() = default;
135 …void ReplacePreEdge(const std::shared_ptr<OperatorInfo> &op, const std::shared_ptr<Edge> &new_edge…
136 …void ReplaceSuccEdge(const std::shared_ptr<OperatorInfo> &op, const std::shared_ptr<Edge> &new_edg…
137 …void ReplacePreEdges(const std::shared_ptr<OperatorInfo> &op, const std::shared_ptr<Edge> &new_edg…
138 …void ReplaceSuccEdges(const std::shared_ptr<OperatorInfo> &op, const std::shared_ptr<Edge> &new_ed…
|
| D | tmp_identity_info.h | 30 class TmpIdentityInfo : public OperatorInfo {
37 … : OperatorInfo(name, inputs_shape, outputs_shape, attrs, std::make_shared<TmpIdentityCost>()) {} in OperatorInfo() function
|
| D | gathernd_info.h | 32 class GatherNdInfo : public OperatorInfo {
36 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<GatherNdCost>()… in GatherNdInfo()
|
| D | select_info.h | 32 class SelectInfo : public OperatorInfo {
36 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<SelectCost>()) … in SelectInfo()
|
| D | split_info.h | 31 class SplitInfo : public OperatorInfo {
35 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<SplitCost>()) {} in SplitInfo()
|
| D | concat_info.h | 32 class ConcatInfo : public OperatorInfo {
36 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<ConcatCost>()) … in ConcatInfo()
|
| D | loss_info.h | 35 class SoftmaxCrossEntropyWithLogitsInfo : public OperatorInfo {
39 …: OperatorInfo(name, inputs_shape, outputs_shape, attrs, std::make_shared<SoftmaxCrossEntropyWithL… in SoftmaxCrossEntropyWithLogitsInfo()
|
| D | scatter_update_info.h | 32 class ScatterUpdateInfo : public OperatorInfo {
36 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<ScatterUpdateCo… in ScatterUpdateInfo()
|
| D | transpose_info.h | 34 class TransposeInfo : public OperatorInfo {
38 : OperatorInfo(name, inputs_shape, outputs_shape, attrs, std::make_shared<TransposeCost>()) {} in TransposeInfo()
|
| D | uniform_real_info.h | 32 class UniformRealInfo : public OperatorInfo {
36 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<MaxPoolCost>())… in UniformRealInfo()
|
| D | broadcast_to_info.h | 35 class BroadcastToInfo : public OperatorInfo {
39 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<BroadcastToCost… in BroadcastToInfo()
|
| D | topk_info.h | 33 class TopKInfo : public OperatorInfo {
37 … : OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<TopKCost>()) {} in TopKInfo()
|
| D | gatherd_info.h | 32 class GatherDInfo : public OperatorInfo {
36 …: OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<GatherDCost>())… in GatherDInfo()
|
| D | pack_info.h | 32 class StackInfo : public OperatorInfo {
36 … : OperatorInfo(operator_name, inputs_shape, outputs_shape, attrs, std::make_shared<PackCost>()) {} in StackInfo()
|