Home
last modified time | relevance | path

Searched refs:node_stats (Results 1 – 25 of 42) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/common/stats/
Dnode-stats_test.cc34 NodeStats node_stats(learner_config, GradientStats(1e-8f, 1e-8f)); in TEST() local
35 EXPECT_EQ(0, node_stats.weight_contribution[0]); in TEST()
36 EXPECT_EQ(0, node_stats.gain); in TEST()
43 NodeStats node_stats(learner_config, GradientStats(7.32f, 1.63f)); in TEST() local
44 EXPECT_EQ(0, node_stats.weight_contribution[0]); in TEST()
45 EXPECT_EQ(0, node_stats.gain); in TEST()
52 NodeStats node_stats(learner_config, GradientStats(7.32f, 1.63f)); in TEST() local
53 EXPECT_EQ(0, node_stats.weight_contribution[0]); in TEST()
54 EXPECT_EQ(0, node_stats.gain); in TEST()
61 NodeStats node_stats(learner_config, GradientStats(7.32f, 1.63f)); in TEST() local
[all …]
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_memory.cc163 for (const auto& node_stats : dev_stats.node_stats()) { in InferFromTrace() local
164 node_placement[node_stats.node_name()] = dev_stats.device(); in InferFromTrace()
179 for (const auto& node_stats : dev_stats.node_stats()) { in InferFromTrace() local
180 for (int i = 0; i < node_stats.output_size(); ++i) { in InferFromTrace()
181 const auto& output = node_stats.output(i); in InferFromTrace()
184 node_stats.node_name(), i, &live_tensors, &device_tensors); in InferFromTrace()
192 Costs::MicroSeconds(node_stats.all_start_micros()); in InferFromTrace()
198 Costs::MicroSeconds(node_stats.all_start_micros() + in InferFromTrace()
199 node_stats.op_end_rel_micros())); in InferFromTrace()
202 auto it = node_map.find(node_stats.node_name()); in InferFromTrace()
[all …]
Dmeasuring_cost_estimator.cc96 for (const NodeExecStats& node_stats : stepstats.node_stats()) { in PredictCosts() local
98 node_stats.all_end_rel_micros() + node_stats.all_start_micros(); in PredictCosts()
Dvirtual_scheduler.cc1036 NodeExecStats* node_stats = device_stepstats->add_node_stats(); in GenerateRunMetadata() local
1040 NodeOutput* no = node_stats->add_output(); in GenerateRunMetadata()
1054 node_stats->set_timeline_label(node_def->op()); in GenerateRunMetadata()
1055 node_stats->set_node_name(node_def->name()); in GenerateRunMetadata()
1056 node_stats->set_op_start_rel_micros(0); in GenerateRunMetadata()
1057 node_stats->set_all_start_micros( in GenerateRunMetadata()
1059 node_stats->set_op_end_rel_micros( in GenerateRunMetadata()
1062 node_stats->set_all_end_rel_micros( in GenerateRunMetadata()
1065 auto* mem_stats = node_stats->mutable_memory_stats(); in GenerateRunMetadata()
/external/tensorflow/tensorflow/python/client/
Dtimeline.py404 for ns in device_stats.node_stats:
488 for node_stats in dev_stats.node_stats:
489 tid = node_stats.thread_id
490 node_name = node_stats.node_name
491 start_time = node_stats.all_start_micros
492 end_time = node_stats.all_start_micros + node_stats.all_end_rel_micros
493 for index, output in enumerate(node_stats.output):
522 for node_stats in dev_stats.node_stats:
523 tid = node_stats.thread_id
524 start_time = node_stats.all_start_micros
[all …]
Dsession_clusterspec_prop_test.py89 node_stats
91 for node_stats in dev_stats.node_stats
93 dev_stats.device and 'Const' == node_stats.node_name
133 node_stats
135 for node_stats in dev_stats.node_stats
137 dev_stats.device and 'Const' == node_stats.node_name
161 node_stats
163 for node_stats in dev_stats.node_stats
165 == dev_stats.device and 'Const' == node_stats.node_name
234 node_stats
[all …]
Dtimeline_test.py128 node_stats = dev_stats.node_stats.add()
129 node_stats.node_name = 'RecvTensor'
130 node_stats.all_start_micros = 12345
131 node_stats.op_end_rel_micros = 42
132 node_stats.timeline_label = ('[1024B] edge_160_conv2/biases/read from '
/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.cc329 for (const auto& node_stats : dev_stats.hardware_stats->node_stats()) { in BuildCostModel() local
330 string node_name = node_stats.node_name(); in BuildCostModel()
343 time + node_stats.op_end_rel_micros()); in BuildCostModel()
345 name_to_hw_node_stats.emplace(node_name, node_stats); in BuildCostModel()
351 const NodeExecStats& stats = dev_stats.regular_stats->node_stats(i); in BuildCostModel()
371 string node_name = dev_stats.regular_stats->node_stats(i).node_name(); in BuildCostModel()
394 NodeExecStatsWrapper* node_stats) { in Save() argument
395 if (!node_stats) return; in Save()
396 VLOG(1) << "Save dev " << device << " node stats " << node_stats->stats(); in Save()
404 delete node_stats; in Save()
[all …]
Dstep_stats_collector.h176 void Save(const string& device, NodeExecStatsWrapper* node_stats);
/external/tensorflow/tensorflow/python/profiler/
Dpprof_profiler.py378 for node_stats in device_step_stats.node_stats:
379 if node_stats.node_name == '_SOURCE' or node_stats.node_name == '_SINK':
382 node_stats,
383 node_to_op_type[node_stats.node_name],
384 node_to_traceback[node_stats.node_name])
Dpprof_profiler_test.py77 device1.node_stats.extend([node1])
Dtfprof_logger.py42 for node_stat in dev_stat.node_stats:
/external/tensorflow/tensorflow/python/debug/cli/
Dprofile_analyzer_cli.py464 for node_stats in device_step_stats.node_stats:
465 if node_stats.node_name == "_SOURCE" or node_stats.node_name == "_SINK":
469 node_stats,
470 node_to_file_path.get(node_stats.node_name, ""),
471 node_to_line_number.get(node_stats.node_name, 0),
472 node_to_func_name.get(node_stats.node_name, ""),
473 node_to_op_type.get(node_stats.node_name, ""))
Dprofile_analyzer_cli_test.py100 device1.node_stats.extend([node1, node2])
130 device1.node_stats.extend([node1])
134 device2.node_stats.extend([node1])
197 device1.node_stats.extend([node1, node2])
249 device1.node_stats.extend([node1, node2])
303 device1.node_stats.extend([node1, node2])
/external/tensorflow/tensorflow/compiler/tests/
Ddense_layer_test.py38 for node_stats in dev_stats.node_stats:
39 labels.append(node_stats.timeline_label)
Djit_test.py71 for node_stats in dev_stats.node_stats:
72 labels.append(node_stats.timeline_label)
/external/tensorflow/tensorflow/core/profiler/internal/cpu/
Dhost_tracer_test.cc94 run_metadata.step_stats().dev_stats(0).node_stats(), in TEST()
110 run_metadata.step_stats().dev_stats(0).node_stats(), in ValidateResult()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc150 auto* node_stats = dev_stats->add_node_stats(); in MaybeCopyInputToExpectedDevice() local
151 node_stats->set_node_name("_Send"); in MaybeCopyInputToExpectedDevice()
152 node_stats->set_all_start_micros(pre_time_nanos / in MaybeCopyInputToExpectedDevice()
154 node_stats->set_all_start_nanos(pre_time_nanos); in MaybeCopyInputToExpectedDevice()
156 node_stats->set_op_end_rel_micros((now_nanos - pre_time_nanos) / in MaybeCopyInputToExpectedDevice()
158 node_stats->set_op_end_rel_nanos(now_nanos - pre_time_nanos); in MaybeCopyInputToExpectedDevice()
159 node_stats->set_all_end_rel_micros((now_nanos - pre_time_nanos) / in MaybeCopyInputToExpectedDevice()
161 node_stats->set_all_end_rel_nanos(now_nanos - pre_time_nanos); in MaybeCopyInputToExpectedDevice()
/external/tensorflow/tensorflow/core/util/
Dstat_summarizer.cc135 step_stats.dev_stats(0).node_stats(0).all_start_micros(); in ProcessStepStats()
139 for (const auto& ns : ds.node_stats()) { in ProcessStepStats()
/external/tensorflow/tensorflow/core/grappler/clusters/
Dvirtual_cluster_test.cc97 for (const auto& node : dev_stat.node_stats()) { in TEST_F()
/external/tensorflow/tensorflow/core/framework/
Dstep_stats.proto79 repeated NodeExecStats node_stats = 2; field
/external/tensorflow/tensorflow/core/profiler/lib/
Dprofiler_session.cc97 run_metadata->step_stats().dev_stats(device_id).node_stats()) { in ConvertRunMetadataToTraceEvent()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmemory_optimizer.cc995 for (const auto& node_stats : dev_stats.node_stats()) { in IdentifySwappingCandidates() local
998 Costs::MicroSeconds(node_stats.all_start_micros() + in IdentifySwappingCandidates()
999 node_stats.op_end_rel_micros()); in IdentifySwappingCandidates()
1000 op_completion_times.emplace(node_stats.node_name(), exec_time); in IdentifySwappingCandidates()
/external/mesa3d/src/gallium/drivers/r600/sb/
Dsb_ir.cpp449 void container_node::collect_stats(node_stats& s) { in collect_stats()
512 void node_stats::dump() { in dump()
Dsb_if_conversion.cpp129 node_stats s; in check_and_convert()

12