Home
last modified time | relevance | path

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

12345

/external/tensorflow/tensorflow/python/debug/wrappers/
Dframework.py182 def __init__(self, fetches, feed_dict, run_options, run_metadata, argument
201 self.run_metadata = run_metadata
276 run_metadata=None, argument
295 if run_metadata is not None:
296 _check_type(run_metadata, config_pb2.RunMetadata)
297 self.run_metadata = run_metadata
394 run_metadata=None, argument
465 run_metadata=run_metadata)
469 OnRunStartRequest(fetches, feed_dict, options, run_metadata,
476 run_start_resp, fetches, feed_dict, options, run_metadata,
[all …]
/external/tensorflow/tensorflow/python/client/
Dtimeline_test.py44 run_metadata = config_pb2.RunMetadata()
50 run_metadata=run_metadata)
51 self.assertTrue(run_metadata.HasField('step_stats'))
52 tl = timeline.Timeline(run_metadata.step_stats)
60 run_metadata = config_pb2.RunMetadata()
66 sess.run(result, options=run_options, run_metadata=run_metadata)
67 self.assertTrue(run_metadata.HasField('step_stats'))
68 step_stats = run_metadata.step_stats
92 run_metadata = config_pb2.RunMetadata()
98 sess.run(result, options=run_options, run_metadata=run_metadata)
[all …]
Dsession_clusterspec_prop_test.py74 run_metadata = config_pb2.RunMetadata()
75 output = sess.run(const, options=run_options, run_metadata=run_metadata)
80 for dev_stats in run_metadata.step_stats.dev_stats
118 run_metadata = config_pb2.RunMetadata()
119 output = sess.run(const, options=run_options, run_metadata=run_metadata)
124 for dev_stats in run_metadata.step_stats.dev_stats
146 run_metadata = config_pb2.RunMetadata()
147 output = sess.run(const, options=run_options, run_metadata=run_metadata)
152 for dev_stats in run_metadata.step_stats.dev_stats
219 run_metadata = config_pb2.RunMetadata()
[all …]
Dsession_ref.cc133 std::vector<Tensor>* outputs, RunMetadata* run_metadata) { in RecordRun() argument
165 target_node_names, outputs, run_metadata); in RecordRun()
175 if (run_metadata) { in RecordRun()
176 *resp->mutable_metadata() = *run_metadata; in RecordRun()
330 RunMetadata* run_metadata) { in RecordRunCallable() argument
339 run_metadata); in RecordRunCallable()
342 if (run_metadata) { in RecordRunCallable()
343 *resp->mutable_metadata() = *run_metadata; in RecordRunCallable()
415 RunMetadata* run_metadata) { in Run() argument
417 target_node_names, outputs, run_metadata); in Run()
[all …]
Dsession.py63 def run(self, fetches, feed_dict=None, options=None, run_metadata=None): argument
859 def run(self, fetches, feed_dict=None, options=None, run_metadata=None): argument
965 run_metadata_ptr = tf_session.TF_NewBuffer() if run_metadata else None
970 if run_metadata:
972 run_metadata.ParseFromString(compat.as_bytes(proto_data))
1104 def _run(self, handle, fetches, feed_dict, options, run_metadata): argument
1192 feed_dict_tensor, options, run_metadata)
1267 run_metadata=None): argument
1271 run_metadata_ptr = tf_session.TF_NewBuffer() if run_metadata else None
1279 if run_metadata:
[all …]
/external/tensorflow/tensorflow/compiler/tests/
Dasync_comp_test.py29 def RunMetadataLabels(run_metadata): argument
32 for dev_stats in run_metadata.step_stats.dev_stats:
43 def MetadataHasXlaRunOp(run_metadata): argument
47 return InLabels(RunMetadataLabels(run_metadata), "_XlaRun")
64 run_metadata = config_pb2.RunMetadata()
68 run_metadata=run_metadata,
72 hasXlaRunOp = MetadataHasXlaRunOp(run_metadata)
78 run_metadata = config_pb2.RunMetadata()
82 run_metadata=run_metadata,
85 hasXlaRunOp = MetadataHasXlaRunOp(run_metadata)
Djit_test.py64 def RunMetadataLabels(run_metadata): argument
67 for dev_stats in run_metadata.step_stats.dev_stats:
78 def MetadataHasXlaRunOp(run_metadata): argument
82 return InLabels(RunMetadataLabels(run_metadata), "_XlaRun")
114 run_metadata = config_pb2.RunMetadata()
118 run_metadata)
122 self.assert_(MetadataHasXlaRunOp(run_metadata))
254 run_metadata = config_pb2.RunMetadata()
262 run_metadata=run_metadata,
269 self.assert_(MetadataHasXlaRunOp(run_metadata))
[all …]
Ddense_layer_test.py32 def GetRunMetadataLabels(run_metadata): argument
35 for dev_stats in run_metadata.step_stats.dev_stats:
74 run_metadata = config_pb2.RunMetadata()
78 run_metadata=run_metadata,
82 labels = GetRunMetadataLabels(run_metadata)
99 run_metadata = config_pb2.RunMetadata()
103 run_metadata=run_metadata,
107 labels = GetRunMetadataLabels(run_metadata)
122 run_metadata = config_pb2.RunMetadata()
126 run_metadata=run_metadata,
[all …]
/external/tensorflow/tensorflow/python/profiler/
Dpprof_profiler_test.py34 run_metadata = config_pb2.RunMetadata()
38 profiles = pprof_profiler.get_profiles(graph, run_metadata)
41 graph, run_metadata, output_dir)
46 run_metadata = config_pb2.RunMetadata()
54 profiles = pprof_profiler.get_profiles(graph, run_metadata)
57 graph, run_metadata, output_dir)
62 run_metadata = config_pb2.RunMetadata()
70 run_metadata = config_pb2.RunMetadata()
71 device1 = run_metadata.step_stats.dev_stats.add()
121 profiles = pprof_profiler.get_profiles(graph, run_metadata)
[all …]
Dmodel_analyzer_test.py168 run_metadata=run_meta)
195 run_metadata=run_meta)
283 run_metadata=run_meta)
311 run_metadata=run_meta)
346 run_metadata=run_meta)
393 run_metadata=run_meta)
426 run_metadata=run_meta)
489 run_metadata=run_meta)
555 run_metadata=run_meta)
717 run_metadata = config_pb2.RunMetadata()
[all …]
/external/tensorflow/tensorflow/python/debug/cli/
Dprofile_analyzer_cli_test.py74 run_metadata = config_pb2.RunMetadata()
76 prof_analyzer = profile_analyzer_cli.ProfileAnalyzer(graph, run_metadata)
93 run_metadata = config_pb2.RunMetadata()
94 device1 = run_metadata.step_stats.dev_stats.add()
109 prof_analyzer = profile_analyzer_cli.ProfileAnalyzer(graph, run_metadata)
123 run_metadata = config_pb2.RunMetadata()
124 device1 = run_metadata.step_stats.dev_stats.add()
128 device2 = run_metadata.step_stats.dev_stats.add()
139 prof_analyzer = profile_analyzer_cli.ProfileAnalyzer(graph, run_metadata)
153 run_metadata = config_pb2.RunMetadata()
[all …]
/external/tensorflow/tensorflow/core/grappler/costs/
Danalytical_cost_estimator.cc158 RunMetadata* run_metadata, in PredictCosts() argument
183 if (run_metadata) { in PredictCosts()
184 cost_graph = run_metadata->mutable_cost_graph(); in PredictCosts()
231 *costs = scheduler_->Summary(run_metadata); in PredictCosts()
232 } else if (run_metadata) { in PredictCosts()
233 scheduler_->GenerateRunMetadata(run_metadata); in PredictCosts()
238 if (run_metadata) { in PredictCosts()
239 VLOG(1) << GetStatsStringFromRunMetadata(*run_metadata, verbose); in PredictCosts()
241 RunMetadata run_metadata; in PredictCosts() local
242 scheduler_->GenerateRunMetadata(&run_metadata); in PredictCosts()
[all …]
/external/tensorflow/tensorflow/python/debug/lib/
Dsession_debug_testlib.py156 run_metadata = config_pb2.RunMetadata()
160 run_metadata=run_metadata)
164 len(run_metadata.partition_graphs))
166 self._dump_root, partition_graphs=run_metadata.partition_graphs,
200 run_metadata = config_pb2.RunMetadata()
203 sess.run(w, options=run_options, run_metadata=run_metadata)
206 len(run_metadata.partition_graphs))
209 self._dump_root, partition_graphs=run_metadata.partition_graphs)
236 run_metadata = config_pb2.RunMetadata()
237 r = sess.run(w, options=run_options, run_metadata=run_metadata)
[all …]
Ddebug_graph_reconstruction_test.py84 run_metadata = config_pb2.RunMetadata()
86 run_metadata=run_metadata)
89 non_debug_graph_defs = run_metadata.partition_graphs
93 run_metadata = config_pb2.RunMetadata()
95 run_metadata=run_metadata)
100 self._dump_dir, partition_graphs=run_metadata.partition_graphs,
114 run_metadata.partition_graphs[i]))
Dsession_debug_file_test.py83 run_metadata = config_pb2.RunMetadata()
86 sess.run(w, options=run_options, run_metadata=run_metadata)
89 len(run_metadata.partition_graphs))
92 run_dump_root, partition_graphs=run_metadata.partition_graphs)
Dsession_debug_multi_gpu_test.py71 run_metadata = config_pb2.RunMetadata()
74 sess.run(w, options=run_options, run_metadata=run_metadata))
77 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/cc/saved_model/
Dloader.cc178 std::vector<Tensor>* outputs, RunMetadata* run_metadata, in RunOnce() argument
199 outputs, run_metadata); in RunOnce()
217 RunMetadata run_metadata; in RunInitOp() local
219 nullptr /* outputs */, &run_metadata, session); in RunInitOp()
259 RunMetadata run_metadata; in RunRestore() local
261 nullptr /* outputs */, &run_metadata, session); in RunRestore()
374 std::vector<Tensor>* outputs, RunMetadata* run_metadata) override { in Run() argument
376 target_node_names, outputs, run_metadata); in Run()
407 RunMetadata* run_metadata) override { in RunCallable() argument
409 run_metadata); in RunCallable()
[all …]
/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/python/grappler/
Dlayout_optimizer_test.py286 output_val = sess.run(output, run_metadata=metadata)
324 output_val = sess.run(output, run_metadata=metadata, feed_dict={dim: 3})
357 output_val = sess.run(output, run_metadata=metadata, feed_dict={dim: 3})
392 output_val = sess.run(output, run_metadata=metadata)
424 output_val = sess.run(output, run_metadata=metadata)
454 output_val = sess.run(output, run_metadata=metadata)
486 output_val = sess.run(output, run_metadata=metadata)
517 output_val = sess.run(output, run_metadata=metadata)
548 output_val = sess.run(output, run_metadata=metadata)
578 output_val = sess.run(output, run_metadata=metadata)
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlift_variables_test_pass.h73 tensorflow::RunMetadata run_metadata; in Run() local
75 outputs, &run_metadata); in Run()
83 tensorflow::RunMetadata* run_metadata) override { in Run() argument
85 run_metadata, tensorflow::thread::ThreadPoolOptions()); in Run()
93 tensorflow::RunMetadata* run_metadata, in Run() argument
/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()

12345