Searched refs:generate_fatbin (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | gpu_kernel_to_blob_pass.cc | 56 bool generate_fatbin, bool print_ptx, bool enable_ftz) { in GpuKernelToBlobPass() argument
61 generate_fatbin_ = generate_fatbin; in GpuKernelToBlobPass()
256 bool generate_fatbin, bool print_ptx, bool enable_ftz) { in CreateGpuKernelToBlobPass() argument
258 blob_annotation, architectures, generate_fatbin, print_ptx, enable_ftz); in CreateGpuKernelToBlobPass()
|
| D | passes.h | 71 ArrayRef<std::string> architectures = {}, bool generate_fatbin = true,
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| D | kernel_creator.cc | 395 bool generate_fatbin, bool print_ptx, in GenerateDeviceCode() argument
405 gpu_binary_attr_name, architectures, generate_fatbin, print_ptx, in GenerateDeviceCode()
433 bool embed_memref_prints, bool generate_fatbin, bool print_ptx, in GenerateKernelForTfCode() argument
455 architectures, generate_fatbin, in GenerateKernelForTfCode()
|
| D | kernel_creator.h | 42 bool embed_memref_prints = false, bool generate_fatbin = true,
|