Searched defs:InplaceAssignerInfo (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/ |
| D | inplace_assign_builder.cc | 38 … const std::vector<std::pair<InplaceAssignerInfo, AnfNodePtr>> ¶meters_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>> ¶meters_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()
|
| D | inplace_assign_builder.h | 26 struct InplaceAssignerInfo { struct
|
| D | add_stitch_atomic_clean_gpu.cc | 30 …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()
|
| D | tsa_atomic_add_to_first_tensor.cc | 149 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
|