Searched refs:EndInstruction (Results 1 – 1 of 1) sorted by relevance
344 Status EndInstruction();597 Status MemoryUsageTracker::EndInstruction() { in EndInstruction() function in xla::__anon6c3ee80d0111::MemoryUsageTracker949 TF_RETURN_IF_ERROR(tracker.EndInstruction()); in ComputePeakMemory()1167 TF_RETURN_IF_ERROR(memory_tracker.EndInstruction()); in RematerializeComputation()