diff options
Diffstat (limited to 'tensorflow/compiler/xla/service/elemental_ir_emitter.h')
| -rw-r--r-- | tensorflow/compiler/xla/service/elemental_ir_emitter.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/elemental_ir_emitter.h b/tensorflow/compiler/xla/service/elemental_ir_emitter.h index 26dff0d96f..d199473374 100644 --- a/tensorflow/compiler/xla/service/elemental_ir_emitter.h +++ b/tensorflow/compiler/xla/service/elemental_ir_emitter.h @@ -105,6 +105,9 @@ class ElementalIrEmitter { virtual StatusOr<llvm::Value*> EmitLog(PrimitiveType prim_type, llvm::Value* value) const; + virtual StatusOr<llvm::Value*> EmitLog1p(PrimitiveType prim_type, + llvm::Value* value) const; + virtual StatusOr<llvm::Value*> EmitSin(PrimitiveType prim_type, llvm::Value* value) const; @@ -114,6 +117,9 @@ class ElementalIrEmitter { virtual StatusOr<llvm::Value*> EmitExp(PrimitiveType prim_type, llvm::Value* value) const; + virtual StatusOr<llvm::Value*> EmitExpm1(PrimitiveType prim_type, + llvm::Value* value) const; + virtual StatusOr<llvm::Value*> EmitPow(PrimitiveType prim_type, llvm::Value* lhs, llvm::Value* rhs) const; |