Searched refs:output_indexs (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | kernel_build_ascend.cc | 139 … auto output_indexs = AnfAlgo::GetNodeAttr<std::vector<size_t>>(pre_node, kAttrAtomicOutputIndexs); in CalCleanZerosSize() local
141 for (auto index : output_indexs) { in CalCleanZerosSize()
253 std::vector<size_t> output_indexs = {}; in IsAtomicNode() local
255 output_indexs = AnfAlgo::GetNodeAttr<std::vector<size_t>>(kernel_node, kAttrAtomicOutputIndexs); in IsAtomicNode()
261 output_indexs.emplace_back(i); in IsAtomicNode()
266 if (!output_indexs.empty()) { in IsAtomicNode()
267 std::set<size_t> s(output_indexs.begin(), output_indexs.end()); in IsAtomicNode()
268 output_indexs.assign(s.begin(), s.end()); in IsAtomicNode()
269 AnfAlgo::SetNodeAttr(kAttrAtomicOutputIndexs, MakeValue(output_indexs), kernel_node); in IsAtomicNode()
283 return !(workspace_indexs.empty() && output_indexs.empty()); in IsAtomicNode()
|
| /third_party/mindspore/tests/ut/cpp/parallel/auto_parallel/ |
| D | edge_costmodel_test.cc | 155 std::vector<size_t> output_indexs, input_indexs; in TEST_F() local
156 output_indexs.push_back(0); in TEST_F()
160 std::make_shared<Edge>(edge_name, matmul1, matmul5, output_indexs, input_indexs, true); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | graph_costmodel.cc | 838 std::vector<size_t> output_indexs, input_indexs; in UpdateEdgesIncidentToNodes() local
839 output_indexs = ith_edge->prev_op_output_indexs(); in UpdateEdgesIncidentToNodes()
842 …std::make_shared<Edge>(new_edge_name, op1, ith_edge->next_operator(), output_indexs, input_indexs,… in UpdateEdgesIncidentToNodes()
865 std::vector<size_t> output_indexs, input_indexs; in UpdateEdgesIncidentToNodes() local
866 output_indexs = ith_edge->prev_op_output_indexs(); in UpdateEdgesIncidentToNodes()
868 …new_edge = std::make_shared<Edge>(new_edge_name, op1, destination, output_indexs, input_indexs, tr… in UpdateEdgesIncidentToNodes()
1036 std::vector<size_t> output_indexs, input_indexs; in EliminationOp() local
1043 output_indexs = edge_u_op->prev_op_output_indexs(); in EliminationOp()
1046 output_indexs.push_back(output_index); in EliminationOp()
1058 new_edge = std::make_shared<Edge>(new_edge_name, u, v, output_indexs, input_indexs, true); in EliminationOp()
[all …]
|