Searched refs:EmitRsqrt (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | elemental_ir_emitter.h | 76 StatusOr<llvm::Value*> EmitRsqrt(PrimitiveType prim_type,
|
| D | elemental_ir_emitter.cc | 252 StatusOr<llvm::Value*> GpuElementalIrEmitter::EmitRsqrt(PrimitiveType prim_type, in EmitRsqrt() function in xla::gpu::GpuElementalIrEmitter
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.h | 134 virtual StatusOr<llvm::Value*> EmitRsqrt(PrimitiveType prim_type,
|
| D | elemental_ir_emitter.cc | 465 return EmitRsqrt(op->shape().element_type(), operand_value); in EmitFloatUnaryOp()
965 TF_ASSIGN_OR_RETURN(llvm::Value * rsqrt_max, EmitRsqrt(prim_type, max)); in EmitRsqrtComplexAbs()
966 TF_ASSIGN_OR_RETURN(llvm::Value * rsqrt_sqrt, EmitRsqrt(prim_type, sqrt)); in EmitRsqrtComplexAbs()
968 TF_ASSIGN_OR_RETURN(llvm::Value * rsqrt_min, EmitRsqrt(prim_type, min)); in EmitRsqrtComplexAbs()
1396 StatusOr<llvm::Value*> ElementalIrEmitter::EmitRsqrt(PrimitiveType prim_type, in EmitRsqrt() function in xla::ElementalIrEmitter
|