Searched refs:EmitDeviceMathCall (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | elemental_ir_emitter.cc | 79 StatusOr<llvm::Value*> GpuElementalIrEmitter::EmitDeviceMathCall( in EmitDeviceMathCall() function in xla::gpu::GpuElementalIrEmitter
179 return EmitDeviceMathCall(TargetDeviceFunctionID::kFmod, in EmitFloatBinaryOp()
197 return EmitDeviceMathCall(TargetDeviceFunctionID::kPow, in EmitPowerOp()
204 return EmitDeviceMathCall(TargetDeviceFunctionID::kLog, {value}, {prim_type}, in EmitLog()
210 return EmitDeviceMathCall(TargetDeviceFunctionID::kLog1p, {value}, in EmitLog1p()
216 return EmitDeviceMathCall(TargetDeviceFunctionID::kSin, {value}, {prim_type}, in EmitSin()
222 return EmitDeviceMathCall(TargetDeviceFunctionID::kCos, {value}, {prim_type}, in EmitCos()
228 return EmitDeviceMathCall(TargetDeviceFunctionID::kExp, {value}, {prim_type}, in EmitExp()
234 return EmitDeviceMathCall(TargetDeviceFunctionID::kExpm1, {value}, in EmitExpm1()
242 return EmitDeviceMathCall(TargetDeviceFunctionID::kPow, {lhs, rhs}, in EmitPow()
[all …]
|
| D | elemental_ir_emitter.h | 121 StatusOr<llvm::Value*> EmitDeviceMathCall(
|