Searched refs:EmitLibdeviceMathCall (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | elemental_ir_emitter.cc | 77 StatusOr<llvm::Value*> GpuElementalIrEmitter::EmitLibdeviceMathCall( in EmitLibdeviceMathCall() function in xla::gpu::GpuElementalIrEmitter
177 return EmitLibdeviceMathCall("__nv_fmod", {lhs_value, rhs_value}, in EmitFloatBinaryOp()
195 return EmitLibdeviceMathCall("__nv_pow", {lhs_value, rhs_value}, in EmitPowerOp()
201 return EmitLibdeviceMathCall("__nv_erfcinv", {value}, {prim_type}, prim_type); in EmitErfcInv()
206 return EmitLibdeviceMathCall("__nv_log", {value}, {prim_type}, prim_type); in EmitLog()
211 return EmitLibdeviceMathCall("__nv_log1p", {value}, {prim_type}, prim_type); in EmitLog1p()
216 return EmitLibdeviceMathCall("__nv_sin", {value}, {prim_type}, prim_type); in EmitSin()
221 return EmitLibdeviceMathCall("__nv_cos", {value}, {prim_type}, prim_type); in EmitCos()
226 return EmitLibdeviceMathCall("__nv_exp", {value}, {prim_type}, prim_type); in EmitExp()
231 return EmitLibdeviceMathCall("__nv_expm1", {value}, {prim_type}, prim_type); in EmitExpm1()
[all …]
|
| D | elemental_ir_emitter.h | 122 StatusOr<llvm::Value*> EmitLibdeviceMathCall(
|