Searched refs:stats_collector (Results 1 – 17 of 17) sorted by relevance
438 std::shared_ptr<SimpleStepStatsCollector> stats_collector; in RunAsync() local440 stats_collector = absl::make_unique<SimpleStepStatsCollector>(); in RunAsync()442 f_opts.stats_collector = stats_collector.get(); in RunAsync()450 const std::shared_ptr<SimpleStepStatsCollector>& stats_collector, in RunAsync()467 {static_cast<float>(stats_collector->processing_time())}); in RunAsync()470 model->AddProcessingTime(prefix, stats_collector->processing_time()); in RunAsync()479 std::move(stats_collector), std::placeholders::_1); in RunAsync()
256 opts->stats_collector = ctx->stats_collector(); in SetRunOptions()
227 params.stats_collector = args.stats_collector; in RunAsync()
218 run_opts.stats_collector = ctx->stats_collector(); in RunFunction()
256 opts.stats_collector = ctx->stats_collector(); in ComputeAsync()
121 opts->stats_collector = ctx->stats_collector(); in SetRunOptions()
514 opts.stats_collector = last_task_context->stats_collector(); in ProcessFuncBatch()
87 StepStatsCollectorInterface* stats_collector = nullptr; member
1020 StepStatsCollector stats_collector(&stats); in TEST_F() local1022 opts.stats_collector = &stats_collector; in TEST_F()1030 stats_collector.FinalizeAndSwap(&stats); in TEST_F()1064 StepStatsCollector stats_collector(&stats); in TEST_F() local1066 opts.stats_collector = &stats_collector; in TEST_F()1074 stats_collector.FinalizeAndSwap(&stats); in TEST_F()
487 opts.stats_collector = ctx->stats_collector(); in ComputeAsync()937 exec_args->stats_collector = run_opts.stats_collector; in ExecutorArgsFromOptions()
80 args.stats_collector = &step_stats_collector_; in Run()
559 args.stats_collector = run_state.collector.get(); in RunInternal()902 args.stats_collector = run_state->collector.get(); in PRunSetup()
1388 stats_collector_(args.stats_collector), in ExecutorState()1643 params.stats_collector = stats_collector_; in Process()
260 params.stats_collector = step_stats_collector.get(); in Run()333 opts.stats_collector = step_stats_collector.get(); in Run()
669 StepStatsCollectorInterface* stats_collector = nullptr; member1096 StepStatsCollectorInterface* stats_collector() const { in stats_collector() function1097 return params_->stats_collector; in stats_collector()
621 StepStatsCollectorInterface* stats_collector = nullptr; member
494 args.stats_collector = collector; in StartParallelExecutors()