Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dtarget_constants.h28 constexpr char kDataLayout[] = "e-i64:64-i128:128-v16:16-v32:32-n16:32:64"; variable
37 constexpr char kDataLayout[] = variable
Dgpu_transfer_manager.cc150 /*pointer_size=*/llvm::DataLayout(xla::gpu::nvptx::kDataLayout) in CreateNVPTXTransferManager()
157 /*pointer_size=*/llvm::DataLayout(xla::gpu::amdgpu::kDataLayout) in CreateAMDGPUTransferManager()
Damdgpu_compiler.cc93 amdgpu::kDataLayout) {} in AMDGPUCompiler()
Dnvptx_compiler.cc281 nvptx::kDataLayout) {} in NVPTXCompiler()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dmlir_gpu_test_base.cc47 llvm_module->setDataLayout(amdgpu::kDataLayout); in CompileMlirModule()
50 llvm_module->setDataLayout(nvptx::kDataLayout); in CompileMlirModule()
Dhlo_to_llvm_ir.cc62 /*data_layout=*/xla::gpu::nvptx::kDataLayout, in CompileAndPrintLlvmIr()
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dgpu_kernel_to_blob_pass.cc151 llvmModule->setDataLayout(xla::gpu::nvptx::kDataLayout); in GetGpuBinaryBlob()