Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dtf_kernel_to_llvm_pass.cc55 gpu_binary_annotation_(gpu_binary_annotation) {} in ConvertLaunchFuncOpToTfRuntimeCallPattern()
80 llvm::SmallString<32> gpu_binary_annotation_; member in mlir::kernel_gen::transforms::__anon1a4b4e700111::ConvertLaunchFuncOpToTfRuntimeCallPattern
161 kernel_module->getAttrOfType<StringAttr>(gpu_binary_annotation_); in matchAndRewrite()
164 << "missing " << gpu_binary_annotation_ << " attribute"; in matchAndRewrite()