Home
last modified time | relevance | path

Searched refs:cover_index (Results 1 – 1 of 1) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dcudnn_inplace_fusion.cc101 bool ExistDependencyFromAcc2Cover(const std::vector<AnfNodeIndex> &inplace_node, size_t cover_index in ExistDependencyFromAcc2Cover() argument
106 size_t acc_index = cover_index == 1 ? 0 : 1; in ExistDependencyFromAcc2Cover()
107 const CNodePtr &cover_node = inplace_node[cover_index].node->cast<CNodePtr>(); in ExistDependencyFromAcc2Cover()
139 size_t cover_index = (first_channel >= second_channel) ? 0 : 1; in GetCoverIndex() local
140 bool ret = ExistDependencyFromAcc2Cover(inplace_node, cover_index); in GetCoverIndex()
144 return {cover_index, true}; in GetCoverIndex()
160 void CheckInplaceNodeInputs(std::vector<AnfNodeIndex> *inplace_node, size_t cover_index, const Func… in CheckInplaceNodeInputs() argument
176 size_t acc_index = cover_index == 1 ? 0 : 1; in CheckInplaceNodeInputs()
177 const CNodePtr &cover_node = inplace_node->at(cover_index).node->cast<CNodePtr>(); in CheckInplaceNodeInputs()
213 auto [cover_index, order_required] = GetCoverIndex(*inplace_node); in SetNodeAttr()
[all …]