Home
last modified time | relevance | path

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

12345

/external/tensorflow/tensorflow/compiler/tests/
Djit_test.py68 def RunMetadataLabels(run_metadata): argument
71 for dev_stats in run_metadata.step_stats.dev_stats:
82 def MetadataHasXlaRunOp(run_metadata): argument
86 return InLabels(RunMetadataLabels(run_metadata), "_XlaRun")
118 run_metadata = config_pb2.RunMetadata()
122 run_metadata)
126 self.assert_(MetadataHasXlaRunOp(run_metadata))
258 run_metadata = config_pb2.RunMetadata()
266 run_metadata=run_metadata,
273 self.assert_(MetadataHasXlaRunOp(run_metadata))
[all …]
Ddense_layer_test.py35 def GetRunMetadataLabels(run_metadata): argument
38 for dev_stats in run_metadata.step_stats.dev_stats:
77 run_metadata = config_pb2.RunMetadata()
81 run_metadata=run_metadata,
85 labels = GetRunMetadataLabels(run_metadata)
102 run_metadata = config_pb2.RunMetadata()
106 run_metadata=run_metadata,
110 labels = GetRunMetadataLabels(run_metadata)
125 run_metadata = config_pb2.RunMetadata()
129 run_metadata=run_metadata,
[all …]
/external/tensorflow/tensorflow/python/debug/wrappers/
Dframework.py188 def __init__(self, fetches, feed_dict, run_options, run_metadata, argument
207 self.run_metadata = run_metadata
282 run_metadata=None, argument
301 if run_metadata is not None:
302 _check_type(run_metadata, config_pb2.RunMetadata)
303 self.run_metadata = run_metadata
401 run_metadata=None, argument
472 run_metadata=run_metadata)
476 OnRunStartRequest(fetches, feed_dict, options, run_metadata,
483 run_start_resp, fetches, feed_dict, options, run_metadata,
[all …]
/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.py78 run_metadata = config_pb2.RunMetadata()
79 output = sess.run(const, options=run_options, run_metadata=run_metadata)
84 for dev_stats in run_metadata.step_stats.dev_stats
122 run_metadata = config_pb2.RunMetadata()
123 output = sess.run(const, options=run_options, run_metadata=run_metadata)
128 for dev_stats in run_metadata.step_stats.dev_stats
150 run_metadata = config_pb2.RunMetadata()
151 output = sess.run(const, options=run_options, run_metadata=run_metadata)
156 for dev_stats in run_metadata.step_stats.dev_stats
223 run_metadata = config_pb2.RunMetadata()
[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 …]
Dsession.py64 def run(self, fetches, feed_dict=None, options=None, run_metadata=None): argument
850 def run(self, fetches, feed_dict=None, options=None, run_metadata=None): argument
956 run_metadata_ptr = tf_session.TF_NewBuffer() if run_metadata else None
961 if run_metadata:
963 run_metadata.ParseFromString(compat.as_bytes(proto_data))
1094 def _run(self, handle, fetches, feed_dict, options, run_metadata): argument
1183 feed_dict_tensor, options, run_metadata)
1257 run_metadata=None): argument
1261 run_metadata_ptr = tf_session.TF_NewBuffer() if run_metadata else None
1269 if run_metadata:
[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.py167 run_metadata=run_meta)
193 run_metadata=run_meta)
289 run_metadata=run_meta)
318 run_metadata=run_meta)
353 run_metadata=run_meta)
400 run_metadata=run_meta)
433 run_metadata=run_meta)
498 run_metadata=run_meta)
563 run_metadata=run_meta)
724 run_metadata = config_pb2.RunMetadata()
[all …]
/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/grappler/costs/
Danalytical_cost_estimator.cc152 RunMetadata* run_metadata, in PredictCosts() argument
177 if (run_metadata) { in PredictCosts()
178 cost_graph = run_metadata->mutable_cost_graph(); in PredictCosts()
221 *costs = scheduler_->Summary(run_metadata); in PredictCosts()
222 } else if (run_metadata) { in PredictCosts()
223 scheduler_->GenerateRunMetadata(run_metadata); in PredictCosts()
228 if (run_metadata) { in PredictCosts()
229 VLOG(1) << GetStatsStringFromRunMetadata(*run_metadata, verbose); in PredictCosts()
231 RunMetadata run_metadata; in PredictCosts() local
232 scheduler_->GenerateRunMetadata(&run_metadata); in PredictCosts()
[all …]
/external/tensorflow/tensorflow/core/profiler/internal/gpu/
Ddevice_tracer_test.cc137 RunMetadata run_metadata; in TEST_F() local
138 TF_EXPECT_OK(tracer->CollectData(&run_metadata)); in TEST_F()
139 EXPECT_EQ(run_metadata.step_stats().dev_stats_size(), 0); in TEST_F()
147 RunMetadata run_metadata; in TEST_F() local
148 Status status = tracer->CollectData(&run_metadata); in TEST_F()
217 RunMetadata run_metadata; in TEST_F() local
218 TF_ASSERT_OK(tracer->CollectData(&run_metadata)); in TEST_F()
221 EXPECT_GE(run_metadata.step_stats().dev_stats_size(), 1) in TEST_F()
222 << "Saw stats: " << run_metadata.DebugString(); in TEST_F()
240 RunMetadata run_metadata; in TEST_F() local
[all …]
/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/cc/client/
Dclient_session.cc109 RunMetadata* run_metadata) const { in Run()
127 target_node_names, outputs, run_metadata); in Run()
134 RunMetadata* run_metadata, in Run() argument
153 target_node_names, outputs, run_metadata, in Run()
166 RunMetadata* run_metadata) { in RunCallable() argument
168 run_metadata); in RunCallable()
174 RunMetadata* run_metadata, in RunCallable() argument
177 run_metadata, options); in RunCallable()
Dclient_session.h94 std::vector<Tensor>* outputs, RunMetadata* run_metadata) const;
101 std::vector<Tensor>* outputs, RunMetadata* run_metadata,
125 RunMetadata* run_metadata);
138 RunMetadata* run_metadata,
/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/tools/android/inference_interface/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/cc/saved_model/
Dloader.cc117 std::vector<Tensor>* outputs, RunMetadata* run_metadata, in RunOnce() argument
138 outputs, run_metadata); in RunOnce()
156 RunMetadata run_metadata; in RunInitOp() local
158 nullptr /* outputs */, &run_metadata, session); in RunInitOp()
230 RunMetadata run_metadata; in RunRestore() local
232 nullptr /* outputs */, &run_metadata, session); in RunRestore()
402 std::vector<Tensor>* outputs, RunMetadata* run_metadata) override { in Run() argument
404 target_node_names, outputs, run_metadata); in Run()
435 RunMetadata* run_metadata) override { in RunCallable() argument
437 run_metadata); in RunCallable()
[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/core/util/
Dstat_summarizer_test.cc68 RunMetadata run_metadata; in TEST() local
71 &run_metadata)); in TEST()
74 stats.ProcessStepStats(run_metadata.step_stats()); in TEST()
/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)
/external/tensorflow/tensorflow/core/profiler/internal/cpu/
Dhost_tracer_test.cc91 RunMetadata run_metadata; in TEST() local
92 TF_ASSERT_OK(tracer->CollectData(&run_metadata)); in TEST()
94 EXPECT_EQ(run_metadata.step_stats().dev_stats_size(), 1); in TEST()
95 EXPECT_EQ(run_metadata.step_stats().dev_stats(0).node_stats_size(), 6); in TEST()
97 run_metadata.step_stats().dev_stats(0).node_stats(), in TEST()

12345