Searched refs:GpuCompiler (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_compiler.h | 43 class GpuCompiler : public LLVMCompiler {
45 GpuCompiler(se::Platform::Id platform_id, const char* target_triple,
47 ~GpuCompiler() override {} in ~GpuCompiler()
146 TF_DISALLOW_COPY_AND_ASSIGN(GpuCompiler);
168 GpuCompiler* compiler, mlir::ModuleOp module, std::string module_name,
|
| D | gpu_compiler.cc | 140 GpuCompiler::GpuCompiler(se::Platform::Id platform_id, in GpuCompiler() function in xla::gpu::GpuCompiler
149 Status GpuCompiler::OptimizeHloModule( in OptimizeHloModule()
359 Status GpuCompiler::PrepareHloModuleForIrEmitting(HloModule* hlo_module) { in PrepareHloModuleForIrEmitting()
402 Status GpuCompiler::OptimizeHloPostLayoutAssignment( in OptimizeHloPostLayoutAssignment()
489 StatusOr<std::unique_ptr<HloModule>> GpuCompiler::RunHloPasses( in RunHloPasses()
513 GpuCompiler::RunHloPassesAndBufferAssignement( in RunHloPassesAndBufferAssignement()
529 return GpuCompiler::GetSizeOfShape(buffer_value.shape(), pointer_size_); in RunHloPassesAndBufferAssignement()
572 return GpuCompiler::GetSizeOfShape(buffer_value.shape(), pointer_size); in CompileModuleToLlvmIrImpl()
676 GpuCompiler::CompileToTargetBinary(const HloModuleConfig& module_config, in CompileToTargetBinary()
855 StatusOr<std::unique_ptr<Executable>> GpuCompiler::RunBackend( in RunBackend()
[all …]
|
| D | amdgpu_compiler.h | 31 class AMDGPUCompiler : public GpuCompiler {
|
| D | amdgpu_compiler.cc | 92 : GpuCompiler(stream_executor::rocm::kROCmPlatformId, amdgpu::kTargetTriple, in AMDGPUCompiler()
|
| D | nvptx_compiler.cc | 172 TF_RETURN_IF_ERROR(GpuCompiler::OptimizeHloPostLayoutAssignment( in OptimizeHloPostLayoutAssignment()
280 : GpuCompiler(stream_executor::cuda::kCudaPlatformId, nvptx::kTargetTriple, in NVPTXCompiler()
|
| D | nvptx_compiler.h | 36 class NVPTXCompiler : public GpuCompiler {
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | llvm_compiler_test.cc | 40 class GpuDummyCompiler : public GpuCompiler {
42 GpuDummyCompiler() : GpuCompiler(kDummyTestId, kDummyTriple, kDummyLayout) {} in GpuDummyCompiler()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | mlir_gpu_test_base.cc | 76 static_cast<GpuCompiler*>(backend_->compiler()), module, "TestModule", in CompileMlirModule()
|