Searched refs:OpToFuncCallLowering (Results 1 – 4 of 4) sorted by relevance
| /external/llvm-project/mlir/lib/Conversion/GPUToROCDL/ |
| D | LowerGpuOpsToROCDLOps.cpp | 100 patterns.insert<OpToFuncCallLowering<AbsFOp>>(converter, "__ocml_fabs_f32", in populateGpuToROCDLConversionPatterns()
102 patterns.insert<OpToFuncCallLowering<CeilFOp>>(converter, "__ocml_ceil_f32", in populateGpuToROCDLConversionPatterns()
104 patterns.insert<OpToFuncCallLowering<CosOp>>(converter, "__ocml_cos_f32", in populateGpuToROCDLConversionPatterns()
106 patterns.insert<OpToFuncCallLowering<ExpOp>>(converter, "__ocml_exp_f32", in populateGpuToROCDLConversionPatterns()
108 patterns.insert<OpToFuncCallLowering<FloorFOp>>(converter, "__ocml_floor_f32", in populateGpuToROCDLConversionPatterns()
110 patterns.insert<OpToFuncCallLowering<LogOp>>(converter, "__ocml_log_f32", in populateGpuToROCDLConversionPatterns()
112 patterns.insert<OpToFuncCallLowering<Log10Op>>(converter, "__ocml_log10_f32", in populateGpuToROCDLConversionPatterns()
114 patterns.insert<OpToFuncCallLowering<Log2Op>>(converter, "__ocml_log2_f32", in populateGpuToROCDLConversionPatterns()
116 patterns.insert<OpToFuncCallLowering<SinOp>>(converter, "__ocml_sin_f32", in populateGpuToROCDLConversionPatterns()
118 patterns.insert<OpToFuncCallLowering<TanhOp>>(converter, "__ocml_tanh_f32", in populateGpuToROCDLConversionPatterns()
|
| /external/llvm-project/mlir/lib/Conversion/GPUToNVVM/ |
| D | LowerGpuOpsToNVVMOps.cpp | 176 patterns.insert<OpToFuncCallLowering<AbsFOp>>(converter, "__nv_fabsf", in populateGpuToNVVMConversionPatterns()
178 patterns.insert<OpToFuncCallLowering<CeilFOp>>(converter, "__nv_ceilf", in populateGpuToNVVMConversionPatterns()
180 patterns.insert<OpToFuncCallLowering<CosOp>>(converter, "__nv_cosf", in populateGpuToNVVMConversionPatterns()
182 patterns.insert<OpToFuncCallLowering<ExpOp>>(converter, "__nv_expf", in populateGpuToNVVMConversionPatterns()
184 patterns.insert<OpToFuncCallLowering<FloorFOp>>(converter, "__nv_floorf", in populateGpuToNVVMConversionPatterns()
186 patterns.insert<OpToFuncCallLowering<LogOp>>(converter, "__nv_logf", in populateGpuToNVVMConversionPatterns()
188 patterns.insert<OpToFuncCallLowering<Log10Op>>(converter, "__nv_log10f", in populateGpuToNVVMConversionPatterns()
190 patterns.insert<OpToFuncCallLowering<Log2Op>>(converter, "__nv_log2f", in populateGpuToNVVMConversionPatterns()
192 patterns.insert<OpToFuncCallLowering<RsqrtOp>>(converter, "__nv_rsqrtf", in populateGpuToNVVMConversionPatterns()
194 patterns.insert<OpToFuncCallLowering<SinOp>>(converter, "__nv_sinf", in populateGpuToNVVMConversionPatterns()
[all …]
|
| /external/llvm-project/mlir/lib/Conversion/GPUCommon/ |
| D | OpToFuncCallLowering.h | 32 struct OpToFuncCallLowering : public ConvertToLLVMPattern { struct
34 explicit OpToFuncCallLowering(LLVMTypeConverter &lowering_, StringRef f32Func, in OpToFuncCallLowering() argument
|
| /external/tensorflow/third_party/mlir/ |
| D | BUILD | 1890 "lib/Conversion/GPUCommon/OpToFuncCallLowering.h",
|