Home
last modified time | relevance | path

Searched defs:InplaceAssignerInfo (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dinplace_assign_builder.cc38const std::vector<std::pair<InplaceAssignerInfo, AnfNodePtr>> &parameters_infos, size_t idx) { in CreateAssign()
68 …const AnfNodePtr &composite_node, const std::vector<std::pair<InplaceAssignerInfo, AnfNodePtr>> &i… in CorrectKernelBuildInfo()
96 const std::vector<std::pair<InplaceAssignerInfo, AnfNodePtr>> &parameters_infos) const { in CreateAssignNodeAndCorrectReturn() argument
180 const std::vector<std::pair<InplaceAssignerInfo, AnfNodePtr>> &info_and_inplace_assignee_addr) { in ProcessOriginCNode()
219 const std::vector<std::pair<InplaceAssignerInfo, AnfNodePtr>> &info_and_inplace_assignee_addr, in FindOriginCNodeUsers()
265 const std::vector<std::pair<InplaceAssignerInfo, AnfNodePtr>> &info_and_inplace_assignee_addr, in ProcessOriginCNodeUser()
Dinplace_assign_builder.h26 struct InplaceAssignerInfo { struct
Dadd_stitch_atomic_clean_gpu.cc30 …const AnfNodePtr &composite_node, const std::vector<std::pair<InplaceAssignerInfo, AnfNodePtr>> &i… in CorrectKernelBuildInfo()
85 const std::vector<std::pair<InplaceAssignerInfo, AnfNodePtr>> &info_and_inplace_assignee_addr) { in ProcessOriginCNode()
Dtsa_atomic_add_to_first_tensor.cc149 const std::vector<std::tuple<InplaceAssignerInfo, AnfNodePtr, size_t>> &outer_infos) const { in ChangeKernelBuildInfo() argument
179 const std::vector<std::tuple<InplaceAssignerInfo, AnfNodePtr, size_t>> &outer_nodes) const { in ProcessOriginalCNode() argument