Searched refs:BeginInstruction (Results 1 – 1 of 1) sorted by relevance
339 Status BeginInstruction(Item* item);570 Status MemoryUsageTracker::BeginInstruction(Item* item) { in BeginInstruction() function in xla::__anon6c3ee80d0111::MemoryUsageTracker944 TF_RETURN_IF_ERROR(tracker.BeginInstruction(item)); in ComputePeakMemory()1016 TF_RETURN_IF_ERROR(memory_tracker.BeginInstruction(item)); in RematerializeComputation()