Searched refs:tf_stats (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | print_model_analysis.cc | 41 TFStats* tf_stats) { in RunProfile() argument
48 tf_stats->BuildAllViews(); in RunProfile()
49 return Advisor(tf_stats).Advise(option_pb).SerializeAsString(); in RunProfile()
51 tf_stats->BuildView(command); in RunProfile()
69 ret = tf_stats->ShowMultiGraphNode(command, opts).SerializeAsString(); in RunProfile()
71 ret = tf_stats->ShowGraphNode(command, opts).SerializeAsString(); in RunProfile()
81 return tf_stats->ShowMultiGraphNode(command, opts).SerializeAsString(); in RunProfile()
83 return tf_stats->ShowGraphNode(command, opts).SerializeAsString(); in RunProfile()
202 TFStats tf_stats(std::move(graph_ptr), std::move(run_meta_ptr), in PrintModelAnalysis() local
205 return RunProfile(*command, *options, &tf_stats); in PrintModelAnalysis()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 63 tensorflow::AllocatorStats tf_stats; in GetStats() local
64 tf_stats.num_allocs = se_stats->num_allocs; in GetStats()
65 tf_stats.bytes_in_use = se_stats->bytes_in_use; in GetStats()
66 tf_stats.peak_bytes_in_use = se_stats->peak_bytes_in_use; in GetStats()
67 tf_stats.largest_alloc_size = se_stats->largest_alloc_size; in GetStats()
68 tf_stats.bytes_limit = se_stats->bytes_limit; in GetStats()
69 tf_stats.bytes_reserved = se_stats->bytes_reserved; in GetStats()
70 tf_stats.peak_bytes_reserved = se_stats->peak_bytes_reserved; in GetStats()
71 tf_stats.bytes_reservable_limit = se_stats->bytes_reservable_limit; in GetStats()
72 tf_stats.largest_free_block_bytes = se_stats->largest_free_block_bytes; in GetStats()
[all …]
|
| /external/tensorflow/tensorflow/core/profiler/convert/ |
| D | op_stats_to_tf_stats_test.cc | 128 const TfStatsDatabase tf_stats = ConvertOpStatsToTfStats(op_stats); in TEST() local
130 EXPECT_EQ(tf_stats.device_type(), op_stats.run_environment().device_type()); in TEST()
133 EXPECT_EQ(4, tf_stats.with_idle().tf_stats_record_size()); in TEST()
135 const TfStatsRecord& record_0 = tf_stats.with_idle().tf_stats_record(0); in TEST()
143 const TfStatsRecord& record_1 = tf_stats.with_idle().tf_stats_record(1); in TEST()
154 const TfStatsRecord& record_2 = tf_stats.with_idle().tf_stats_record(2); in TEST()
|
| /external/tensorflow/tensorflow/core/profiler/protobuf/ |
| D | BUILD | 140 srcs = ["tf_stats.proto"],
|