Searched refs:gpu_executable (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | BUILD | 48 "//tensorflow/compiler/xla/service/gpu:gpu_executable",
120 "//tensorflow/compiler/xla/service/gpu:gpu_executable",
147 "//tensorflow/compiler/xla/service/gpu:gpu_executable",
174 "//tensorflow/compiler/xla/service/gpu:gpu_executable",
201 "//tensorflow/compiler/xla/service/gpu:gpu_executable",
228 "//tensorflow/compiler/xla/service/gpu:gpu_executable",
255 "//tensorflow/compiler/xla/service/gpu:gpu_executable",
279 "//tensorflow/compiler/xla/service/gpu:gpu_executable",
545 "//tensorflow/compiler/xla/service/gpu:gpu_executable",
|
| D | gemm_rewrite_test.cc | 47 GpuExecutable* gpu_executable = in CheckNumberOfAllocations() local
50 gpu_executable->GetAllocations(); in CheckNumberOfAllocations()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_compiler.cc | 949 auto* gpu_executable = new GpuExecutable( in RunBackend() local
958 gpu_executable->set_ir_module_string(ir_module_string_before_opt); in RunBackend()
960 return std::unique_ptr<Executable>(gpu_executable); in RunBackend()
1144 auto* gpu_executable = new GpuExecutable( in CompileLmhloToExecutable() local
1149 return std::unique_ptr<Executable>(gpu_executable); in CompileLmhloToExecutable()
|
| D | BUILD | 219 ":gpu_executable",
255 ":gpu_executable",
553 name = "gpu_executable",
564 "gpu_executable.cc",
587 "gpu_executable.h",
744 ":gpu_executable",
772 ":gpu_executable",
1247 ":gpu_executable",
|