Searched refs:alloc_size_by_instruction (Results 1 – 2 of 2) sorted by relevance
260 int64 alloc_size_by_instruction, in AccountForSubcomputationMemory() argument298 const HloInstruction* instruction, int64 alloc_size_by_instruction,
463 const HloInstruction* instruction, int64 alloc_size_by_instruction, in AccountForSubcomputationMemory() argument484 max_subcomputation_bytes -= alloc_size_by_instruction; in AccountForSubcomputationMemory()