Home
last modified time | relevance | path

Searched refs:RecordStep (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dlog_memory.h62 static void RecordStep(int64 step_id, const string& handle);
Dlog_memory.cc41 void LogMemory::RecordStep(const int64 step_id, const string& handle) { in RecordStep() function in tensorflow::LogMemory
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session.cc785 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()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.cc498 LogMemory::RecordStep(args.step_id, handle); in StartParallelExecutors()