| /external/tensorflow/tensorflow/core/profiler/internal/ |
| tfprof_node.cc | 173 int64 peak_bytes = 0; local
177 peak_bytes += mem.peak_bytes();
185 peak_bytes += exec_mem.host_persistent_bytes() +
191 exec_mem.set_peak_bytes(peak_bytes);
|
| tfprof_node_show.cc | 41 mutable_proto()->set_peak_bytes(node->peak_bytes(step));
103 proto().peak_bytes());
159 mutable_proto()->set_peak_bytes(node->peak_bytes());
210 proto().peak_bytes());
|
| tfprof_node.h | 130 int64 peak_bytes() const { function in class:tensorflow::tfprof::ExecStep
131 int64 peak_bytes = 0; local
133 peak_bytes += exec.peak_bytes();
135 return peak_bytes;
572 int64 peak_bytes(int64 step) const { GRAPH_NODE_BYTES(peak); } function in class:tensorflow::tfprof::TFGraphNode
809 peak_bytes_ += node->peak_bytes(step);
845 int64 peak_bytes() const { return peak_bytes_; } function in class:tensorflow::tfprof::TFMultiGraphNode
|
| tfprof_op.cc | 237 node->proto().peak_bytes()));
|
| tfprof_show.cc | 178 info.push_back(FormatNodeMemory(node, node->proto().peak_bytes(),
|
| tfprof_code.cc | 221 sample_pb->mutable_value()->Add(gn->peak_bytes(node->node->step()));
638 attrs.push_back(FormatNodeMemory(node, node->proto().peak_bytes(),
|
| /external/tensorflow/tensorflow/python/profiler/ |
| profiler_test.py | 179 .select(['micros', 'bytes', 'peak_bytes',
206 self.assertEqual(n1.peak_bytes, 0)
216 self.assertGreater(n2.peak_bytes, 0)
|
| model_analyzer_test.py | 333 'peak_bytes', 'residual_bytes',
475 self.assertGreaterEqual(n.peak_bytes, mpb)
572 ['peak_bytes', 'residual_bytes', 'output_bytes']).build()
714 options['select'] = ('bytes', 'peak_bytes', 'output_bytes',
726 self.assertGreater(n.peak_bytes, 0)
729 self.assertEqual(n.peak_bytes, n2.peak_bytes)
763 options['select'] = ('bytes', 'peak_bytes', 'output_bytes',
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| direct_session_with_tracking_alloc_test.cc | 346 EXPECT_LT(0, node_stat.memory(0).peak_bytes());
|
| executor.cc | 86 int64 peak = all.peak_bytes();
[all...] |
