Searched refs:stats_collector (Results 1 – 11 of 11) sorted by relevance
104 opts->stats_collector = ctx->stats_collector(); in SetRunOptions()
248 opts.stats_collector = ctx->stats_collector(); in ComputeAsync()
555 args.stats_collector = run_state.collector.get(); in Run()623 TF_RETURN_IF_ERROR(tracer->Collect(args.stats_collector)); in Run()673 if (args.stats_collector) { in Run()674 args.stats_collector->Finalize(); in Run()688 args.stats_collector->BuildCostModel(&cost_model_manager_, device_to_graph); in Run()781 args.stats_collector = run_state->collector.get(); in PRunSetup()
302 opts.stats_collector = ctx->stats_collector(); in ComputeAsync()779 exec_args->stats_collector = run_opts.stats_collector; in Run()865 exec_args->stats_collector = run_opts.stats_collector; in Run()
86 StepStatsCollector* stats_collector = nullptr; member
757 StepStatsCollector stats_collector(&stats); in TEST_F() local759 opts.stats_collector = &stats_collector; in TEST_F()767 stats_collector.FinalizeAndSwap(&stats); in TEST_F()
1301 stats_collector_(args.stats_collector), in ExecutorState()1520 params.stats_collector = stats_collector_; in Process()
564 StepStatsCollector* stats_collector = nullptr; member958 StepStatsCollector* stats_collector() const { in stats_collector() function959 return params_->stats_collector; in stats_collector()
481 StepStatsCollector* stats_collector = nullptr; member
474 args.stats_collector = collector; in StartParallelExecutors()
80 args.stats_collector = &step_stats_collector_; in Run()