Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dtarget_util.cc112 TargetDeviceFunctionID func_id) { in GetDeviceFunctionRoot()
114 case TargetDeviceFunctionID::kAtan2: { in GetDeviceFunctionRoot()
117 case TargetDeviceFunctionID::kCos: { in GetDeviceFunctionRoot()
120 case TargetDeviceFunctionID::kErfcinv: { in GetDeviceFunctionRoot()
123 case TargetDeviceFunctionID::kExp: { in GetDeviceFunctionRoot()
126 case TargetDeviceFunctionID::kExpm1: { in GetDeviceFunctionRoot()
129 case TargetDeviceFunctionID::kFmod: { in GetDeviceFunctionRoot()
132 case TargetDeviceFunctionID::kHypot: { in GetDeviceFunctionRoot()
135 case TargetDeviceFunctionID::kLog: { in GetDeviceFunctionRoot()
138 case TargetDeviceFunctionID::kLog1p: { in GetDeviceFunctionRoot()
[all …]
Delemental_ir_emitter.cc80 TargetDeviceFunctionID funcid, absl::Span<llvm::Value* const> operands, in EmitDeviceMathCall()
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 …]
Dtarget_util.h48 enum class TargetDeviceFunctionID { enum
86 std::string ObtainDeviceFunctionName(TargetDeviceFunctionID func_id,
Delemental_ir_emitter.h122 TargetDeviceFunctionID funcid, absl::Span<llvm::Value* const> operands,