Searched refs:FormatNodeMemory (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_show.cc | 137 string TFShow::FormatNodeMemory(ShowNode* node, int64 bytes, in FormatNodeMemory() function in tensorflow::tfprof::TFShow
174 info.push_back(FormatNodeMemory(node, node->proto().requested_bytes(), in FormatNode()
178 info.push_back(FormatNodeMemory(node, node->proto().peak_bytes(), in FormatNode()
182 info.push_back(FormatNodeMemory(node, node->proto().residual_bytes(), in FormatNode()
186 info.push_back(FormatNodeMemory(node, node->proto().output_bytes(), in FormatNode()
|
| D | tfprof_code.cc | 619 string TFCode::FormatNodeMemory(CodeNode* node, int64 bytes, in FormatNodeMemory() function in tensorflow::tfprof::TFCode
634 attrs.push_back(FormatNodeMemory(node, node->proto().requested_bytes(), in FormatNode()
638 attrs.push_back(FormatNodeMemory(node, node->proto().peak_bytes(), in FormatNode()
642 attrs.push_back(FormatNodeMemory(node, node->proto().residual_bytes(), in FormatNode()
646 attrs.push_back(FormatNodeMemory(node, node->proto().output_bytes(), in FormatNode()
|
| D | tfprof_code.h | 86 string FormatNodeMemory(CodeNode* node, int64 bytes, int64 total_bytes) const;
|
| D | tfprof_show.h | 71 string FormatNodeMemory(ShowNode* node, int64 bytes, int64 total_bytes) const;
|