Searched refs:alloc_size_by_instruction (Results 1 – 2 of 2) sorted by relevance
233 int64 alloc_size_by_instruction, in AccountForSubcomputationMemory() argument257 const HloInstruction* instruction, int64 alloc_size_by_instruction,
243 int64 alloc_size_by_instruction = 0; in RunComputation() local276 alloc_size_by_instruction += size_fn_(*buffer); in RunComputation()284 instruction, alloc_size_by_instruction, *memory_by_computation_); in RunComputation()529 const HloInstruction* instruction, int64 alloc_size_by_instruction, in AccountForSubcomputationMemory() argument550 max_subcomputation_bytes -= alloc_size_by_instruction; in AccountForSubcomputationMemory()