Home
last modified time | relevance | path

Searched refs:run_metadata (Results 1 – 25 of 124) 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.py36 def GetRunMetadataLabels(run_metadata): argument
39 for dev_stats in run_metadata.step_stats.dev_stats:
78 run_metadata = config_pb2.RunMetadata()
82 run_metadata=run_metadata,
86 labels = GetRunMetadataLabels(run_metadata)
103 run_metadata = config_pb2.RunMetadata()
107 run_metadata=run_metadata,
111 labels = GetRunMetadataLabels(run_metadata)
126 run_metadata = config_pb2.RunMetadata()
130 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.cc132 std::vector<Tensor>* outputs, RunMetadata* run_metadata) { in RecordRun() argument
164 target_node_names, outputs, run_metadata); in RecordRun()
174 if (run_metadata) { in RecordRun()
175 *resp->mutable_metadata() = *run_metadata; in RecordRun()
329 RunMetadata* run_metadata) { in RecordRunCallable() argument
338 run_metadata); in RecordRunCallable()
341 if (run_metadata) { in RecordRunCallable()
342 *resp->mutable_metadata() = *run_metadata; in RecordRunCallable()
414 RunMetadata* run_metadata) { in Run() argument
416 target_node_names, outputs, run_metadata); in Run()
[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/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.py88 run_metadata = config_pb2.RunMetadata()
90 run_metadata=run_metadata)
93 non_debug_graph_defs = run_metadata.partition_graphs
97 run_metadata = config_pb2.RunMetadata()
99 run_metadata=run_metadata)
104 self._dump_dir, partition_graphs=run_metadata.partition_graphs,
118 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/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()
146 RunMetadata run_metadata; in TEST_F() local
147 Status status = tracer->CollectData(&run_metadata); in TEST_F()
213 RunMetadata run_metadata; in TEST_F() local
214 TF_ASSERT_OK(tracer->CollectData(&run_metadata)); in TEST_F()
217 EXPECT_GE(run_metadata.step_stats().dev_stats_size(), 1) in TEST_F()
218 << "Saw stats: " << run_metadata.DebugString(); in TEST_F()
236 RunMetadata run_metadata; in TEST_F() local
[all …]
/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/cc/saved_model/
Dloader.cc155 std::vector<Tensor>* outputs, RunMetadata* run_metadata, in RunOnce() argument
176 outputs, run_metadata); in RunOnce()
194 RunMetadata run_metadata; in RunInitOp() local
196 nullptr /* outputs */, &run_metadata, session); in RunInitOp()
233 RunMetadata run_metadata; in RunRestore() local
235 nullptr /* outputs */, &run_metadata, session); in RunRestore()
346 std::vector<Tensor>* outputs, RunMetadata* run_metadata) override { in Run() argument
348 target_node_names, outputs, run_metadata); in Run()
379 RunMetadata* run_metadata) override { in RunCallable() argument
381 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/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/compiler/mlir/tensorflow/transforms/
Dlift_variables_test_pass.cc70 tensorflow::RunMetadata run_metadata; in Run() local
72 outputs, &run_metadata); in Run()
80 tensorflow::RunMetadata* run_metadata) override { in Run() argument
82 run_metadata, tensorflow::thread::ThreadPoolOptions()); in Run()
90 tensorflow::RunMetadata* run_metadata, in Run() argument
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/python/grappler/
Dlayout_optimizer_test.py290 output_val = sess.run(output, run_metadata=metadata)
328 output_val = sess.run(output, run_metadata=metadata, feed_dict={dim: 3})
361 output_val = sess.run(output, run_metadata=metadata, feed_dict={dim: 3})
396 output_val = sess.run(output, run_metadata=metadata)
428 output_val = sess.run(output, run_metadata=metadata)
458 output_val = sess.run(output, run_metadata=metadata)
490 output_val = sess.run(output, run_metadata=metadata)
521 output_val = sess.run(output, run_metadata=metadata)
552 output_val = sess.run(output, run_metadata=metadata)
582 output_val = sess.run(output, run_metadata=metadata)
[all …]
/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/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()

12345