Searched refs:mutable_step_stats (Results 1 – 10 of 10) sorted by relevance
105 run_graph_response->mutable_step_stats()->add_dev_stats()->set_device( in BuildRunGraphResponse()123 ASSERT_EQ(1, response->mutable_step_stats()->dev_stats_size()); in CheckRunGraphResponse()124 EXPECT_EQ("/cpu:0", response->mutable_step_stats()->dev_stats(0).device()); in CheckRunGraphResponse()139 *run_step_response->mutable_metadata()->mutable_step_stats() = in BuildRunStepResponse()140 *run_graph_response->mutable_step_stats(); in BuildRunStepResponse()
635 StepStats* InMemoryRunGraphResponse::mutable_step_stats() { in mutable_step_stats() function in tensorflow::InMemoryRunGraphResponse703 StepStats* OwnedProtoRunGraphResponse::mutable_step_stats() { in mutable_step_stats() function in tensorflow::OwnedProtoRunGraphResponse704 return response_.mutable_step_stats(); in mutable_step_stats()774 StepStats* NonOwnedProtoRunGraphResponse::mutable_step_stats() { in mutable_step_stats() function in tensorflow::NonOwnedProtoRunGraphResponse775 return response_->mutable_step_stats(); in mutable_step_stats()
491 virtual StepStats* mutable_step_stats() = 0;526 StepStats* mutable_step_stats() override;559 StepStats* mutable_step_stats() override;586 StepStats* mutable_step_stats() override;
324 labeled_step_stats->mutable_step_stats()->Swap(&step_stats); in RetrieveLogs()337 labeled_step_stats->mutable_step_stats()->Swap(&step_stats); in RetrieveLogs()
196 collector = new StepStatsCollector(response->mutable_step_stats()); in DoRunGraph()241 response->mutable_step_stats()->MergeFrom(run_metadata.step_stats()); in DoRunGraph()
786 pss->step_stats[i].Swap(run_graph_resp->mutable_step_stats()); in RunPartitionsHelper()980 resp->mutable_step_stats()->Swap(&step_stats_proto); in ProcessStats()
111 StepStats* step_stats = run_metadata->mutable_step_stats(); in CollectData()
210 run_metadata->mutable_step_stats()->add_dev_stats()->Swap(&dev_stats); in CollectData()
1198 StepStats* stepstats = metadata->mutable_step_stats(); in GenerateRunMetadata()
672 new StepStatsCollector(run_metadata->mutable_step_stats())); in RunInternal()