Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
H A Dgpu_kernel_to_blob_pass.cc62 gpu::GPUModuleOp gpu_module = getOperation(); in runOnOperation() local
77 gpu::GPUModuleOp gpu_module) { in GetGpuBinaryBlob()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/
H A Dkernel_ops_pattern.cc341 auto gpu_module = rewriter.create<mlir::gpu::GPUModuleOp>(loc, "gpu_module"); in Rewrite() local
H A Dlmhlo_to_jitrt.cc373 auto gpu_module = kernel->getParentOfType<mlir::gpu::GPUModuleOp>(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter_unnested.cc1600 auto gpu_module = kernel_func->getParentOfType<mlir::gpu::GPUModuleOp>(); in EmitLaunchFunc() local