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.cc54 GpuKernelToBlobPass(mlir::StringRef blob_annotation, in GpuKernelToBlobPass() argument
57 if (!blob_annotation.empty()) { in GpuKernelToBlobPass()
58 blob_annotation_ = blob_annotation.str(); in GpuKernelToBlobPass()
255 mlir::StringRef blob_annotation, ArrayRef<std::string> architectures, in CreateGpuKernelToBlobPass() argument
258 blob_annotation, architectures, generate_fatbin, print_ptx, enable_ftz); in CreateGpuKernelToBlobPass()
Dtf_kernel_to_llvm_pass.cc238 explicit TFKernelToLLVMPass(StringRef blob_annotation) { in TFKernelToLLVMPass() argument
239 if (!blob_annotation.empty()) { in TFKernelToLLVMPass()
240 blob_annotation_ = blob_annotation.str(); in TFKernelToLLVMPass()
290 StringRef blob_annotation) { in CreateTFKernelToLLVMPass() argument
291 return std::make_unique<TFKernelToLLVMPass>(blob_annotation); in CreateTFKernelToLLVMPass()
Dpasses.h51 mlir::StringRef blob_annotation = {});
70 mlir::StringRef blob_annotation = {},