Searched refs:shared_memory_usage (Results 1 – 2 of 2) sorted by relevance
39 shared_memory_usage.erase(instr); in Invalidate()45 absl::flat_hash_map<const HloInstruction*, int64_t> shared_memory_usage; member
335 auto it_and_inserted = cache->shared_memory_usage.emplace(&instr, -1); in SharedMemoryUsage()