Home
last modified time | relevance | path

Searched refs:run_options (Results 1 – 25 of 115) sorted by relevance

12345

/external/tensorflow/tensorflow/cc/saved_model/
Dloader_test.cc106 RunOptions run_options; in TEST_F() local
111 TF_ASSERT_OK(LoadSavedModel(session_options, run_options, export_dir, in TEST_F()
120 RunOptions run_options; in TEST_F() local
124 TF_ASSERT_OK(LoadSavedModel(session_options, run_options, export_dir, in TEST_F()
131 RunOptions run_options; in TEST_F() local
136 Status st = LoadSavedModel(session_options, run_options, export_dir, in TEST_F()
147 RunOptions run_options; in TEST_F() local
152 Status st = LoadSavedModel(session_options, run_options, export_dir, in TEST_F()
168 RunOptions run_options; in TEST_F() local
172 Status st = LoadSavedModel(session_options, run_options, export_dir, in TEST_F()
[all …]
Dloader.cc112 Status RunOnce(const RunOptions& run_options, in RunOnce() argument
120 *callable_options.mutable_run_options() = run_options; in RunOnce()
146 Status RunInitOp(const RunOptions& run_options, const string& export_dir, in RunInitOp() argument
155 return RunOnce(run_options, inputs, {}, {init_op_name}, in RunInitOp()
196 Status RunRestore(const RunOptions& run_options, const string& export_dir, in RunRestore() argument
229 return RunOnce(run_options, inputs, {}, {string(restore_op_name)}, in RunRestore()
260 const RunOptions& run_options, in LoadSavedModelInternal() argument
275 RunRestore(run_options, export_dir, in LoadSavedModelInternal()
288 TF_RETURN_IF_ERROR(RunInitOp(run_options, export_dir, bundle->meta_graph_def, in LoadSavedModelInternal()
302 const RunOptions& run_options, const string& export_dir, in LoadSavedModel() argument
[all …]
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc56 const ExecutableRunOptions& run_options, const Backend& backend) { in ValidateExecutionOptions() argument
81 if (run_options.stream() != nullptr) { in ValidateExecutionOptions()
82 if (!run_options.stream()->ok()) { in ValidateExecutionOptions()
88 run_options.stream()->parent()->platform(); in ValidateExecutionOptions()
97 if (run_options.device_ordinal() != -1) { in ValidateExecutionOptions()
106 int run_device_ordinal = run_options.device_ordinal(); in ValidateExecutionOptions()
108 run_device_ordinal = run_options.stream() != nullptr in ValidateExecutionOptions()
109 ? run_options.stream()->parent()->device_ordinal() in ValidateExecutionOptions()
129 if (!run_options.allocator()) { in ValidateExecutionOptions()
133 if (run_options.allocator()->platform() != backend.platform()) { in ValidateExecutionOptions()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc137 const ExecutableRunOptions* run_options, in ExecuteComputeFunction() argument
188 compute_function_(result_buffer, run_options, nullptr, buffer_pointers.data(), in ExecuteComputeFunction()
209 const ServiceExecutableRunOptions* run_options, in CreateResultShapedBuffer() argument
211 se::Stream* stream = run_options->stream(); in CreateResultShapedBuffer()
214 /*on_device_shape=*/result_shape(), run_options->allocator(), in CreateResultShapedBuffer()
268 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream() argument
273 ExecuteAsyncOnStreamImpl(run_options, arguments, hlo_execution_profile)); in ExecuteOnStream()
274 TF_RETURN_IF_ERROR(run_options->stream()->BlockHostUntilDone()); in ExecuteOnStream()
279 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream() argument
286 return ExecuteAsyncOnStreamImpl(run_options, arguments, nullptr); in ExecuteAsyncOnStream()
[all …]
Dcpu_runtime.cc109 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length, in __xla_cpu_runtime_AcquireInfeedBufferForDequeue() argument
112 run_options ? run_options->stream()->parent()->device_ordinal() : 0; in __xla_cpu_runtime_AcquireInfeedBufferForDequeue()
133 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length, in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue() argument
136 run_options ? run_options->stream()->parent()->device_ordinal() : 0; in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue()
152 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length, in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation() argument
155 run_options ? run_options->stream()->parent()->device_ordinal() : 0; in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation()
176 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length, in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation() argument
179 run_options ? run_options->stream()->parent()->device_ordinal() : 0; in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation()
Dcpu_executable.h59 const ServiceExecutableRunOptions* run_options,
64 const ServiceExecutableRunOptions* run_options,
95 const ServiceExecutableRunOptions* run_options,
120 Status ExecuteComputeFunction(const ExecutableRunOptions* run_options,
128 const ServiceExecutableRunOptions* run_options,
Dcpu_runtime.h107 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length,
123 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length,
129 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length,
142 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length,
Druntime_fork_join.cc74 const xla::ExecutableRunOptions* run_options = in __xla_cpu_runtime_ParallelForkJoin() local
76 CHECK_NE(run_options, nullptr); in __xla_cpu_runtime_ParallelForkJoin()
77 CHECK_NE(run_options->intra_op_thread_pool(), nullptr); in __xla_cpu_runtime_ParallelForkJoin()
88 run_options->intra_op_thread_pool()->enqueueNoNotification( in __xla_cpu_runtime_ParallelForkJoin()
Druntime_conv2d.cc35 const xla::ExecutableRunOptions* run_options = in __xla_cpu_runtime_EigenConvF32() local
38 *run_options->intra_op_thread_pool(), out, lhs, rhs, input_batch, in __xla_cpu_runtime_EigenConvF32()
54 const xla::ExecutableRunOptions* run_options = in __xla_cpu_runtime_EigenConvF16() local
57 *run_options->intra_op_thread_pool(), out, lhs, rhs, input_batch, in __xla_cpu_runtime_EigenConvF16()
/external/tensorflow/tensorflow/compiler/xla/service/
Dexecutable.cc33 absl::Span<const ServiceExecutableRunOptions> run_options, in ExecuteOnStreams() argument
35 TF_RET_CHECK(run_options.size() == arguments.size()); in ExecuteOnStreams()
38 return_values.reserve(run_options.size()); in ExecuteOnStreams()
40 if (run_options.size() == 1) { in ExecuteOnStreams()
42 ExecuteOnStream(&run_options[0], arguments[0], in ExecuteOnStreams()
48 for (size_t i = 0; i < run_options.size(); ++i) { in ExecuteOnStreams()
53 ExecuteAsyncOnStream(&run_options[i], arguments[i])); in ExecuteOnStreams()
56 for (const auto& options : run_options) { in ExecuteOnStreams()
64 const ServiceExecutableRunOptions* run_options, ExecutionProfile* profile, in ExecuteOnStreamWrapper() argument
66 se::Stream* stream = run_options->stream(); in ExecuteOnStreamWrapper()
[all …]
Dexecutable.h83 const ServiceExecutableRunOptions* run_options,
90 const ServiceExecutableRunOptions* run_options,
103 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream() argument
111 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream() argument
122 absl::Span<const ServiceExecutableRunOptions> run_options,
138 const ServiceExecutableRunOptions* run_options, ExecutionProfile* profile,
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_utils.py26 def add_debug_tensor_watch(run_options, argument
61 watch_opts = run_options.debug_options.debug_tensor_watch_opts
62 run_options.debug_options.global_step = global_step
82 def watch_graph(run_options, argument
168 run_options,
176 run_options.debug_options.reset_disk_byte_usage = reset_disk_byte_usage
179 def watch_graph_with_blacklists(run_options, argument
261 run_options,
269 run_options.debug_options.reset_disk_byte_usage = reset_disk_byte_usage
Dsession_debug_testlib.py153 run_options = config_pb2.RunOptions(output_partition_graphs=True)
155 run_options,
164 options=run_options,
195 run_options = config_pb2.RunOptions(output_partition_graphs=True)
200 run_options, "%s/read" % u_name, 0, debug_urls=debug_urls)
203 run_options, "%s/read" % v_name, 0, debug_urls=debug_urls)
208 sess.run(w, options=run_options, run_metadata=run_metadata)
231 run_options = config_pb2.RunOptions(output_partition_graphs=True)
234 run_options,
239 run_options, "v", 0, ["DebugNumericSummary"], debug_urls=debug_urls)
[all …]
/external/tensorflow/tensorflow/python/client/
Dsession_ref.cc127 Status RecordRun(Session* session, const RunOptions& run_options, in RecordRun() argument
137 *req->mutable_options() = run_options; in RecordRun()
158 if (&run_options == kEmptyRunOptions()) { in RecordRun()
162 RUN_WITH_TIMESTAMP(Run, run_options, inputs, output_tensor_names, in RecordRun()
185 Status RecordCreate(Session* session, const RunOptions& run_options, in RecordCreate() argument
192 if (&run_options == kEmptyRunOptions()) { in RecordCreate()
195 RUN_WITH_TIMESTAMP(Create, run_options, graph); in RecordCreate()
206 Status RecordExtend(Session* session, const RunOptions& run_options, in RecordExtend() argument
213 if (&run_options == kEmptyRunOptions()) { in RecordExtend()
216 RUN_WITH_TIMESTAMP(Extend, run_options, graph); in RecordExtend()
[all …]
Dtimeline_test.py46 run_options = config_pb2.RunOptions(
53 options=run_options,
62 run_options = config_pb2.RunOptions(
70 sess.run(result, options=run_options, run_metadata=run_metadata)
94 run_options = config_pb2.RunOptions(
102 sess.run(result, options=run_options, run_metadata=run_metadata)
140 run_options = config_pb2.RunOptions(
153 sess.run(result, options=run_options, run_metadata=run_metadata)
170 run_options = config_pb2.RunOptions(
182 sess.run(result, options=run_options, run_metadata=run_metadata)
/external/tensorflow/tensorflow/python/ops/
Dcollective_ops_test.py44 run_options = config_pb2.RunOptions()
46 run_options.experimental.collective_graph_key = 1
47 results = sess.run([colred0, colred1], options=run_options)
105 run_options = config_pb2.RunOptions()
106 run_options.experimental.collective_graph_key = 1
107 results = sess.run([out0, out1], options=run_options)
126 run_options = config_pb2.RunOptions()
128 run_options.experimental.collective_graph_key = 1
129 results = sess.run([colred0, colred1], options=run_options)
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_test_base.cc151 ExecutableRunOptions run_options; in DefaultExecutableRunOptions() local
152 run_options.set_intra_op_thread_pool(thread_pool_wrapper_->device.get()); in DefaultExecutableRunOptions()
153 run_options.set_allocator(GetOrCreateAllocator(local_client_->platform())); in DefaultExecutableRunOptions()
154 return run_options; in DefaultExecutableRunOptions()
169 const ExecutableRunOptions& run_options) { in ExecuteLocallyOrDie() argument
170 return ExecuteLocally(computation, arguments, build_options, run_options) in ExecuteLocallyOrDie()
185 const ExecutableRunOptions& run_options) { in ExecuteLocally() argument
193 TF_ASSIGN_OR_RETURN(auto ret, executable->Run(arguments, run_options)); in ExecuteLocally()
197 auto* stream = run_options.stream(); in ExecuteLocally()
/external/tensorflow/tensorflow/contrib/session_bundle/
Dsession_bundle.cc131 Status RunRestoreOp(const RunOptions& run_options, const StringPiece export_dir, in RunRestoreOp() argument
144 return session->Run(run_options, inputs, {}, {string(restore_op_name)}, in RunRestoreOp()
148 Status RunInitOp(const RunOptions& run_options, const StringPiece export_dir, in RunInitOp() argument
155 return session->Run(run_options, inputs, {}, {string(init_op_name)}, in RunInitOp()
160 const SessionOptions& options, const RunOptions& run_options, in LoadSessionBundleFromPathUsingRunOptionsInternal() argument
163 LOG(INFO) << "Using RunOptions: " << DebugStringIfAvailable(run_options); in LoadSessionBundleFromPathUsingRunOptionsInternal()
208 RunRestoreOp(run_options, export_dir, asset_files, in LoadSessionBundleFromPathUsingRunOptionsInternal()
219 TF_RETURN_IF_ERROR(RunInitOp(run_options, export_dir, asset_files, in LoadSessionBundleFromPathUsingRunOptionsInternal()
238 const RunOptions& run_options, in LoadSessionBundleFromPathUsingRunOptions() argument
243 options, run_options, export_dir, bundle); in LoadSessionBundleFromPathUsingRunOptions()
Dsession_bundle_test.cc196 RunOptions run_options; in TEST() local
197 run_options.set_trace_level(RunOptions::FULL_TRACE); in TEST()
201 session_options, run_options, export_path, &bundle)); in TEST()
215 RunOptions run_options; in TEST() local
216 run_options.set_inter_op_thread_pool(threadpool_index); in TEST()
220 session_options, run_options, export_path, &bundle)); in TEST()
234 RunOptions run_options; in TEST() local
235 run_options.set_inter_op_thread_pool(invalid_threadpool_index); in TEST()
239 session_options, run_options, export_path, &bundle); in TEST()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutable.cc49 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream() argument
52 se::Stream* stream = run_options->stream(); in ExecuteOnStream()
92 run_options->stream(), *arguments[p])); in ExecuteOnStream()
108 result_literal.shape(), run_options->allocator(), in ExecuteOnStream()
111 run_options->stream(), result_literal, result)); in ExecuteOnStream()
125 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream() argument
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session.cc138 Status GrpcSession::Create(const RunOptions& run_options, in Create() argument
141 call_options.SetTimeout(run_options.timeout_in_ms()); in Create()
175 Status GrpcSession::Extend(const RunOptions& run_options, in Extend() argument
178 call_options.SetTimeout(run_options.timeout_in_ms()); in Extend()
183 const RunOptions& run_options, in RunHelper() argument
194 *req->mutable_options() = run_options; in RunHelper()
196 if (run_options.timeout_in_ms() == 0) { in RunHelper()
269 Status GrpcSession::Run(const RunOptions& run_options, in Run() argument
275 return RunHelper(run_options, inputs, output_tensor_names, target_node_names, in Run()
283 RunOptions run_options; in Run() local
[all …]
/external/tensorflow/tensorflow/java/src/main/native/
Dsaved_model_bundle_jni.cc25 jbyteArray config, jbyteArray run_options) { in Java_org_tensorflow_SavedModelBundle_load() argument
45 if (run_options != nullptr) { in Java_org_tensorflow_SavedModelBundle_load()
46 size_t sz = env->GetArrayLength(run_options); in Java_org_tensorflow_SavedModelBundle_load()
48 jbyte* run_options_data = env->GetByteArrayElements(run_options, nullptr); in Java_org_tensorflow_SavedModelBundle_load()
51 env->ReleaseByteArrayElements(run_options, run_options_data, JNI_ABORT); in Java_org_tensorflow_SavedModelBundle_load()
/external/tensorflow/tensorflow/cc/training/
Dqueue_runner.cc91 const RunOptions& run_options) { in StartAndCollectCostGraph() argument
92 SetRunArgumentsAndCostGraph(run_options); in StartAndCollectCostGraph()
124 const RunOptions& run_options) { in StartAndCollectCostGraph() argument
125 SetRunArgumentsAndCostGraph(run_options); in StartAndCollectCostGraph()
217 void QueueRunner::SetRunArgumentsAndCostGraph(const RunOptions& run_options) { in SetRunArgumentsAndCostGraph() argument
223 run_options_ = run_options; in SetRunArgumentsAndCostGraph()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc67 const ServiceExecutableRunOptions* run_options, in ExecuteThunks() argument
70 se::Stream* main_stream = run_options->stream(); in ExecuteThunks()
94 run_options->BorrowStream(executor->device_ordinal())); in ExecuteThunks()
237 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream() argument
240 DeviceMemoryAllocator* memory_allocator = run_options->allocator(); in ExecuteOnStream()
247 se::StreamExecutor* executor = run_options->stream()->parent(); in ExecuteOnStream()
284 TF_RETURN_IF_ERROR(ExecuteThunks(run_options, *buffer_allocations, in ExecuteOnStream()
350 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream() argument
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc360 xla::ExecutableRunOptions run_options; in Compute() local
361 run_options.set_stream(stream); in Compute()
362 run_options.set_allocator(platform_info_.allocator()); in Compute()
363 run_options.set_intra_op_thread_pool(&ctx->eigen_cpu_device()); in Compute()
364 run_options.set_rng_seed(GetXLARandomSeed()); in Compute()
368 auto run_result = executable->Run(launch_context.arguments(), run_options); in Compute()
534 xla::ExecutableRunOptions run_options; in Compute() local
535 run_options.set_stream(stream); in Compute()
536 run_options.set_allocator(platform_info_.allocator()); in Compute()
537 run_options.set_intra_op_thread_pool(&ctx->eigen_cpu_device()); in Compute()
[all …]

12345