Home
last modified time | relevance | path

Searched refs:EmitAtan2 (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Delemental_ir_emitter.h42 StatusOr<llvm::Value*> EmitAtan2(PrimitiveType prim_type, llvm::Value* lhs,
Delemental_ir_emitter.cc35 StatusOr<llvm::Value*> CpuElementalIrEmitter::EmitAtan2(PrimitiveType prim_type, in EmitAtan2() function in xla::cpu::CpuElementalIrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.h88 StatusOr<llvm::Value*> EmitAtan2(PrimitiveType prim_type, llvm::Value* lhs,
Delemental_ir_emitter.cc251 StatusOr<llvm::Value*> GpuElementalIrEmitter::EmitAtan2(PrimitiveType prim_type, in EmitAtan2() function in xla::gpu::GpuElementalIrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.h116 virtual StatusOr<llvm::Value*> EmitAtan2(PrimitiveType prim_type,
Delemental_ir_emitter.cc491 TF_ASSIGN_OR_RETURN(auto angle, EmitAtan2(component_type, b, a)); in EmitComplexUnaryOp()
504 TF_ASSIGN_OR_RETURN(auto angle, EmitAtan2(component_type, b, a_plus_one)); in EmitComplexUnaryOp()
750 return EmitAtan2(op->shape().element_type(), lhs_value, rhs_value); in EmitFloatBinaryOp()
775 TF_ASSIGN_OR_RETURN(auto arg_lhs, EmitAtan2(component_type, b, a)); in EmitComplexPower()
1150 StatusOr<llvm::Value*> ElementalIrEmitter::EmitAtan2(PrimitiveType prim_type, in EmitAtan2() function in xla::ElementalIrEmitter