Home
last modified time | relevance | path

Searched defs:gpu_module (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dgpu_kernel_to_blob_pass.cc67 mlir::gpu::GPUModuleOp gpu_module = getOperation(); in runOnOperation() local
82 mlir::gpu::GPUModuleOp gpu_module) { in GetGpuBinaryBlob()
/external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/
Dpasses.cc253 auto gpu_module = kernel->getParentOfType<mlir::gpu::GPUModuleOp>(); in runOnFunction() local