Home
last modified time | relevance | path

Searched refs:run_metadata (Results 1 – 25 of 123) sorted by relevance

12345

/external/tensorflow/tensorflow/python/client/
Dtimeline_test.py48 run_metadata = config_pb2.RunMetadata()
54 run_metadata=run_metadata)
55 self.assertTrue(run_metadata.HasField('step_stats'))
56 tl = timeline.Timeline(run_metadata.step_stats)
64 run_metadata = config_pb2.RunMetadata()
70 sess.run(result, options=run_options, run_metadata=run_metadata)
71 self.assertTrue(run_metadata.HasField('step_stats'))
72 step_stats = run_metadata.step_stats
96 run_metadata = config_pb2.RunMetadata()
102 sess.run(result, options=run_options, run_metadata=run_metadata)
[all …]
Dsession_clusterspec_prop_test.py84 run_metadata = config_pb2.RunMetadata()
85 output = sess.run(const, options=run_options, run_metadata=run_metadata)
90 for dev_stats in run_metadata.step_stats.dev_stats
128 run_metadata = config_pb2.RunMetadata()
129 output = sess.run(const, options=run_options, run_metadata=run_metadata)
134 for dev_stats in run_metadata.step_stats.dev_stats
156 run_metadata = config_pb2.RunMetadata()
157 output = sess.run(const, options=run_options, run_metadata=run_metadata)
162 for dev_stats in run_metadata.step_stats.dev_stats
229 run_metadata = config_pb2.RunMetadata()
[all …]
Dsession.py56 def run(self, fetches, feed_dict=None, options=None, run_metadata=None): argument
820 def run(self, fetches, feed_dict=None, options=None, run_metadata=None): argument
926 run_metadata_ptr = tf_session.TF_NewBuffer() if run_metadata else None
931 if run_metadata:
933 run_metadata.ParseFromString(compat.as_bytes(proto_data))
1064 def _run(self, handle, fetches, feed_dict, options, run_metadata): argument
1153 feed_dict_tensor, options, run_metadata)
1226 run_metadata=None): argument
1230 run_metadata_ptr = tf_session.TF_NewBuffer() if run_metadata else None
1238 if run_metadata:
[all …]
Dsession_ref.cc131 std::vector<Tensor>* outputs, RunMetadata* run_metadata) { in RecordRun() argument
163 target_node_names, outputs, run_metadata); in RecordRun()
173 if (run_metadata) { in RecordRun()
174 *resp->mutable_metadata() = *run_metadata; in RecordRun()
328 RunMetadata* run_metadata) { in RecordRunCallable() argument
337 run_metadata); in RecordRunCallable()
340 if (run_metadata) { in RecordRunCallable()
341 *resp->mutable_metadata() = *run_metadata; in RecordRunCallable()
413 RunMetadata* run_metadata) { in Run() argument
415 target_node_names, outputs, run_metadata); in Run()
[all …]
/external/tensorflow/tensorflow/compiler/tests/
Djit_test.py67 def RunMetadataLabels(run_metadata): argument
70 for dev_stats in run_metadata.step_stats.dev_stats:
81 def MetadataHasXlaRunOp(run_metadata): argument
85 return InLabels(RunMetadataLabels(run_metadata), "_XlaRun")
111 run_metadata = config_pb2.RunMetadata()
115 run_metadata)
119 self.assert_(MetadataHasXlaRunOp(run_metadata))
252 run_metadata = config_pb2.RunMetadata()
260 run_metadata=run_metadata,
267 self.assert_(MetadataHasXlaRunOp(run_metadata))
[all …]
Ddense_layer_test.py34 def GetRunMetadataLabels(run_metadata): argument
37 for dev_stats in run_metadata.step_stats.dev_stats:
76 run_metadata = config_pb2.RunMetadata()
80 run_metadata=run_metadata,
84 labels = GetRunMetadataLabels(run_metadata)
101 run_metadata = config_pb2.RunMetadata()
105 run_metadata=run_metadata,
109 labels = GetRunMetadataLabels(run_metadata)
130 run_metadata = config_pb2.RunMetadata()
134 run_metadata=run_metadata,
[all …]
/external/tensorflow/tensorflow/python/debug/wrappers/
Dframework.py201 def __init__(self, fetches, feed_dict, run_options, run_metadata, argument
220 self.run_metadata = run_metadata
299 run_metadata=None, argument
318 if run_metadata is not None:
319 _check_type(run_metadata, config_pb2.RunMetadata)
320 self.run_metadata = run_metadata
421 run_metadata=None, argument
480 run_metadata=run_metadata)
484 OnRunStartRequest(fetches, feed_dict, options, run_metadata,
502 run_metadata = run_metadata or config_pb2.RunMetadata()
[all …]
/external/tensorflow/tensorflow/python/profiler/
Dpprof_profiler_test.py38 run_metadata = config_pb2.RunMetadata()
42 profiles = pprof_profiler.get_profiles(graph, run_metadata)
45 graph, run_metadata, output_dir)
50 run_metadata = config_pb2.RunMetadata()
58 profiles = pprof_profiler.get_profiles(graph, run_metadata)
61 graph, run_metadata, output_dir)
66 run_metadata = config_pb2.RunMetadata()
74 run_metadata = config_pb2.RunMetadata()
75 device1 = run_metadata.step_stats.dev_stats.add()
125 profiles = pprof_profiler.get_profiles(graph, run_metadata)
[all …]
Dmodel_analyzer_test.py169 run_metadata=run_meta)
195 run_metadata=run_meta)
291 run_metadata=run_meta)
320 run_metadata=run_meta)
355 run_metadata=run_meta)
402 run_metadata=run_meta)
435 run_metadata=run_meta)
500 run_metadata=run_meta)
565 run_metadata=run_meta)
724 run_metadata = config_pb2.RunMetadata()
[all …]
/external/tensorflow/tensorflow/core/profiler/internal/cpu/
Dhost_tracer_test.cc88 RunMetadata run_metadata; in TEST() local
89 TF_ASSERT_OK(tracer->CollectData(&run_metadata)); in TEST()
91 EXPECT_EQ(run_metadata.step_stats().dev_stats_size(), 1); in TEST()
92 EXPECT_EQ(run_metadata.step_stats().dev_stats(0).node_stats_size(), 6); in TEST()
94 run_metadata.step_stats().dev_stats(0).node_stats(), in TEST()
106 void ValidateResult(const RunMetadata& run_metadata, const string& trace_name) { in ValidateResult() argument
110 run_metadata.step_stats().dev_stats(0).node_stats(), in ValidateResult()
116 RunMetadata run_metadata; in TEST() local
121 TF_ASSERT_OK(tracer->CollectData(&run_metadata)); in TEST()
126 ValidateResult(run_metadata, "hello"); in TEST()
/external/tensorflow/tensorflow/python/debug/cli/
Dprofile_analyzer_cli_test.py78 run_metadata = config_pb2.RunMetadata()
80 prof_analyzer = profile_analyzer_cli.ProfileAnalyzer(graph, run_metadata)
97 run_metadata = config_pb2.RunMetadata()
98 device1 = run_metadata.step_stats.dev_stats.add()
113 prof_analyzer = profile_analyzer_cli.ProfileAnalyzer(graph, run_metadata)
127 run_metadata = config_pb2.RunMetadata()
128 device1 = run_metadata.step_stats.dev_stats.add()
132 device2 = run_metadata.step_stats.dev_stats.add()
143 prof_analyzer = profile_analyzer_cli.ProfileAnalyzer(graph, run_metadata)
157 run_metadata = config_pb2.RunMetadata()
[all …]
/external/tensorflow/tensorflow/core/profiler/lib/
Dprofiler_session.cc39 void AssignLanes(RunMetadata* run_metadata) { in AssignLanes() argument
41 device_id < run_metadata->step_stats().dev_stats_size(); ++device_id) { in AssignLanes()
43 run_metadata->mutable_step_stats()->mutable_dev_stats(device_id); in AssignLanes()
69 void ConvertRunMetadataToTraceEvent(RunMetadata* run_metadata, in ConvertRunMetadataToTraceEvent() argument
72 AssignLanes(run_metadata); in ConvertRunMetadataToTraceEvent()
76 device_id < run_metadata->step_stats().dev_stats_size(); ++device_id) { in ConvertRunMetadataToTraceEvent()
79 run_metadata->mutable_step_stats()->mutable_dev_stats(device_id); in ConvertRunMetadataToTraceEvent()
97 run_metadata->step_stats().dev_stats(device_id).node_stats()) { in ConvertRunMetadataToTraceEvent()
136 RunMetadata run_metadata; in SerializeToString() local
138 profiler->CollectData(&run_metadata).IgnoreError(); in SerializeToString()
[all …]
/external/tensorflow/tensorflow/core/grappler/costs/
Danalytical_cost_estimator.cc136 RunMetadata* run_metadata, in PredictCosts() argument
149 if (run_metadata) { in PredictCosts()
150 cost_graph = run_metadata->mutable_cost_graph(); in PredictCosts()
192 *costs = scheduler_->Summary(run_metadata); in PredictCosts()
196 if (run_metadata) { in PredictCosts()
197 VLOG(1) << GetStatsStringFromRunMetadata(*run_metadata, verbose); in PredictCosts()
199 RunMetadata run_metadata; in PredictCosts() local
200 scheduler_->GenerateRunMetadata(&run_metadata); in PredictCosts()
201 VLOG(1) << GetStatsStringFromRunMetadata(run_metadata, verbose); in PredictCosts()
/external/tensorflow/tensorflow/python/debug/lib/
Dsession_debug_testlib.py161 run_metadata = config_pb2.RunMetadata()
165 run_metadata=run_metadata)
169 len(run_metadata.partition_graphs))
171 self._dump_root, partition_graphs=run_metadata.partition_graphs,
205 run_metadata = config_pb2.RunMetadata()
208 sess.run(w, options=run_options, run_metadata=run_metadata)
211 len(run_metadata.partition_graphs))
214 self._dump_root, partition_graphs=run_metadata.partition_graphs)
241 run_metadata = config_pb2.RunMetadata()
242 r = sess.run(w, options=run_options, run_metadata=run_metadata)
[all …]
Ddebug_graph_reconstruction_test.py89 run_metadata = config_pb2.RunMetadata()
91 run_metadata=run_metadata)
94 non_debug_graph_defs = run_metadata.partition_graphs
98 run_metadata = config_pb2.RunMetadata()
100 run_metadata=run_metadata)
105 self._dump_dir, partition_graphs=run_metadata.partition_graphs,
119 run_metadata.partition_graphs[i]))
Dsession_debug_file_test.py87 run_metadata = config_pb2.RunMetadata()
90 sess.run(w, options=run_options, run_metadata=run_metadata)
93 len(run_metadata.partition_graphs))
96 run_dump_root, partition_graphs=run_metadata.partition_graphs)
Dsession_debug_multi_gpu_test.py75 run_metadata = config_pb2.RunMetadata()
78 sess.run(w, options=run_options, run_metadata=run_metadata))
81 self._dump_root, partition_graphs=run_metadata.partition_graphs)
/external/tensorflow/tensorflow/core/profiler/internal/runtime/
Deager_profiler.cc28 Status TraceCollector::CollectData(RunMetadata* run_metadata) { in CollectData() argument
29 run_metadata->MergeFrom(run_metadata_); in CollectData()
52 Status EagerProfiler::CollectData(RunMetadata* run_metadata) { in CollectData() argument
53 return collector_.CollectData(run_metadata); in CollectData()
/external/tensorflow/tensorflow/core/profiler/g3doc/
Dpython_api.md60 run_metadata = tf.RunMetadata()
64 run_metadata=run_metadata)
79 run_meta=run_metadata,
87 run_meta=run_metadata,
128 run_metadata=run_meta)
138 run_metadata=run_meta2)
147 run_metadata=run_meta3)
/external/tensorflow/tensorflow/contrib/stat_summarizer/python/
Dstat_summarizer_test.py45 run_metadata = config_pb2.RunMetadata()
48 sess.run(product, options=run_options, run_metadata=run_metadata)
50 ss.ProcessStepStatsStr(run_metadata.step_stats.SerializeToString())
/external/tensorflow/tensorflow/contrib/android/jni/
Drun_stats_jni.cc56 jbyteArray run_metadata) { in RUN_STATS_METHOD()
59 jbyte* data = env->GetByteArrayElements(run_metadata, nullptr); in RUN_STATS_METHOD()
60 int size = static_cast<int>(env->GetArrayLength(run_metadata)); in RUN_STATS_METHOD()
69 env->ReleaseByteArrayElements(run_metadata, data, JNI_ABORT); in RUN_STATS_METHOD()
/external/tensorflow/tensorflow/python/grappler/
Dlayout_optimizer_test.py269 output_val = sess.run(output, run_metadata=metadata)
307 output_val = sess.run(output, run_metadata=metadata, feed_dict={dim: 3})
340 output_val = sess.run(output, run_metadata=metadata, feed_dict={dim: 3})
375 output_val = sess.run(output, run_metadata=metadata)
407 output_val = sess.run(output, run_metadata=metadata)
437 output_val = sess.run(output, run_metadata=metadata)
469 output_val = sess.run(output, run_metadata=metadata)
500 output_val = sess.run(output, run_metadata=metadata)
531 output_val = sess.run(output, run_metadata=metadata)
561 output_val = sess.run(output, run_metadata=metadata)
[all …]
/external/tensorflow/tensorflow/cc/framework/
Dwhile_gradients_test.cc65 RunMetadata* run_metadata = nullptr) { in Run() argument
75 &out_tensors, run_metadata)); in Run()
221 RunMetadata run_metadata; in TEST_F() local
223 &run_metadata); in TEST_F()
226 ASSERT_EQ(run_metadata.partition_graphs().size(), 2); in TEST_F()
227 for (const GraphDef& partition_graph : run_metadata.partition_graphs()) { in TEST_F()
/external/tensorflow/tensorflow/java/src/main/native/
Dsession_jni.cc150 unique_tf_buffer run_metadata( in Java_org_tensorflow_Session_run() local
179 static_cast<int>(ntargets), run_metadata.get(), status); in Java_org_tensorflow_Session_run()
196 if (run_metadata != nullptr) { in Java_org_tensorflow_Session_run()
197 ret = env->NewByteArray(run_metadata->length); in Java_org_tensorflow_Session_run()
198 env->SetByteArrayRegion(ret, 0, run_metadata->length, in Java_org_tensorflow_Session_run()
199 reinterpret_cast<const jbyte*>(run_metadata->data)); in Java_org_tensorflow_Session_run()
/external/tensorflow/tensorflow/examples/tutorials/mnist/
Dmnist_softmax_xla.py66 run_metadata = tf.RunMetadata()
81 run_metadata=run_metadata)
82 trace = timeline.Timeline(step_stats=run_metadata.step_stats)

12345