Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dgpu_kernel_to_blob_pass.cc53 GpuKernelToBlobPass(mlir::StringRef blob_annotation, in GpuKernelToBlobPass() argument
56 if (!blob_annotation.empty()) { in GpuKernelToBlobPass()
57 blob_annotation_ = blob_annotation.str(); in GpuKernelToBlobPass()
233 mlir::StringRef blob_annotation, ArrayRef<std::string> architectures, in CreateGpuKernelToBlobPass() argument
236 blob_annotation, architectures, print_ptx, print_llvmir, enable_ftz); in CreateGpuKernelToBlobPass()
Dtf_kernel_to_llvm_pass.cc243 explicit TFKernelToLLVMPass(StringRef blob_annotation) { in TFKernelToLLVMPass() argument
244 if (!blob_annotation.empty()) { in TFKernelToLLVMPass()
245 blob_annotation_ = blob_annotation.str(); in TFKernelToLLVMPass()
301 StringRef blob_annotation) { in CreateTFKernelToLLVMPass() argument
302 return std::make_unique<TFKernelToLLVMPass>(blob_annotation); in CreateTFKernelToLLVMPass()
Dpasses.h59 mlir::StringRef blob_annotation = {});
81 mlir::StringRef blob_annotation = {},