Searched refs:EmitSqrt (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | elemental_ir_emitter.h | 73 StatusOr<llvm::Value*> EmitSqrt(PrimitiveType prim_type,
|
| D | elemental_ir_emitter.cc | 246 StatusOr<llvm::Value*> GpuElementalIrEmitter::EmitSqrt(PrimitiveType prim_type, in EmitSqrt() function in xla::gpu::GpuElementalIrEmitter
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.h | 128 virtual StatusOr<llvm::Value*> EmitSqrt(PrimitiveType prim_type,
|
| D | elemental_ir_emitter.cc | 463 return EmitSqrt(op->shape().element_type(), operand_value); in EmitFloatUnaryOp()
917 TF_ASSIGN_OR_RETURN(llvm::Value * sqrt, EmitSqrt(prim_type, one_p_div_sq)); in EmitComplexAbsHelper()
945 TF_ASSIGN_OR_RETURN(llvm::Value * sqrt_max, EmitSqrt(prim_type, max)); in EmitSqrtComplexAbs()
1390 StatusOr<llvm::Value*> ElementalIrEmitter::EmitSqrt(PrimitiveType, in EmitSqrt() function in xla::ElementalIrEmitter
1398 TF_ASSIGN_OR_RETURN(auto sqrt, EmitSqrt(prim_type, value)); in EmitRsqrt()
|