Home
last modified time | relevance | path

Searched refs:blob_annotation (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
H A Dgpu_kernel_to_blob_pass.cc51 GpuKernelToBlobPass(StringRef blob_annotation, in GpuKernelToBlobPass() argument
54 if (!blob_annotation.empty()) blob_annotation_ = blob_annotation.str(); in GpuKernelToBlobPass()
260 StringRef blob_annotation, ArrayRef<std::string> architectures, in CreateGpuKernelToBlobPass() argument
263 blob_annotation, architectures, print_ptx, print_llvmir, enable_ftz); in CreateGpuKernelToBlobPass()
H A Dtf_kernel_to_llvm_pass.cc247 explicit TFKernelToLLVMPass(StringRef blob_annotation) { in TFKernelToLLVMPass() argument
248 if (!blob_annotation.empty()) { in TFKernelToLLVMPass()
249 blob_annotation_ = blob_annotation.str(); in TFKernelToLLVMPass()
309 StringRef blob_annotation) { in CreateTFKernelToLLVMPass() argument
310 return std::make_unique<TFKernelToLLVMPass>(blob_annotation); in CreateTFKernelToLLVMPass()
H A Dpasses.h60 mlir::StringRef blob_annotation = {});
71 mlir::StringRef blob_annotation = {},