Searched refs:set_step_id (Results 1 – 12 of 12) sorted by relevance
42 step.set_step_id(step_id); in RecordStep()51 allocation.set_step_id(step_id); in RecordTensorAllocation()69 output.set_step_id(step_id); in RecordTensorOutput()80 allocation.set_step_id(step_id); in RecordRawAllocation()93 deallocation.set_step_id(step_id); in RecordRawDeallocation()
103 metadata.set_step_id(kInvalidStepId); in GenerateMemoryProfile()144 metadata.set_step_id(stat.IntValue()); in GenerateMemoryProfile()185 snapshot.mutable_activity_metadata()->set_step_id(last_valid_step_id + 1); in UpdateStepId()283 special_allocation->set_step_id(step_id); in InsertSpecialAllocations()300 special_allocation->set_step_id(step_id); in InsertSpecialAllocations()
323 labeled_step_stats->set_step_id(step_id); in RetrieveLogs()336 labeled_step_stats->set_step_id(step_id); in RetrieveLogs()
315 void InMemoryRunGraphRequest::set_step_id(int64 step_id) { step_id_ = step_id; } in set_step_id() function in tensorflow::InMemoryRunGraphRequest406 proto_version_->set_step_id(step_id()); in ToProto()454 void MutableProtoRunGraphRequest::set_step_id(int64 step_id) { in set_step_id() function in tensorflow::MutableProtoRunGraphRequest455 request_.set_step_id(step_id); in set_step_id()
279 req->set_step_id(opts.step_id); in Run()379 cleanup_req->set_step_id(step_id); in CleanUp()
312 virtual void set_step_id(int64 step_id) = 0;354 void set_step_id(int64 step_id) override;414 void set_step_id(int64 step_id) override;
45 req_.set_step_id(step_id); in RecvBufCall()
75 run_graph_request->set_step_id(13); in BuildRunGraphRequest()
685 c->req->set_step_id(step_id); in RunPartitionsHelper()879 req_.set_step_id(step_id); in CleanupBroadcastHelper()
70 req_.set_step_id(step_id); in Init()
276 cleanup_function->set_step_id(step_id); in CleanUp()
1129 mc->set_step_id(c->step_id); in RecordMemoryMapInternal()