Searched refs:node_stat (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/profiler/internal/advisor/ |
| D | tfprof_advisor_test.cc | 54 NodeExecStats node_stat; in CreateNode() local
55 node_stat.set_all_start_micros(start_miros); in CreateNode()
56 node_stat.set_op_end_rel_micros(end_rel_micros); in CreateNode()
58 node_stat); in CreateNode()
61 node_stat); in CreateNode()
64 node_stat); in CreateNode()
|
| /external/tensorflow/tensorflow/python/profiler/ |
| D | tfprof_logger.py | 42 for node_stat in dev_stat.node_stats:
43 if not node_stat.output:
46 op = graph.get_operation_by_name(node_stat.node_name)
50 if len(node_stat.output) != len(op.outputs):
53 for (i, node_stat_out) in enumerate(node_stat.output):
64 (node_stat.node_name, e))
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_stats.cc | 284 for (const NodeExecStats& node_stat : dev_stat.node_stats()) { in AddRunMeta() local
285 string name = node_stat.node_name(); in AddRunMeta()
287 auto split_pos = node_stat.node_name().find(":"); in AddRunMeta()
288 if (split_pos != node_stat.node_name().npos) { in AddRunMeta()
289 name = node_stat.node_name().substr(0, split_pos); in AddRunMeta()
297 nodes_map_.at(name)->AddStepStat(step, dev_stat.device(), node_stat); in AddRunMeta()
301 node->second->AddStepStat(step, dev_stat.device(), node_stat); in AddRunMeta()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | direct_session_with_tracking_alloc_test.cc | 358 for (const auto& node_stat : dev_stat.node_stats()) { in TEST() local
359 if (node_stat.node_name() == y->name()) { in TEST()
360 EXPECT_LT(0, node_stat.memory(0).total_bytes()); in TEST()
361 EXPECT_LT(0, node_stat.memory(0).live_bytes()); in TEST()
362 EXPECT_LT(0, node_stat.memory(0).peak_bytes()); in TEST()
|
| /external/tensorflow/tensorflow/python/profiler/internal/ |
| D | run_metadata_test.py | 56 for node_stat in dev_stat.node_stats:
57 nname = node_stat.node_name
61 ret[dev].append(node_stat)
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils.cc | 464 for (const auto& node_stat : dev_stat.node_stats()) { in GetStatsStringFromRunMetadata() local
465 for (const auto& node_output : node_stat.output()) { in GetStatsStringFromRunMetadata()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | control_flow_ops_py_test.py | 1675 for node_stat in run_metadata.step_stats.dev_stats[0].node_stats:
1676 for output in node_stat.output:
|