Searched refs:RecordStep (Results 1 – 4 of 4) sorted by relevance
62 static void RecordStep(int64 step_id, const std::string& handle);
40 void LogMemory::RecordStep(const int64 step_id, const string& handle) { in RecordStep() function in tensorflow::LogMemory
890 LogMemory::RecordStep(step_id, run_state_args.handle); in Run()1002 LogMemory::RecordStep(args.step_id, run_state_args.handle); in PRunSetup()1988 LogMemory::RecordStep(step_id, run_state_args.handle); in RunCallable()
539 LogMemory::RecordStep(args.step_id, handle); in StartParallelExecutors()