Home
last modified time | relevance | path

Searched refs:gpu_executable_run_options (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dexecutable_run_options.cc106 const gpu::GpuExecutableRunOptions* gpu_executable_run_options) { in set_gpu_executable_run_options() argument
107 gpu_executable_run_options_ = gpu_executable_run_options; in set_gpu_executable_run_options()
112 ExecutableRunOptions::gpu_executable_run_options() const { in gpu_executable_run_options() function in xla::ExecutableRunOptions
H A Dexecutable_run_options.h155 const gpu::GpuExecutableRunOptions* gpu_executable_run_options);
156 const gpu::GpuExecutableRunOptions* gpu_executable_run_options() const;
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dgpu_device.cc341 gpu::GpuExecutableRunOptions* gpu_executable_run_options) { in BuildDistributedDevices() argument
389 gpu_executable_run_options->set_gpu_global_device_ids( in BuildDistributedDevices()
394 gpu_executable_run_options->set_nccl_unique_id_callback( in BuildDistributedDevices()
H A DBUILD326 "//tensorflow/compiler/xla/service/gpu:gpu_executable_run_options",
427 "//tensorflow/compiler/xla/service/gpu:gpu_executable_run_options",
466 "//tensorflow/compiler/xla/service/gpu:gpu_executable_run_options",
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_executable_run_options.cc59 run_options.run_options().gpu_executable_run_options(); in NcclExecuteParams()
H A DBUILD70 ":gpu_executable_run_options",
102 name = "gpu_executable_run_options",
103 srcs = ["gpu_executable_run_options.cc"],
104 hdrs = ["gpu_executable_run_options.h"],
431 ":gpu_executable_run_options",
509 ":gpu_executable_run_options",
629 ":gpu_executable_run_options",
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/kernels/
H A DBUILD35 "//tensorflow/compiler/xla/service/gpu:gpu_executable_run_options",
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/
H A DBUILD130 "//tensorflow/compiler/xla/service/gpu:gpu_executable_run_options",
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A DBUILD262 "//tensorflow/compiler/xla/service/gpu:gpu_executable_run_options",
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A DBUILD624 "//tensorflow/compiler/xla/service/gpu:gpu_executable_run_options",