| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | executable.cc | 81 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream() argument
85 ExecuteAsyncOnStream(run_options, arguments, hlo_execution_profile); in ExecuteOnStream()
86 Status blocking_status = run_options->stream()->BlockHostUntilDone(); in ExecuteOnStream()
103 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream() argument
112 ExecuteAsyncOnStream(run_options, std::move(args), in ExecuteAsyncOnStream()
118 const ServiceExecutableRunOptions* run_options, in ExecuteOnStream() argument
122 run_options, std::move(arguments), hlo_execution_profile); in ExecuteOnStream()
123 Status blocking_status = run_options->stream()->BlockHostUntilDone(); in ExecuteOnStream()
130 absl::Span<const ServiceExecutableRunOptions> run_options, in ExecuteOnStreams() argument
132 TF_RET_CHECK(run_options.size() == arguments.size()); in ExecuteOnStreams()
[all …]
|
| /external/tensorflow/tensorflow/cc/saved_model/ |
| D | saved_model_bundle_test.cc | 117 RunOptions run_options; in TEST_F() local
122 TF_ASSERT_OK(LoadSavedModel(session_options, run_options, export_dir, in TEST_F()
131 RunOptions run_options; in TEST_F() local
135 TF_ASSERT_OK(LoadSavedModel(session_options, run_options, export_dir, in TEST_F()
143 RunOptions run_options; in TEST_F() local
147 TF_ASSERT_OK(LoadSavedModel(session_options, run_options, export_dir, in TEST_F()
159 RunOptions run_options; in TEST_F() local
163 TF_ASSERT_OK(LoadSavedModel(session_options, run_options, export_dir, in TEST_F()
172 TF_ASSERT_OK(RestoreSession(run_options, actual_bundle.meta_graph_def, in TEST_F()
179 RunOptions run_options; in TEST_F() local
[all …]
|
| D | saved_model_bundle_lite_test.cc | 98 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 …]
|
| D | loader.cc | 174 Status RunOnce(const RunOptions& run_options, in RunOnce() argument
182 *callable_options.mutable_run_options() = run_options; in RunOnce()
208 Status RunInitOp(const RunOptions& run_options, const string& export_dir, in RunInitOp() argument
218 return RunOnce(run_options, inputs, {}, {init_op_name}, in RunInitOp()
224 Status RunRestore(const RunOptions& run_options, const string& export_dir, in RunRestore() argument
260 return RunOnce(run_options, inputs, {}, {string(restore_op_name)}, in RunRestore()
279 const RunOptions& run_options, in LoadSavedModelInternal() argument
289 TF_RETURN_IF_ERROR(RestoreSession(run_options, bundle->meta_graph_def, in LoadSavedModelInternal()
295 const RunOptions& run_options, const string& export_dir, in LoadSavedModel() argument
302 const Status status = LoadSavedModelInternal(session_options, run_options, in LoadSavedModel()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 56 const ExecutableRunOptions& run_options, const Backend& backend) { in ValidateExecutionOptions() argument
57 if (run_options.stream() != nullptr) { in ValidateExecutionOptions()
58 if (!run_options.stream()->ok()) { in ValidateExecutionOptions()
64 run_options.stream()->parent()->platform(); in ValidateExecutionOptions()
73 if (run_options.device_ordinal() != -1) { in ValidateExecutionOptions()
82 int run_device_ordinal = run_options.device_ordinal(); in ValidateExecutionOptions()
84 run_device_ordinal = run_options.stream() != nullptr in ValidateExecutionOptions()
85 ? run_options.stream()->parent()->device_ordinal() in ValidateExecutionOptions()
105 if (!run_options.allocator()) { in ValidateExecutionOptions()
109 if (run_options.allocator()->platform() != backend.platform()) { in ValidateExecutionOptions()
[all …]
|
| D | local_client.h | 53 ExecutableRunOptions run_options);
58 ExecutableRunOptions run_options);
64 ExecutableRunOptions run_options);
69 ExecutableRunOptions run_options);
80 std::vector<ExecutionInput> arguments, ExecutableRunOptions run_options);
87 Status ValidateExecutionOptions(const ExecutableRunOptions& run_options,
95 ExecutableRunOptions run_options);
105 const ExecutableRunOptions& run_options, in AsyncCallAndBlockHostUntilDone() argument
108 RunHelper(argument_shapes, run_options)); in AsyncCallAndBlockHostUntilDone()
109 ExecutableRunOptions options = options_and_stream.first.run_options(); in AsyncCallAndBlockHostUntilDone()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.h | 140 const xla::ExecutableRunOptions* run_options, int32_t buffer_length,
156 const xla::ExecutableRunOptions* run_options, int32_t buffer_length,
162 const xla::ExecutableRunOptions* run_options, int32_t buffer_length,
175 const xla::ExecutableRunOptions* run_options, int32_t buffer_length,
186 const xla::ExecutableRunOptions* run_options,
193 const xla::ExecutableRunOptions* run_options, int32_t channel_id_present,
198 const xla::ExecutableRunOptions* run_options, int32_t channel_id_present,
205 const xla::ExecutableRunOptions* run_options, void* output_buffer);
208 const xla::ExecutableRunOptions* run_options, void* output_buffer);
|
| D | cpu_executable.cc | 154 const ExecutableRunOptions* run_options, in ExecuteComputeFunction() argument
192 compute_function_(nullptr, run_options, nullptr, buffer_pointers.data(), in ExecuteComputeFunction()
197 if (run_options->execution_profile()) { in ExecuteComputeFunction()
199 run_options->execution_profile()->set_compute_time_ns( in ExecuteComputeFunction()
203 run_options->execution_profile()->set_compute_cycle_count( in ExecuteComputeFunction()
219 const ServiceExecutableRunOptions* run_options, in CreateResultShapedBuffer() argument
222 se::Stream* stream = run_options->stream(); in CreateResultShapedBuffer()
224 run_options->allocator(), in CreateResultShapedBuffer()
290 run_options->allocator()->Allocate( in CreateResultShapedBuffer()
319 const ServiceExecutableRunOptions* run_options, in ExecuteAsyncOnStream() argument
[all …]
|
| D | cpu_runtime.cc | 261 int GetDeviceOrdinal(const xla::ExecutableRunOptions* run_options) { in GetDeviceOrdinal() argument
262 if (!run_options) { in GetDeviceOrdinal()
264 } else if (run_options->device_ordinal() != -1) { in GetDeviceOrdinal()
265 return run_options->device_ordinal(); in GetDeviceOrdinal()
267 return run_options->stream()->parent()->device_ordinal(); in GetDeviceOrdinal()
301 const xla::ExecutableRunOptions* run_options, int32_t buffer_length, in __xla_cpu_runtime_AcquireInfeedBufferForDequeue() argument
303 int device_ordinal = GetDeviceOrdinal(run_options); in __xla_cpu_runtime_AcquireInfeedBufferForDequeue()
324 const xla::ExecutableRunOptions* run_options, int32_t buffer_length, in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue() argument
326 int device_ordinal = GetDeviceOrdinal(run_options); in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue()
342 const xla::ExecutableRunOptions* run_options, int32_t buffer_length, in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation() argument
[all …]
|
| D | runtime_conv2d.cc | 35 const xla::ExecutableRunOptions* run_options = in __xla_cpu_runtime_EigenConv2DF32() local
37 XLA_LIGHTWEIGHT_CHECK(run_options->intra_op_thread_pool() != nullptr); in __xla_cpu_runtime_EigenConv2DF32()
39 *run_options->intra_op_thread_pool(), out, lhs, rhs, input_batch, in __xla_cpu_runtime_EigenConv2DF32()
57 const xla::ExecutableRunOptions* run_options = in __xla_cpu_runtime_EigenConv2DF16() local
59 XLA_LIGHTWEIGHT_CHECK(run_options->intra_op_thread_pool() != nullptr); in __xla_cpu_runtime_EigenConv2DF16()
61 *run_options->intra_op_thread_pool(), out, lhs, rhs, input_batch, in __xla_cpu_runtime_EigenConv2DF16()
|
| D | runtime_conv3d.cc | 37 const xla::ExecutableRunOptions* run_options = in __xla_cpu_runtime_EigenConv3DF32() local
39 XLA_LIGHTWEIGHT_CHECK(run_options->intra_op_thread_pool() != nullptr); in __xla_cpu_runtime_EigenConv3DF32()
41 *run_options->intra_op_thread_pool(), out, lhs, rhs, input_batch, input_x, in __xla_cpu_runtime_EigenConv3DF32()
62 const xla::ExecutableRunOptions* run_options = in __xla_cpu_runtime_EigenConv3DF16() local
64 XLA_LIGHTWEIGHT_CHECK(run_options->intra_op_thread_pool() != nullptr); in __xla_cpu_runtime_EigenConv3DF16()
66 *run_options->intra_op_thread_pool(), out, lhs, rhs, input_batch, input_x, in __xla_cpu_runtime_EigenConv3DF16()
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | debug_utils.py | 21 def add_debug_tensor_watch(run_options, argument
56 watch_opts = run_options.debug_options.debug_tensor_watch_opts
57 run_options.debug_options.global_step = global_step
77 def watch_graph(run_options, argument
169 run_options,
186 run_options,
194 run_options.debug_options.reset_disk_byte_usage = reset_disk_byte_usage
197 def watch_graph_with_denylists(run_options, argument
278 run_options,
286 run_options.debug_options.reset_disk_byte_usage = reset_disk_byte_usage
|
| D | session_debug_testlib.py | 148 run_options = config_pb2.RunOptions(output_partition_graphs=True)
150 run_options,
159 options=run_options,
190 run_options = config_pb2.RunOptions(output_partition_graphs=True)
195 run_options, "%s/read" % u_name, 0, debug_urls=debug_urls)
198 run_options, "%s/read" % v_name, 0, debug_urls=debug_urls)
203 sess.run(w, options=run_options, run_metadata=run_metadata)
226 run_options = config_pb2.RunOptions(output_partition_graphs=True)
229 run_options,
234 run_options, "v", 0, ["DebugNumericSummary"], debug_urls=debug_urls)
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| D | tpu_executable.cc | 28 se_options.allocator = ApiConverter::ToC(options.run_options().allocator()); in ToC()
29 se_options.device_ordinal = options.run_options().device_ordinal(); in ToC()
30 if (options.run_options().host_to_device_stream() != nullptr) { in ToC()
33 options.run_options().host_to_device_stream()->implementation()) in ToC()
39 if (options.run_options().device_assignment() != nullptr) { in ToC()
41 options.run_options() in ToC()
52 se_options.rng_seed = options.run_options().rng_seed(); in ToC()
53 se_options.run_id = options.run_options().run_id().ToInt(); in ToC()
54 se_options.launch_id = options.run_options().launch_id(); in ToC()
56 CHECK_EQ(options.run_options().then_execute_function(), nullptr) in ToC()
[all …]
|
| D | tpu_op_executable.cc | 38 const xla::ServiceExecutableRunOptions& run_options, in LoadProgramAndEnqueueToStream() argument
60 int32_t 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()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable_run_options.cc | 54 const ServiceExecutableRunOptions& run_options, se::Stream* stream) in NcclExecuteParams() argument
56 run_id(run_options.run_options().run_id()), in NcclExecuteParams()
57 device_assn(run_options.run_options().device_assignment()) { in NcclExecuteParams()
59 run_options.run_options().gpu_executable_run_options(); in NcclExecuteParams()
|
| D | jitrt_custom_calls.cc | 408 Error operator()(const ServiceExecutableRunOptions* run_options,
419 Error LaunchFunc::operator()(const ServiceExecutableRunOptions* run_options, in operator ()() argument
426 se::Stream* stream = run_options->stream(); in operator ()()
501 Error operator()(const ServiceExecutableRunOptions* run_options,
514 Error Gemm::operator()(const ServiceExecutableRunOptions* run_options, in operator ()() argument
527 se::Stream* stream = run_options->stream(); in operator ()()
575 Error operator()(const ServiceExecutableRunOptions* run_options,
590 const ServiceExecutableRunOptions* run_options, in operator ()() argument
598 se::Stream* stream = run_options->stream(); in operator ()()
805 const ServiceExecutableRunOptions* run_options, in operator ()()
[all …]
|
| D | gpu_executable.cc | 285 const ServiceExecutableRunOptions* run_options) { in CheckCompatibilityWithServiceExecutableRunOptions() argument
286 se::Stream* main_stream = run_options->stream(); in CheckCompatibilityWithServiceExecutableRunOptions()
315 Status MaybeSyncAndProfile(const ServiceExecutableRunOptions* run_options,
320 const ServiceExecutableRunOptions* run_options, in ExecuteThunks() argument
323 se::Stream* main_stream = run_options->stream(); in ExecuteThunks()
327 run_options->BorrowStream(executor->device_ordinal()); in ExecuteThunks()
345 *run_options, buffer_allocations, main_stream, in ExecuteThunks()
349 return MaybeSyncAndProfile(run_options, start_micros, in ExecuteThunks()
353 Status MaybeSyncAndProfile(const ServiceExecutableRunOptions* run_options, in MaybeSyncAndProfile() argument
374 if (run_options->run_options().execution_profile()) { in MaybeSyncAndProfile()
[all …]
|
| /external/tensorflow/tensorflow/python/client/ |
| D | session_ref.cc | 129 Status RecordRun(Session* session, const RunOptions& run_options, in RecordRun() argument
139 *req->mutable_options() = run_options; in RecordRun()
160 if (&run_options == kEmptyRunOptions()) { in RecordRun()
164 RUN_WITH_TIMESTAMP(Run, run_options, inputs, output_tensor_names, in RecordRun()
187 Status RecordCreate(Session* session, const RunOptions& run_options, in RecordCreate() argument
194 if (&run_options == kEmptyRunOptions()) { in RecordCreate()
197 RUN_WITH_TIMESTAMP(Create, run_options, graph); in RecordCreate()
208 Status RecordExtend(Session* session, const RunOptions& run_options, in RecordExtend() argument
215 if (&run_options == kEmptyRunOptions()) { in RecordExtend()
218 RUN_WITH_TIMESTAMP(Extend, run_options, graph); in RecordExtend()
[all …]
|
| D | timeline_test.py | 42 run_options = config_pb2.RunOptions(
49 options=run_options,
58 run_options = config_pb2.RunOptions(
66 sess.run(result, options=run_options, run_metadata=run_metadata)
90 run_options = config_pb2.RunOptions(
98 sess.run(result, options=run_options, run_metadata=run_metadata)
136 run_options = config_pb2.RunOptions(
149 sess.run(result, options=run_options, run_metadata=run_metadata)
166 run_options = config_pb2.RunOptions(
178 sess.run(result, options=run_options, run_metadata=run_metadata)
|
| /external/tensorflow/tensorflow/core/public/ |
| D | session.h | 142 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/ |
| D | grpc_session.cc | 139 Status GrpcSession::Create(const RunOptions& run_options, in Create() argument
141 return Create(run_options, GraphDef(graph)); in Create()
150 Status GrpcSession::Create(const RunOptions& run_options, GraphDef&& graph) { in Create() argument
152 call_options.SetTimeout(run_options.timeout_in_ms()); in Create()
183 Status GrpcSession::Extend(const RunOptions& run_options, in Extend() argument
185 return Extend(run_options, GraphDef(graph)); in Extend()
194 Status GrpcSession::Extend(const RunOptions& run_options, GraphDef&& graph) { in Extend() argument
196 call_options.SetTimeout(run_options.timeout_in_ms()); in Extend()
201 const RunOptions& run_options, in RunHelper() argument
212 *req->mutable_options() = run_options; in RunHelper()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_test_base.cc | 154 ExecutableRunOptions run_options; in DefaultExecutableRunOptions() local
155 run_options.set_intra_op_thread_pool(thread_pool_wrapper_->device.get()); in DefaultExecutableRunOptions()
156 run_options.set_allocator(GetOrCreateAllocator(local_client_->platform())); in DefaultExecutableRunOptions()
157 return run_options; in DefaultExecutableRunOptions()
172 const ExecutableRunOptions& run_options) { in ExecuteLocallyOrDie() argument
173 return ExecuteLocally(computation, arguments, build_options, run_options) in ExecuteLocallyOrDie()
188 const ExecutableRunOptions& run_options) { in ExecuteLocally() argument
197 TF_ASSIGN_OR_RETURN(auto ret, executables[0]->Run(arguments, run_options)); in ExecuteLocally()
201 auto* stream = run_options.stream(); in ExecuteLocally()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_op_executable.cc | 38 const xla::ServiceExecutableRunOptions& run_options, in LoadProgramAndEnqueueToStream() argument
60 int32_t 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()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executable_base.cc | 42 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()
116 run_options->stream(), argument_buffers[p])); in ExecuteAsyncOnStream()
126 Evaluate(run_options, *computation, arg_literals)); in ExecuteAsyncOnStream()
137 run_options->allocator(), &arguments, stream)); in ExecuteAsyncOnStream()
140 run_options->stream(), result_literal, result.Result())); in ExecuteAsyncOnStream()
144 ExecutionProfile* profile = run_options->run_options().execution_profile(); in ExecuteAsyncOnStream()
|