Searched refs:alloc_size_by_instruction (Results 1 – 2 of 2) sorted by relevance
266 int64_t alloc_size_by_instruction, in AccountForSubcomputationMemory() argument304 const HloInstruction* instruction, int64_t alloc_size_by_instruction,
466 const HloInstruction* instruction, int64_t alloc_size_by_instruction, in AccountForSubcomputationMemory() argument487 max_subcomputation_bytes -= alloc_size_by_instruction; in AccountForSubcomputationMemory()