Searched defs:alloc_size_by_instruction (Results 1 – 2 of 2) sorted by relevance
233 int64 alloc_size_by_instruction, in AccountForSubcomputationMemory()
243 int64 alloc_size_by_instruction = 0; in RunComputation() local529 const HloInstruction* instruction, int64 alloc_size_by_instruction, in AccountForSubcomputationMemory()