Searched refs:set_step_id (Results 1 – 12 of 12) sorted by relevance
43 step.set_step_id(step_id); in RecordStep()52 allocation.set_step_id(step_id); in RecordTensorAllocation()70 output.set_step_id(step_id); in RecordTensorOutput()81 allocation.set_step_id(step_id); in RecordRawAllocation()94 deallocation.set_step_id(step_id); in RecordRawDeallocation()
177 labeled_step_stats->set_step_id(step_id); in RetrieveLogs()190 labeled_step_stats->set_step_id(step_id); in RetrieveLogs()
317 void InMemoryRunGraphRequest::set_step_id(int64 step_id) { step_id_ = step_id; } in set_step_id() function in tensorflow::InMemoryRunGraphRequest402 proto_version_->set_step_id(step_id()); in ToProto()447 void MutableProtoRunGraphRequest::set_step_id(int64 step_id) { in set_step_id() function in tensorflow::MutableProtoRunGraphRequest448 request_.set_step_id(step_id); in set_step_id()
307 virtual void set_step_id(int64 step_id) = 0;347 void set_step_id(int64 step_id) override;404 void set_step_id(int64 step_id) override;
43 req_.set_step_id(step_id); in RecvBufCall()
190 req->set_step_id(step_id); in Run()
75 run_graph_request->set_step_id(13); in BuildRunGraphRequest()
633 c->req->set_step_id(step_id); in RunPartitionsHelper()822 req_.set_step_id(step_id); in CleanupBroadcastHelper()
75 mRes_.set_step_id(step_id); in Init()107 req_.set_step_id(step_id); in Init()
44 req_.set_step_id(step_id); in RecvBufCall()
49 req_.set_step_id(step_id); in GdrRecvTensorCall()
69 req_.set_step_id(step_id); in Init()