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()
|