Home
last modified time | relevance | path

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

12345

/external/tensorflow/tensorflow/compiler/xla/service/
Dexecutable.cc82 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream() argument
86 ExecuteAsyncOnStream(run_options, arguments, hlo_execution_profile); in ExecuteOnStream()
87 Status blocking_status = run_options->stream()->BlockHostUntilDone(); in ExecuteOnStream()
104 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream() argument
113 ExecuteAsyncOnStream(run_options, std::move(args), in ExecuteAsyncOnStream()
119 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream() argument
123 run_options, std::move(arguments), hlo_execution_profile); in ExecuteOnStream()
124 Status blocking_status = run_options->stream()->BlockHostUntilDone(); in ExecuteOnStream()
131 absl::Span<const ServiceExecutableRunOptions> run_options, in ExecuteOnStreams() argument
133 TF_RET_CHECK(run_options.size() == arguments.size()); in ExecuteOnStreams()
[all …]
Dexecutable.h259 const ServiceExecutableRunOptions* run_options,
282 const ServiceExecutableRunOptions* run_options,
289 const ServiceExecutableRunOptions* run_options,
294 const ServiceExecutableRunOptions* run_options,
303 absl::Span<const ServiceExecutableRunOptions> run_options,
320 const ServiceExecutableRunOptions* run_options,
324 const ServiceExecutableRunOptions* run_options,
328 const ServiceExecutableRunOptions* run_options,
332 const ServiceExecutableRunOptions* run_options,
/external/tensorflow/tensorflow/cc/saved_model/
Dsaved_model_bundle_test.cc114 RunOptions run_options; in TEST_F() local
119 TF_ASSERT_OK(LoadSavedModel(session_options, run_options, export_dir, in TEST_F()
128 RunOptions run_options; in TEST_F() local
132 TF_ASSERT_OK(LoadSavedModel(session_options, run_options, export_dir, in TEST_F()
140 RunOptions run_options; in TEST_F() local
144 TF_ASSERT_OK(LoadSavedModel(session_options, run_options, export_dir, in TEST_F()
156 RunOptions run_options; in TEST_F() local
160 TF_ASSERT_OK(LoadSavedModel(session_options, run_options, export_dir, in TEST_F()
169 TF_ASSERT_OK(RestoreSession(run_options, actual_bundle.meta_graph_def, in TEST_F()
176 RunOptions run_options; in TEST_F() local
[all …]
Dsaved_model_bundle_lite_test.cc98 RunOptions run_options; in CheckSavedModelBundle() local
99 run_options.set_output_partition_graphs(true); in CheckSavedModelBundle()
102 bundle.GetSession()->Run(run_options, {{input_name, input}}, in CheckSavedModelBundle()
115 RunOptions run_options; in TEST_F() local
120 TF_ASSERT_OK(LoadSavedModel(session_options, run_options, export_dir, in TEST_F()
129 RunOptions run_options; in TEST_F() local
133 TF_ASSERT_OK(LoadSavedModel(session_options, run_options, export_dir, in TEST_F()
142 RunOptions run_options; in TEST_F() local
146 TF_ASSERT_OK(LoadSavedModel(session_options, run_options, export_dir, in TEST_F()
153 RunOptions run_options; in TEST_F() local
[all …]
Dloader.cc151 Status RunOnce(const RunOptions& run_options, in RunOnce() argument
159 *callable_options.mutable_run_options() = run_options; in RunOnce()
185 Status RunInitOp(const RunOptions& run_options, const string& export_dir, in RunInitOp() argument
195 return RunOnce(run_options, inputs, {}, {init_op_name}, in RunInitOp()
201 Status RunRestore(const RunOptions& run_options, const string& export_dir, in RunRestore() argument
234 return RunOnce(run_options, inputs, {}, {string(restore_op_name)}, in RunRestore()
253 const RunOptions& run_options, in LoadSavedModelInternal() argument
263 TF_RETURN_IF_ERROR(RestoreSession(run_options, bundle->meta_graph_def, in LoadSavedModelInternal()
269 const RunOptions& run_options, const string& export_dir, in LoadSavedModel() argument
274 const Status status = LoadSavedModelInternal(session_options, run_options, in LoadSavedModel()
[all …]
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc55 const ExecutableRunOptions& run_options, const Backend& backend) { in ValidateExecutionOptions() argument
56 if (run_options.stream() != nullptr) { in ValidateExecutionOptions()
57 if (!run_options.stream()->ok()) { in ValidateExecutionOptions()
63 run_options.stream()->parent()->platform(); in ValidateExecutionOptions()
72 if (run_options.device_ordinal() != -1) { in ValidateExecutionOptions()
81 int run_device_ordinal = run_options.device_ordinal(); in ValidateExecutionOptions()
83 run_device_ordinal = run_options.stream() != nullptr in ValidateExecutionOptions()
84 ? run_options.stream()->parent()->device_ordinal() in ValidateExecutionOptions()
104 if (!run_options.allocator()) { in ValidateExecutionOptions()
108 if (run_options.allocator()->platform() != backend.platform()) { in ValidateExecutionOptions()
[all …]
Dlocal_client.h52 ExecutableRunOptions run_options);
57 ExecutableRunOptions run_options);
63 ExecutableRunOptions run_options);
68 ExecutableRunOptions run_options);
79 std::vector<ExecutionInput> arguments, ExecutableRunOptions run_options);
86 Status ValidateExecutionOptions(const ExecutableRunOptions& run_options,
94 ExecutableRunOptions run_options);
104 const ExecutableRunOptions& run_options, in AsyncCallAndBlockHostUntilDone() argument
107 RunHelper(argument_shapes, run_options)); in AsyncCallAndBlockHostUntilDone()
108 ExecutableRunOptions options = options_and_stream.first.run_options(); in AsyncCallAndBlockHostUntilDone()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime.cc233 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length, in __xla_cpu_runtime_AcquireInfeedBufferForDequeue() argument
236 run_options ? run_options->stream()->parent()->device_ordinal() : 0; in __xla_cpu_runtime_AcquireInfeedBufferForDequeue()
257 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length, in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue() argument
260 run_options ? run_options->stream()->parent()->device_ordinal() : 0; in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue()
276 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length, in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation() argument
279 run_options ? run_options->stream()->parent()->device_ordinal() : 0; in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation()
300 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length, in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation() argument
303 run_options ? run_options->stream()->parent()->device_ordinal() : 0; in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation()
587 int GetDeviceOrdinal(const xla::ExecutableRunOptions* run_options) { in GetDeviceOrdinal() argument
588 if (run_options->stream()) { in GetDeviceOrdinal()
[all …]
Dcpu_executable.cc149 const ExecutableRunOptions* run_options, in ExecuteComputeFunction() argument
205 compute_function_(result_buffer, run_options, nullptr, buffer_pointers.data(), in ExecuteComputeFunction()
210 if (run_options->execution_profile()) { in ExecuteComputeFunction()
212 run_options->execution_profile()->set_compute_time_ns( in ExecuteComputeFunction()
216 run_options->execution_profile()->set_compute_cycle_count( in ExecuteComputeFunction()
226 const ServiceExecutableRunOptions* run_options, in CreateResultShapedBuffer() argument
229 se::Stream* stream = run_options->stream(); in CreateResultShapedBuffer()
231 run_options->allocator(), in CreateResultShapedBuffer()
297 run_options->allocator()->Allocate( in CreateResultShapedBuffer()
326 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream() argument
[all …]
Dcpu_runtime.h128 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length,
144 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length,
150 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length,
163 const xla::ExecutableRunOptions* run_options, xla::int32 buffer_length,
174 const xla::ExecutableRunOptions* run_options,
181 const xla::ExecutableRunOptions* run_options, xla::int32 channel_id_present,
187 const xla::ExecutableRunOptions* run_options, xla::int32 channel_id_present,
194 const xla::ExecutableRunOptions* run_options, void* output_buffer);
Druntime_conv2d.cc38 const xla::ExecutableRunOptions* run_options = in __xla_cpu_runtime_EigenConvF32() local
40 XLA_LIGHTWEIGHT_CHECK(run_options->intra_op_thread_pool() != nullptr); in __xla_cpu_runtime_EigenConvF32()
42 *run_options->intra_op_thread_pool(), out, lhs, rhs, input_batch, in __xla_cpu_runtime_EigenConvF32()
62 const xla::ExecutableRunOptions* run_options = in __xla_cpu_runtime_EigenConvF16() local
64 XLA_LIGHTWEIGHT_CHECK(run_options->intra_op_thread_pool() != nullptr); in __xla_cpu_runtime_EigenConvF16()
66 *run_options->intra_op_thread_pool(), out, lhs, rhs, input_batch, in __xla_cpu_runtime_EigenConvF16()
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()
/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
174 run_options,
191 run_options,
199 run_options.debug_options.reset_disk_byte_usage = reset_disk_byte_usage
202 def watch_graph_with_denylists(run_options, argument
283 run_options,
291 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/compiler/xla/service/gpu/
Dgpu_executable.cc95 const ServiceExecutableRunOptions* run_options) { in CheckCompatibilityWithServiceExecutableRunOptions() argument
96 se::Stream* main_stream = run_options->stream(); in CheckCompatibilityWithServiceExecutableRunOptions()
129 const ServiceExecutableRunOptions* run_options, in ExecuteThunks() argument
133 CheckCompatibilityWithServiceExecutableRunOptions(run_options)); in ExecuteThunks()
138 se::Stream* main_stream = run_options->stream(); in ExecuteThunks()
152 run_options->BorrowStream(executor->device_ordinal())); in ExecuteThunks()
185 run_options->run_options().gpu_executable_run_options(); in ExecuteThunks()
189 run_options->run_options().run_id(), in ExecuteThunks()
191 run_options->run_options().device_assignment(), in ExecuteThunks()
215 if (run_options->run_options().then_execute_function()) { in ExecuteThunks()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutable_base.cc42 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream() argument
45 se::Stream* stream = run_options->stream(); in ExecuteAsyncOnStream()
53 int device_ordinal = run_options->device_ordinal(); in ExecuteAsyncOnStream()
114 run_options->stream(), argument_buffers[p])); in ExecuteAsyncOnStream()
124 Evaluate(run_options, *computation, arg_literals)); in ExecuteAsyncOnStream()
131 result_literal.shape(), run_options->allocator(), in ExecuteAsyncOnStream()
134 run_options->stream(), result_literal, result_buffers)); in ExecuteAsyncOnStream()
139 ExecutionProfile* profile = run_options->run_options().execution_profile(); in ExecuteAsyncOnStream()
/external/tensorflow/tensorflow/python/client/
Dsession_ref.cc128 Status RecordRun(Session* session, const RunOptions& run_options, in RecordRun() argument
138 *req->mutable_options() = run_options; in RecordRun()
159 if (&run_options == kEmptyRunOptions()) { in RecordRun()
163 RUN_WITH_TIMESTAMP(Run, run_options, inputs, output_tensor_names, in RecordRun()
186 Status RecordCreate(Session* session, const RunOptions& run_options, in RecordCreate() argument
193 if (&run_options == kEmptyRunOptions()) { in RecordCreate()
196 RUN_WITH_TIMESTAMP(Create, run_options, graph); in RecordCreate()
207 Status RecordExtend(Session* session, const RunOptions& run_options, in RecordExtend() argument
214 if (&run_options == kEmptyRunOptions()) { in RecordExtend()
217 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)
143 run_options = config_pb2.RunOptions(
156 sess.run(result, options=run_options, run_metadata=run_metadata)
173 run_options = config_pb2.RunOptions(
185 sess.run(result, options=run_options, run_metadata=run_metadata)
/external/tensorflow/tensorflow/core/public/
Dsession.h142 virtual Status Create(const RunOptions& run_options, const GraphDef& graph) { in Create() argument
147 virtual Status Extend(const RunOptions& run_options, const GraphDef& graph) { in Extend() argument
153 virtual Status Create(const RunOptions& run_options, GraphDef&& graph) { in Create() argument
154 return Create(run_options, graph); in Create()
156 virtual Status Extend(const RunOptions& run_options, GraphDef&& graph) { in Extend() argument
157 return Extend(run_options, graph); in Extend()
160 virtual Status Close(const RunOptions& run_options) { in Close() argument
171 virtual Status Run(const RunOptions& run_options,
180 virtual Status Run(const RunOptions& run_options, in Run() argument
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session.cc138 Status GrpcSession::Create(const RunOptions& run_options, in Create() argument
140 return Create(run_options, GraphDef(graph)); in Create()
149 Status GrpcSession::Create(const RunOptions& run_options, GraphDef&& graph) { in Create() argument
151 call_options.SetTimeout(run_options.timeout_in_ms()); in Create()
182 Status GrpcSession::Extend(const RunOptions& run_options, in Extend() argument
184 return Extend(run_options, GraphDef(graph)); in Extend()
193 Status GrpcSession::Extend(const RunOptions& run_options, GraphDef&& graph) { in Extend() argument
195 call_options.SetTimeout(run_options.timeout_in_ms()); in Extend()
200 const RunOptions& run_options, in RunHelper() argument
211 *req->mutable_options() = run_options; in RunHelper()
[all …]
Dgrpc_session.h66 Status Create(const RunOptions& run_options, const GraphDef& graph) override;
68 Status Create(const RunOptions& run_options, GraphDef&& graph) override;
75 Status Run(const RunOptions& run_options,
82 Status Extend(const RunOptions& run_options, const GraphDef& graph) override;
84 Status Extend(const RunOptions& run_options, GraphDef&& graph) override;
132 Status RunHelper(const RunOptions& run_options,
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_test_base.cc157 ExecutableRunOptions run_options; in DefaultExecutableRunOptions() local
158 run_options.set_intra_op_thread_pool(thread_pool_wrapper_->device.get()); in DefaultExecutableRunOptions()
159 run_options.set_allocator(GetOrCreateAllocator(local_client_->platform())); in DefaultExecutableRunOptions()
160 return run_options; in DefaultExecutableRunOptions()
175 const ExecutableRunOptions& run_options) { in ExecuteLocallyOrDie() argument
176 return ExecuteLocally(computation, arguments, build_options, run_options) in ExecuteLocallyOrDie()
191 const ExecutableRunOptions& run_options) { in ExecuteLocally() argument
200 TF_ASSIGN_OR_RETURN(auto ret, executables[0]->Run(arguments, run_options)); in ExecuteLocally()
204 auto* stream = run_options.stream(); in ExecuteLocally()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_on_demand_compiler.cc43 se_options.allocator = ApiConverter::ToC(options.run_options().allocator()); in ToC()
44 se_options.device_ordinal = options.run_options().device_ordinal(); in ToC()
45 if (options.run_options().host_to_device_stream() != nullptr) { in ToC()
48 options.run_options().host_to_device_stream()->implementation()) in ToC()
54 if (options.run_options().device_assignment() != nullptr) { in ToC()
56 options.run_options() in ToC()
67 se_options.rng_seed = options.run_options().rng_seed(); in ToC()
68 se_options.run_id = options.run_options().run_id().ToInt(); in ToC()
69 se_options.launch_id = options.run_options().launch_id(); in ToC()
71 CHECK_EQ(options.run_options().then_execute_function(), nullptr) in ToC()
[all …]
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executable.cc38 const ServiceExecutableRunOptions& run_options, in LoadProgramAndEnqueueToStream() argument
60 int32 rng_seed = run_options.run_options().rng_seed(); in LoadProgramAndEnqueueToStream()
63 auto dev_assign = run_options.run_options().device_assignment(); in LoadProgramAndEnqueueToStream()
77 run_options.run_options().stream()->implementation()); in LoadProgramAndEnqueueToStream()
Dtpu_executable_interface.cc169 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream() argument
177 se::Stream* stream = run_options->stream(); in ExecuteAsyncOnStream()
179 CHECK_NE(run_options->allocator(), nullptr); in ExecuteAsyncOnStream()
188 shape, alias_config, run_options->allocator(), &arguments, stream, in ExecuteAsyncOnStream()
189 run_options->run_options().host_to_device_stream())); in ExecuteAsyncOnStream()
221 *run_options, memory_bases, result.Result().root_buffer(), in ExecuteAsyncOnStream()

12345