Searched refs:kAtan2 (Results 1 – 18 of 18) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | target_util.h | 49 kAtan2 = 0, enumerator
|
| D | target_util.cc | 114 case TargetDeviceFunctionID::kAtan2: { in GetDeviceFunctionRoot()
|
| D | elemental_ir_emitter.cc | 261 return EmitDeviceMathCall(TargetDeviceFunctionID::kAtan2, {lhs, rhs}, in EmitAtan2()
|
| D | ir_emitter_unnested.cc | 204 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/ |
| D | hlo_opcode.h | 54 V(kAtan2, "atan2", 2) \
|
| D | instruction_fusion.cc | 142 case HloOpcode::kAtan2: in IsExpensive()
|
| D | hlo_instruction.cc | 930 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()
|
| D | hlo_cost_analysis.cc | 110 opcode == HloOpcode::kAtan2) { in HandleElementwiseOp()
|
| D | hlo_graph_dumper.cc | 942 case HloOpcode::kAtan2: in GetInstructionColor()
|
| D | sharding_propagation.cc | 168 case HloOpcode::kAtan2: in PickRepresentativeOperand()
|
| D | elemental_ir_emitter.cc | 880 case HloOpcode::kAtan2: in EmitFloatBinaryOp()
2280 case HloOpcode::kAtan2: in MakeElementGenerator()
|
| D | hlo_parser.cc | 87 case HloOpcode::kAtan2: in CanInferShape()
1129 case HloOpcode::kAtan2: in ParseInstructionRhs()
|
| D | layout_assignment.cc | 2211 case HloOpcode::kAtan2: in InstructionCanChangeLayout()
|
| D | shape_inference.cc | 982 case HloOpcode::kAtan2: in InferBinaryOpShape()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.cc | 296 return xla::HloOpcode::kAtan2; in MhloToHloOpcode()
|
| D | hlo_function_importer.cc | 749 NoAttributeCase(kAtan2, Atan2Op); in ImportInstructionImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 260 case HloOpcode::kAtan2: in EmitOp()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 3508 case HloOpcode::kAtan2: in BuildDynamicInferenceGraph()
4729 return lhs.builder()->BinaryOp(HloOpcode::kAtan2, lhs, rhs, in Atan2()
|