Home
last modified time | relevance | path

Searched refs:tf_stats (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/profiler/internal/
Dprint_model_analysis.cc39 TFStats* tf_stats) { in RunProfile() argument
46 tf_stats->BuildAllViews(); in RunProfile()
47 return Advisor(tf_stats).Advise(option_pb).SerializeAsString(); in RunProfile()
49 tf_stats->BuildView(command); in RunProfile()
65 ret = tf_stats->ShowMultiGraphNode(command, opts).SerializeAsString(); in RunProfile()
67 ret = tf_stats->ShowGraphNode(command, opts).SerializeAsString(); in RunProfile()
76 return tf_stats->ShowMultiGraphNode(command, opts).SerializeAsString(); in RunProfile()
78 return tf_stats->ShowGraphNode(command, opts).SerializeAsString(); in RunProfile()
197 TFStats tf_stats(std::move(graph_ptr), std::move(run_meta_ptr), in PrintModelAnalysis() local
200 return RunProfile(*command, *options, &tf_stats); in PrintModelAnalysis()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_context.cc62 tensorflow::AllocatorStats tf_stats; in GetStats() local
63 tf_stats.num_allocs = se_stats->num_allocs; in GetStats()
64 tf_stats.bytes_in_use = se_stats->bytes_in_use; in GetStats()
65 tf_stats.peak_bytes_in_use = se_stats->peak_bytes_in_use; in GetStats()
66 tf_stats.largest_alloc_size = se_stats->largest_alloc_size; in GetStats()
67 tf_stats.bytes_limit = se_stats->bytes_limit; in GetStats()
68 return tf_stats; in GetStats()