Home
last modified time | relevance | path

Searched refs:gpu_executable (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
H A DBUILD56 "//tensorflow/compiler/xla/service/gpu:gpu_executable",
223 "//tensorflow/compiler/xla/service/gpu:gpu_executable",
354 "//tensorflow/compiler/xla/service/gpu:gpu_executable",
381 "//tensorflow/compiler/xla/service/gpu:gpu_executable",
429 "//tensorflow/compiler/xla/service/gpu:gpu_executable",
720 "//tensorflow/compiler/xla/service/gpu:gpu_executable",
H A Dgemm_rewrite_test.cc54 GpuExecutable* gpu_executable = in CheckNumberOfAllocations() local
57 gpu_executable->GetAllocations(); in CheckNumberOfAllocations()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_compiler.cc1404 auto gpu_executable, in RunBackend()
1419 gpu_executable->set_ir_module_string(ir_module_string_before_opt); in RunBackend()
1425 DumpingEnabledForHloModule(gpu_executable->module())) { in RunBackend()
1427 *hlo_proto->mutable_hlo_module() = gpu_executable->module().ToProto(); in RunBackend()
1429 gpu_executable->set_hlo_proto(std::move(hlo_proto)); in RunBackend()
1431 gpu_executable->set_debug_info(buffer_assignment->GetStats().ToString()); in RunBackend()
1432 return static_cast<std::unique_ptr<Executable>>(std::move(gpu_executable)); in RunBackend()
H A DBUILD77 ":gpu_executable",
268 ":gpu_executable",
585 name = "gpu_executable",
591 "gpu_executable.cc",
607 "gpu_executable.h",
989 ":gpu_executable",
1598 ":gpu_executable",
1793 ":gpu_executable",
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/
H A DBUILD97 # :lmhlo_to_gpu_binary > xla/service/gpu:gpu_executable > :lmhlo_to_gpu
121 "//tensorflow/compiler/xla/service/gpu:gpu_executable",
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/g3doc/
H A Dgpu_backend.md10 …github.com/tensorflow/tensorflow/blob/master/tensorflow/compiler/xla/service/gpu/gpu_executable.h),
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A DBUILD1150 "//tensorflow/compiler/xla/service/gpu:gpu_executable",