Searched refs:BeginInstruction (Results 1 – 1 of 1) sorted by relevance
507 Status BeginInstruction(Item* item);861 Status MemoryUsageTracker::BeginInstruction(Item* item) { in BeginInstruction() function in xla::__anon95ba67740111::MemoryUsageTracker1803 TF_RETURN_IF_ERROR(tracker.BeginInstruction(item)); in ComputePeakMemory()1880 TF_RETURN_IF_ERROR(memory_tracker.BeginInstruction(item)); in RematerializeComputation()