| /third_party/mindspore/tests/ut/cpp/parallel/tensor_layout/ |
| D | tensor_redistribution_test.cc | 86 TensorRedistribution tensor_redistribution; in TEST_F() local
88 tensor_redistribution.Init(from_layout, to_layout, dev_list); in TEST_F()
90 op_ptr = tensor_redistribution.InferTensorRedistributionOperatorList(); in TEST_F()
139 TensorRedistribution tensor_redistribution; in TEST_F() local
141 tensor_redistribution.Init(from_layout, to_layout, dev_list); in TEST_F()
143 op_ptr = tensor_redistribution.InferTensorRedistributionOperatorList(); in TEST_F()
192 TensorRedistribution tensor_redistribution; in TEST_F() local
194 tensor_redistribution.Init(from_layout, to_layout, dev_list); in TEST_F()
196 op_ptr = tensor_redistribution.InferTensorRedistributionOperatorList(); in TEST_F()
|
| D | redistribution_layout_transfer_test.cc | 64 RedistributionLayoutTransfer tensor_redistribution; in RedistributionLayoutTransferTestFunction() local
65 status = tensor_redistribution.Init(in_tensor_layout, out_tensor_layout); in RedistributionLayoutTransferTestFunction()
67 …std::shared_ptr<ReshapeLayoutTransfer> unified_ptr = tensor_redistribution.UnifyDeviceArrangementA… in RedistributionLayoutTransferTestFunction()
|
| D | reshape_layout_transfer_test.cc | 64 ReshapeLayoutTransfer tensor_redistribution; in InferUnifiedLayout() local
65 status = tensor_redistribution.Init(in_tensor_layout, out_tensor_layout); in InferUnifiedLayout()
67 …std::shared_ptr<ReshapeLayoutTransfer> unified_ptr = tensor_redistribution.UnifyDeviceArrangementA… in InferUnifiedLayout()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | edge_costmodel.cc | 119 TensorRedistribution tensor_redistribution(false); in GetRedistributionCost() local
122 if (tensor_redistribution.Init(prev_op_output_layout, next_op_input_layout, dev_list) == FAILED) { in GetRedistributionCost()
126 if (tensor_redistribution.ComputeCost() == FAILED) { in GetRedistributionCost()
130 double comm_cost = tensor_redistribution.comm_cost(); in GetRedistributionCost()
131 double forward_comm_cost = tensor_redistribution.forward_comm_cost(); in GetRedistributionCost()
132 double backward_comm_cost = tensor_redistribution.backward_comm_cost(); in GetRedistributionCost()
133 double computation_cost = tensor_redistribution.computation_cost(); in GetRedistributionCost()
134 double mem_cost = tensor_redistribution.memory_cost(); in GetRedistributionCost()
|
| D | operator_costmodel.cc | 672 TensorRedistribution tensor_redistribution(false, true); in GetForwardCommCost() local
673 …if (tensor_redistribution.Init(inputs[0].tensor_layout(), outputs[0].tensor_layout(), dev_list) ==… in GetForwardCommCost()
676 if (tensor_redistribution.ComputeCost() == FAILED) { in GetForwardCommCost()
679 return (inputs_type_lengths_[0] * tensor_redistribution.comm_cost()); in GetForwardCommCost()
710 TensorRedistribution tensor_redistribution(false, true); in GetForwardComputationCost() local
711 …if (tensor_redistribution.Init(inputs[0].tensor_layout(), outputs[0].tensor_layout(), dev_list) ==… in GetForwardComputationCost()
714 if (tensor_redistribution.ComputeCost() == FAILED) { in GetForwardComputationCost()
717 return (inputs_type_lengths_[0] * tensor_redistribution.computation_cost()); in GetForwardComputationCost()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel.h | 69 … const CNodePtr &middle_node, int64_t index, TensorRedistribution tensor_redistribution,
79 … const TensorRedistribution &tensor_redistribution, const CNodePtr &pre_node);
|
| D | step_parallel.cc | 430 … const CNodePtr &middle_node, int64_t index, TensorRedistribution tensor_redistribution, in Redistribution() argument
468 if (tensor_redistribution.Init(tensorlayout_in, tensorlayout_out, dev_list) == FAILED) { in Redistribution()
475 …RedistributionOpListPtr redistribution_oplist_ptr = tensor_redistribution.InferTensorRedistributio… in Redistribution()
544 … const TensorRedistribution &tensor_redistribution, const CNodePtr &pre_node) { in StepRedistribution() argument
572 …StepRedistribution(use_cnode, distribute_operator, insert_node_new, tensor_redistribution, pre_nod… in StepRedistribution()
582 …tribution(node_pair, distribute_operator, insert_node_new, node_pair.second, tensor_redistribution, in StepRedistribution()
585 …StepRedistribution(use_cnode, distribute_operator, insert_node_new, tensor_redistribution, pre_nod… in StepRedistribution()
2103 TensorRedistribution tensor_redistribution; in InferSensRedistribution() local
2127 if (tensor_redistribution.Init(stand_alone_layout, loss_layout, dev_list) == FAILED) { in InferSensRedistribution()
2130 …RedistributionOpListPtr sens_redistribution_list = tensor_redistribution.InferTensorRedistribution… in InferSensRedistribution()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | reshape_info.cc | 117 TensorRedistribution tensor_redistribution(!is_generating_costs_, true); in ComputeReplaceOp() local
118 if (tensor_redistribution.Init(input_layout_, output_layout_, dev_list) == FAILED) { in ComputeReplaceOp()
136 …RedistributionOpListPtr redistribution_oplist_ptr = tensor_redistribution.InferTensorRedistributio… in ComputeReplaceOp()
|