Searched refs:mutable_step_stats (Results 1 – 12 of 12) 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()
178 labeled_step_stats->mutable_step_stats()->Swap(&step_stats); in RetrieveLogs()191 labeled_step_stats->mutable_step_stats()->Swap(&step_stats); in RetrieveLogs()
479 virtual StepStats* mutable_step_stats() = 0;514 StepStats* mutable_step_stats() override;547 StepStats* mutable_step_stats() override;574 StepStats* mutable_step_stats() override;
616 StepStats* InMemoryRunGraphResponse::mutable_step_stats() { in mutable_step_stats() function in tensorflow::InMemoryRunGraphResponse684 StepStats* OwnedProtoRunGraphResponse::mutable_step_stats() { in mutable_step_stats() function in tensorflow::OwnedProtoRunGraphResponse685 return response_.mutable_step_stats(); in mutable_step_stats()755 StepStats* NonOwnedProtoRunGraphResponse::mutable_step_stats() { in mutable_step_stats() function in tensorflow::NonOwnedProtoRunGraphResponse756 return response_->mutable_step_stats(); in mutable_step_stats()
183 collector = new StepStatsCollector(response->mutable_step_stats()); in DoRunGraph()
729 pss->step_stats[i].Swap(run_graph_resp->mutable_step_stats()); in RunPartitionsHelper()923 resp->mutable_step_stats()->Swap(&step_stats_proto); in ProcessStats()
43 run_metadata->mutable_step_stats()->mutable_dev_stats(device_id); in AssignLanes()79 run_metadata->mutable_step_stats()->mutable_dev_stats(device_id); in ConvertRunMetadataToTraceEvent()
49 absl::make_unique<StepStatsCollector>(run_metadata->mutable_step_stats()); in CollectData()
61 absl::make_unique<StepStatsCollector>(run_metadata->mutable_step_stats()); in CollectData()
145 auto* step_stats = run_metadata->mutable_step_stats(); in MaybeCopyInputToExpectedDevice()486 maybe_step_stats = ctx->RunMetadataProto()->mutable_step_stats(); in EagerLocalExecute()987 auto* step_stats = ctx->RunMetadataProto()->mutable_step_stats(); in EagerKernelExecute()
1029 StepStats* stepstats = metadata->mutable_step_stats(); in GenerateRunMetadata()
558 new StepStatsCollector(run_metadata->mutable_step_stats())); in RunInternal()