Searched refs:tensor_mem (Results 1 – 2 of 2) sorted by relevance
93 const std::map<int64, std::vector<string>>& tensor_mem) { in EmitCounter() argument110 for (auto it = tensor_mem.rbegin(); it != tensor_mem.rend(); ++it) { in EmitCounter()289 std::map<int64, std::vector<string>> tensor_mem; in GenerateGraphTimeline() local297 tensor_mem[it->second].push_back(tensor_alloc_it.first); in GenerateGraphTimeline()301 dev.first, cur_bytes_in_use, tensor_mem); in GenerateGraphTimeline()
53 const std::map<int64, std::vector<string>>& tensor_mem);