Home
last modified time | relevance | path

Searched refs:stats_aggregator (Results 1 – 25 of 42) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/data/
Dfilter_dataset_op.cc131 auto stats_aggregator = ctx->stats_aggregator(); in GetNextInternal() local
166 if (stats_aggregator) { in GetNextInternal()
169 stats_aggregator->AddScalar( in GetNextInternal()
173 stats_aggregator->IncrementCounter(dataset()->node_name(), in GetNextInternal()
181 if (stats_aggregator) { in GetNextInternal()
184 stats_aggregator->AddScalar( in GetNextInternal()
188 stats_aggregator->IncrementCounter(dataset()->node_name(), in GetNextInternal()
Dprefetch_dataset_op.cc169 const auto& stats_aggregator = ctx->stats_aggregator(); in GetNextInternal() local
212 if (stats_aggregator) { in GetNextInternal()
213 stats_aggregator->AddScalar( in GetNextInternal()
216 stats_aggregator->AddScalar( in GetNextInternal()
367 const auto& stats_aggregator = ctx->stats_aggregator(); in Consume() local
368 if (stats_aggregator) { in Consume()
370 stats_aggregator->AddToHistogram( in Consume()
375 stats_aggregator->AddScalar( in Consume()
378 stats_aggregator->AddScalar( in Consume()
Dcaptured_function.cc853 if (node || ctx->stats_aggregator()) { in Run()
874 if (ctx->stats_aggregator()) { in Run()
877 ctx->stats_aggregator()->AddToHistogram( in Run()
916 if (node || ctx->stats_aggregator()) { in RunWithBorrowedArgs()
936 if (ctx->stats_aggregator()) { in RunWithBorrowedArgs()
939 ctx->stats_aggregator()->AddToHistogram( in RunWithBorrowedArgs()
1018 if (node || ctx->stats_aggregator()) { in RunAsync()
1046 if (ctx->stats_aggregator()) { in RunAsync()
1050 ctx->stats_aggregator()->AddToHistogram( in RunAsync()
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dstats_options.py22 from tensorflow.python.data.experimental.ops import stats_aggregator
48 ty=(stats_aggregator.StatsAggregatorV2,
49 stats_aggregator.StatsAggregatorV1),
Dstats_ops.py29 def set_stats_aggregator(stats_aggregator, prefix="", counter_prefix=""): argument
46 dataset, stats_aggregator, prefix, counter_prefix)
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dstats_dataset_ops.cc121 auto stats_aggregator = ctx->stats_aggregator(); in GetNextInternal() local
122 if (stats_aggregator && !*end_of_sequence) { in GetNextInternal()
124 stats_aggregator->AddToHistogram( in GetNextInternal()
236 auto stats_aggregator = ctx->stats_aggregator(); in GetNextInternal() local
237 if (stats_aggregator && s.ok() && !*end_of_sequence) { in GetNextInternal()
243 stats_aggregator->AddToHistogram( in GetNextInternal()
Dset_stats_aggregator_dataset_op.cc34 std::shared_ptr<StatsAggregator> stats_aggregator, const string& tag, in StatsAggregatorWithTagAndPrefix() argument
36 : wrapped_(stats_aggregator), tag_(tag), prefix_(prefix) {} in StatsAggregatorWithTagAndPrefix()
193 params.stats_aggregator = std::shared_ptr<StatsAggregator>( in ContextWithAggregator()
194 new StatsAggregatorWithTagAndPrefix(resource->stats_aggregator(), in ContextWithAggregator()
Dmap_and_batch_dataset_op.cc382 const auto& stats_aggregator = ctx->stats_aggregator(); in CallCompleted() local
383 if (stats_aggregator) { in CallCompleted()
384 stats_aggregator->AddScalar( in CallCompleted()
574 const auto& stats_aggregator = ctx->stats_aggregator(); in RunnerThread() local
575 if (stats_aggregator) { in RunnerThread()
577 stats_aggregator->AddScalar( in RunnerThread()
Dparse_example_dataset_op.cc557 if (ctx->stats_aggregator()) { in EnsureThreadsStarted()
652 auto stats_aggregator = ctx->stats_aggregator(); in ParseExample() local
653 if (stats_aggregator) { in ParseExample()
691 if (stats_aggregator) { in ParseExample()
692 stats_aggregator->IncrementCounter( in ParseExample()
697 stats_aggregator->IncrementCounter(stats_utils::kFeaturesCount, in ParseExample()
700 stats_aggregator->IncrementCounter( in ParseExample()
704 stats_aggregator->AddToHistogram( in ParseExample()
708 stats_aggregator->AddToHistogram( in ParseExample()
833 ctx->stats_aggregator()->AddScalar( in StatsThread()
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dstats_dataset_ops_test.py26 from tensorflow.python.data.experimental.ops import stats_aggregator
44 aggregator = stats_aggregator.StatsAggregator()
68 aggregator = stats_aggregator.StatsAggregator()
88 aggregator = stats_aggregator.StatsAggregator()
130 aggregator = stats_aggregator.StatsAggregator()
154 aggregator = stats_aggregator.StatsAggregator()
182 aggregator = stats_aggregator.StatsAggregator()
215 aggregator = stats_aggregator.StatsAggregator()
239 aggregator = stats_aggregator.StatsAggregator()
258 aggregator = stats_aggregator.StatsAggregator()
[all …]
Dstats_dataset_test_base.py28 from tensorflow.python.data.experimental.ops import stats_aggregator
43 stats_aggregator._DEFAULT_MAX_QUEUE = 0 # pylint: disable=protected-access
44 stats_aggregator.StatsAggregator = stats_aggregator.StatsAggregatorV2
47 stats_aggregator.StatsAggregator = stats_aggregator.StatsAggregatorV1
269 if isinstance(aggregator, stats_aggregator.StatsAggregatorV1):
271 assert isinstance(aggregator, (stats_aggregator.StatsAggregatorV2))
280 aggregator = stats_aggregator.StatsAggregator()
/external/tensorflow/tensorflow/core/framework/
Dstats_aggregator.h80 StatsAggregatorResource(std::unique_ptr<StatsAggregator> stats_aggregator) in StatsAggregatorResource() argument
81 : stats_aggregator_(stats_aggregator.release()) {} in StatsAggregatorResource()
84 std::shared_ptr<StatsAggregator> stats_aggregator() const { in stats_aggregator() function
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dlatency_all_edges_test.py23 from tensorflow.python.data.experimental.ops import stats_aggregator
37 aggregator = stats_aggregator.StatsAggregator()
64 aggregator = stats_aggregator.StatsAggregator()
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Dstats_dataset_serialization_test.py23 from tensorflow.python.data.experimental.ops import stats_aggregator
100 aggregator = stats_aggregator.StatsAggregator()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DStatsAggregatorSetSummaryWriter.pbtxt4 name: "stats_aggregator"
DSetStatsAggregatorDataset.pbtxt8 name: "stats_aggregator"
DExperimentalSetStatsAggregatorDataset.pbtxt8 name: "stats_aggregator"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_StatsAggregatorSetSummaryWriter.pbtxt4 summary: "Set a summary_writer_interface to record statistics using given stats_aggregator."
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DStatsAggregatorSetSummaryWriter.pbtxt4 name: "stats_aggregator"
DExperimentalSetStatsAggregatorDataset.pbtxt8 name: "stats_aggregator"
DSetStatsAggregatorDataset.pbtxt8 name: "stats_aggregator"
/external/tensorflow/tensorflow/python/data/experimental/benchmarks/
Dparallel_interleave_benchmark.py22 from tensorflow.python.data.experimental.ops import stats_aggregator
117 aggregator = stats_aggregator.StatsAggregator()
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.data.experimental.-stats-aggregator.pbtxt3 …is_instance: "<class \'tensorflow.python.data.experimental.ops.stats_aggregator.StatsAggregatorV2\…
/external/tensorflow/tensorflow/python/data/benchmarks/
Dmap_benchmark.py23 from tensorflow.python.data.experimental.ops import stats_aggregator
80 aggregator = stats_aggregator.StatsAggregator()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.data.experimental.-stats-aggregator.pbtxt3 …is_instance: "<class \'tensorflow.python.data.experimental.ops.stats_aggregator.StatsAggregatorV1\…

12