Searched defs:ref_map (Results 1 – 8 of 8) sorted by relevance
226 void KernelInfo::set_ref_map(const bool &all_ref, const OutputInputRefMap &ref_map) { in set_ref_map()
82 absl::flat_hash_map<Regexp*, int> ref_map; member90 static inline absl::flat_hash_map<Regexp*, int>* ref_map() { in ref_map() function
277 const auto &ref_map = graph->GetRefMap(); in AddSwappableTensors() local
162 const auto &ref_map = graph->GetRefMap(); in OptimizeNopNode() local
266 void RefreshRefFormat(const std::unordered_map<size_t, size_t> &ref_map, const std::vector<std::str… in RefreshRefFormat() argument
811 auto ref_map = kernel_graph->GetRefMap(); in CreateOutputDeviceAddress() local869 auto ref_map = kernel_graph->GetRefMap(); in AllocOutputMemory() local
272 const auto &ref_map = sub_graph->GetRefMap(); in InlineSubGraph() local
2865 … const std::map<session::AnfWithOutIndex, session::AnfWithOutIndex> &ref_map, in DoInline()