Searched refs:callee_usage (Results 1 – 1 of 1) sorted by relevance
945 TF_ASSIGN_OR_RETURN(int64 callee_usage, in ComputePeakMemory()948 std::max<int64>(peak_memory, tracker.memory_usage() + callee_usage); in ComputePeakMemory()963 int64 callee_usage = 0; in CalledComputationsMemoryUsage() local966 callee_usage += computation_peak_memory_.at(computation); in CalledComputationsMemoryUsage()968 return callee_usage; in CalledComputationsMemoryUsage()1014 TF_ASSIGN_OR_RETURN(int64 callee_usage, in RematerializeComputation()1020 << ", callee usage = " << callee_usage << ", [" << instruction_index in RematerializeComputation()1024 while (memory_tracker.memory_usage() + callee_usage > memory_limit_bytes) { in RematerializeComputation()1028 callee_usage) in RematerializeComputation()1039 callee_usage); in RematerializeComputation()[all …]