Searched refs:EmitCos (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | elemental_ir_emitter.h | 64 StatusOr<llvm::Value*> EmitCos(PrimitiveType prim_type,
|
D | elemental_ir_emitter.cc | 220 StatusOr<llvm::Value*> GpuElementalIrEmitter::EmitCos(PrimitiveType prim_type, in EmitCos() function in xla::gpu::GpuElementalIrEmitter
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | elemental_ir_emitter.cc | 457 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
|
D | elemental_ir_emitter.h | 143 virtual StatusOr<llvm::Value*> EmitCos(PrimitiveType prim_type,
|