| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | executable_build_options.cc | 35 ExecutableBuildOptions& ExecutableBuildOptions::set_device_ordinal( in set_device_ordinal() function in xla::ExecutableBuildOptions
108 output.set_device_ordinal(device_ordinal()); in ToProto()
136 output.set_device_ordinal(input.device_ordinal()); in ExecutableBuildOptionsFromProto()
|
| D | executable_build_options.h | 49 ExecutableBuildOptions& set_device_ordinal(int device_ordinal);
|
| D | local_client.h | 110 options.set_device_ordinal(-1); in AsyncCallAndBlockHostUntilDone()
|
| D | local_client.cc | 363 updated_options.set_device_ordinal(default_device_ordinal); in UpdateBuildOptions()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_allocation_test.cc | 91 computation, {}, ExecutableBuildOptions().set_device_ordinal(d), in XLA_TEST_F()
92 ExecutableRunOptions().set_device_ordinal(d).set_allocator(allocator)); in XLA_TEST_F()
|
| D | local_client_execute_test.cc | 620 DefaultExecutableBuildOptions().set_device_ordinal(d), in XLA_TEST_F()
621 DefaultExecutableRunOptions().set_device_ordinal(d)); in XLA_TEST_F()
628 DefaultExecutableBuildOptions().set_device_ordinal(d), in XLA_TEST_F()
629 DefaultExecutableRunOptions().set_device_ordinal(d)); in XLA_TEST_F()
646 DefaultExecutableBuildOptions().set_device_ordinal( in XLA_TEST_F()
648 DefaultExecutableRunOptions().set_device_ordinal( in XLA_TEST_F()
|
| D | multiple_devices_on_host_test.cc | 43 execute_options.set_device_ordinal(device_ordinal); in CompileAndExecute()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | executable_run_options.cc | 35 ExecutableRunOptions& ExecutableRunOptions::set_device_ordinal( in set_device_ordinal() function in xla::ExecutableRunOptions
|
| D | executable_run_options.h | 94 ExecutableRunOptions& set_device_ordinal(int device_ordinal);
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | dumped_computation_to_operation_list.cc | 88 build_options.set_device_ordinal(0); in RealMain()
|
| D | dumped_computation_to_text.cc | 65 build_options.set_device_ordinal(0); in RealMain()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | get_compiler_ir.cc | 58 build_options.set_device_ordinal( in BuildExecutable()
|
| D | xla_compilation_cache.cc | 264 build_options.set_device_ordinal(options.device_ordinal != -1 in GetBuildOptions()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_compile_ops.cc | 162 build_options.set_device_ordinal(device_ref.device_ordinal()); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_runner.cc | 476 run_options.set_device_ordinal(device); in GetServiceRunOptionsForDevice()
|
| D | service.cc | 527 options.set_device_ordinal(stream->parent()->device_ordinal()); in ExecuteAndRegisterResult()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/c/ |
| D | pjrt_c_api_wrapper_impl.cc | 207 ret.executable_build_options.set_device_ordinal(c_option->device_ordinal); in ConvertCCompileOptionstoCppCompileOptions()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | tfrt_cpu_pjrt_client.cc | 380 build_options.set_device_ordinal( in Compile()
1457 run_options.set_device_ordinal(device->local_hardware_id()); in ExecuteHelper()
|
| D | pjrt_stream_executor_client.cc | 2440 build_options.set_device_ordinal( in GetExecutableExtras()
|