Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.h79 StatusOr<llvm::Value*> EmitSqrt(PrimitiveType prim_type,
Delemental_ir_emitter.cc241 StatusOr<llvm::Value*> GpuElementalIrEmitter::EmitSqrt(PrimitiveType prim_type, in EmitSqrt() function in xla::gpu::GpuElementalIrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.h122 virtual StatusOr<llvm::Value*> EmitSqrt(PrimitiveType prim_type,
Delemental_ir_emitter.cc424 return EmitSqrt(op->shape().element_type(), operand_value); in EmitFloatUnaryOp()
1084 StatusOr<llvm::Value*> ElementalIrEmitter::EmitSqrt(PrimitiveType prim_type, in EmitSqrt() function in xla::ElementalIrEmitter
1092 TF_ASSIGN_OR_RETURN(auto sqrt, EmitSqrt(prim_type, value)); in EmitRsqrt()