Home
last modified time | relevance | path

Searched refs:kAtan2 (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dtarget_util.h49 kAtan2 = 0, enumerator
Dtarget_util.cc114 case TargetDeviceFunctionID::kAtan2: { in GetDeviceFunctionRoot()
Delemental_ir_emitter.cc261 return EmitDeviceMathCall(TargetDeviceFunctionID::kAtan2, {lhs, rhs}, in EmitAtan2()
Dir_emitter_unnested.cc204 case HloOpcode::kAtan2: in MayPreventVectorization()
221 case HloOpcode::kAtan2: in MayPreventVectorization()
276 case HloOpcode::kAtan2: in MayPreventVectorization()
297 case HloOpcode::kAtan2: in MayPreventVectorization()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_opcode.h54 V(kAtan2, "atan2", 2) \
Dinstruction_fusion.cc142 case HloOpcode::kAtan2: in IsExpensive()
Dhlo_instruction.cc930 case HloOpcode::kAtan2: in CreateBinary()
1746 case HloOpcode::kAtan2: in CloneWithNewOperands()
2121 case HloOpcode::kAtan2: in IdenticalSlowPath()
2644 case HloOpcode::kAtan2: in IsOpElementwise()
3060 case HloOpcode::kAtan2: in Visit()
Dhlo_cost_analysis.cc110 opcode == HloOpcode::kAtan2) { in HandleElementwiseOp()
Dhlo_graph_dumper.cc942 case HloOpcode::kAtan2: in GetInstructionColor()
Dsharding_propagation.cc168 case HloOpcode::kAtan2: in PickRepresentativeOperand()
Delemental_ir_emitter.cc880 case HloOpcode::kAtan2: in EmitFloatBinaryOp()
2280 case HloOpcode::kAtan2: in MakeElementGenerator()
Dhlo_parser.cc87 case HloOpcode::kAtan2: in CanInferShape()
1129 case HloOpcode::kAtan2: in ParseInstructionRhs()
Dlayout_assignment.cc2211 case HloOpcode::kAtan2: in InstructionCanChangeLayout()
Dshape_inference.cc982 case HloOpcode::kAtan2: in InferBinaryOpShape()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_utils.cc296 return xla::HloOpcode::kAtan2; in MhloToHloOpcode()
Dhlo_function_importer.cc749 NoAttributeCase(kAtan2, Atan2Op); in ImportInstructionImpl()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc260 case HloOpcode::kAtan2: in EmitOp()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc3508 case HloOpcode::kAtan2: in BuildDynamicInferenceGraph()
4729 return lhs.builder()->BinaryOp(HloOpcode::kAtan2, lhs, rhs, in Atan2()