Searched refs:RecordStep (Results 1 – 4 of 4) sorted by relevance
62 static void RecordStep(int64 step_id, const string& handle);
41 void LogMemory::RecordStep(const int64 step_id, const string& handle) { in RecordStep() function in tensorflow::LogMemory
785 LogMemory::RecordStep(step_id, run_state_args.handle); in Run()896 LogMemory::RecordStep(args.step_id, run_state_args.handle); in PRunSetup()1835 LogMemory::RecordStep(step_id, run_state_args.handle); in RunCallable()
498 LogMemory::RecordStep(args.step_id, handle); in StartParallelExecutors()