Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dtarget_util.cc113 TargetDeviceFunctionID func_id) { in GetDeviceFunctionRoot()
115 case TargetDeviceFunctionID::kAtan2: { in GetDeviceFunctionRoot()
118 case TargetDeviceFunctionID::kCos: { in GetDeviceFunctionRoot()
121 case TargetDeviceFunctionID::kErfcinv: { in GetDeviceFunctionRoot()
124 case TargetDeviceFunctionID::kExp: { in GetDeviceFunctionRoot()
127 case TargetDeviceFunctionID::kExpm1: { in GetDeviceFunctionRoot()
130 case TargetDeviceFunctionID::kFmod: { in GetDeviceFunctionRoot()
133 case TargetDeviceFunctionID::kHypot: { in GetDeviceFunctionRoot()
136 case TargetDeviceFunctionID::kLog: { in GetDeviceFunctionRoot()
139 case TargetDeviceFunctionID::kLog1p: { in GetDeviceFunctionRoot()
[all …]
H A Delemental_ir_emitter.cc79 TargetDeviceFunctionID funcid, absl::Span<llvm::Value* const> operands, in EmitDeviceMathCall()
195 return EmitDeviceMathCall(TargetDeviceFunctionID::kFmod, in EmitFloatBinaryOp()
213 return EmitDeviceMathCall(TargetDeviceFunctionID::kPow, in EmitPowerOp()
220 return EmitDeviceMathCall(TargetDeviceFunctionID::kLog, {value}, {prim_type}, in EmitLog()
226 return EmitDeviceMathCall(TargetDeviceFunctionID::kLog1p, {value}, in EmitLog1p()
232 return EmitDeviceMathCall(TargetDeviceFunctionID::kSin, {value}, {prim_type}, in EmitSin()
238 return EmitDeviceMathCall(TargetDeviceFunctionID::kCos, {value}, {prim_type}, in EmitCos()
244 return EmitDeviceMathCall(TargetDeviceFunctionID::kExp, {value}, {prim_type}, in EmitExp()
250 return EmitDeviceMathCall(TargetDeviceFunctionID::kExpm1, {value}, in EmitExpm1()
258 return EmitDeviceMathCall(TargetDeviceFunctionID::kPow, {lhs, rhs}, in EmitPow()
[all …]
H A Dtarget_util.h48 enum class TargetDeviceFunctionID { enum
86 std::string ObtainDeviceFunctionName(TargetDeviceFunctionID func_id,
H A Delemental_ir_emitter.h125 TargetDeviceFunctionID funcid, absl::Span<llvm::Value* const> operands,