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.h39 StatusOr<llvm::Value*> EmitAtan2(PrimitiveType prim_type, llvm::Value* lhs,
Delemental_ir_emitter.cc36 StatusOr<llvm::Value*> CpuElementalIrEmitter::EmitAtan2( in EmitAtan2() function in xla::cpu::CpuElementalIrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.h83 StatusOr<llvm::Value*> EmitAtan2(PrimitiveType prim_type, llvm::Value* lhs,
Delemental_ir_emitter.cc258 StatusOr<llvm::Value*> GpuElementalIrEmitter::EmitAtan2( in EmitAtan2() function in xla::gpu::GpuElementalIrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.h121 virtual StatusOr<llvm::Value*> EmitAtan2(PrimitiveType prim_type,
Delemental_ir_emitter.cc532 EmitAtan2(component_type, b, a, "")); in EmitComplexUnaryOp()
548 EmitAtan2(component_type, b, a_plus_one, "")); in EmitComplexUnaryOp()
881 return EmitAtan2(op->shape().element_type(), lhs_value, rhs_value, in EmitFloatBinaryOp()
992 TF_ASSIGN_OR_RETURN(llvm::Value * t, EmitAtan2(prim_type, b, a, "")); in EmitComplexSqrt()
1048 TF_ASSIGN_OR_RETURN(llvm::Value * t, EmitAtan2(prim_type, b, a, "")); in EmitComplexRsqrt()
1128 TF_ASSIGN_OR_RETURN(auto arg_lhs, EmitAtan2(component_type, b, a, "")); in EmitComplexPower()
1469 StatusOr<llvm::Value*> ElementalIrEmitter::EmitAtan2( in EmitAtan2() function in xla::ElementalIrEmitter