Home
last modified time | relevance | path

Searched refs:EmitCos (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.h64 StatusOr<llvm::Value*> EmitCos(PrimitiveType prim_type,
Delemental_ir_emitter.cc220 StatusOr<llvm::Value*> GpuElementalIrEmitter::EmitCos(PrimitiveType prim_type, in EmitCos() function in xla::gpu::GpuElementalIrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc457 return EmitCos(op->shape().element_type(), operand_value); in EmitFloatUnaryOp()
574 auto cos_b, EmitCos(component_type, EmitExtractImag(operand_value))); in EmitComplexUnaryOp()
585 auto cos_b, EmitCos(component_type, EmitExtractImag(operand_value))); in EmitComplexUnaryOp()
607 TF_ASSIGN_OR_RETURN(auto cos_a, EmitCos(component_type, a)); in EmitComplexUnaryOp()
629 TF_ASSIGN_OR_RETURN(auto cos_a, EmitCos(component_type, a)); in EmitComplexUnaryOp()
691 TF_ASSIGN_OR_RETURN(llvm::Value * cos_b, EmitCos(component_type, b)); in EmitComplexUnaryOp()
996 TF_ASSIGN_OR_RETURN(llvm::Value * cos, EmitCos(prim_type, angle)); in EmitComplexSqrt()
1052 TF_ASSIGN_OR_RETURN(llvm::Value * cos, EmitCos(prim_type, angle)); in EmitComplexRsqrt()
1136 TF_ASSIGN_OR_RETURN(auto cos_q, EmitCos(component_type, q)); in EmitComplexPower()
1408 StatusOr<llvm::Value*> ElementalIrEmitter::EmitCos(PrimitiveType prim_type, in EmitCos() function in xla::ElementalIrEmitter
Delemental_ir_emitter.h143 virtual StatusOr<llvm::Value*> EmitCos(PrimitiveType prim_type,