diff options
Diffstat (limited to 'compiler/exo-tflite/src/Dialect/IR/TFLNodes.h')
-rw-r--r-- | compiler/exo-tflite/src/Dialect/IR/TFLNodes.h | 32 |
1 files changed, 30 insertions, 2 deletions
diff --git a/compiler/exo-tflite/src/Dialect/IR/TFLNodes.h b/compiler/exo-tflite/src/Dialect/IR/TFLNodes.h index f9ff2223f..42de74806 100644 --- a/compiler/exo-tflite/src/Dialect/IR/TFLNodes.h +++ b/compiler/exo-tflite/src/Dialect/IR/TFLNodes.h @@ -166,7 +166,21 @@ private: // TODO TFLDepthwiseConv2D -// TODO TFLDiv +/** + * @brief DIV in TensorFlow Lite + */ +class TFLDiv final : public FixedArityNode<2, TFLNodeImpl<TFLOpcode::DIV>> +{ +public: + TFLDiv() = default; + +public: + loco::Node *x(void) const { return at(0)->node(); } + void x(loco::Node *node) { at(0)->node(node); } + + loco::Node *y(void) const { return at(1)->node(); } + void y(loco::Node *node) { at(1)->node(node); } +}; /** * @brief MAX_POOL_2D in TensorFlow Lite @@ -234,7 +248,21 @@ public: // TODO TFLSqrt -// TODO TFLSub +/** + * @brief SUB in TensorFlow Lite + */ +class TFLSub final : public FixedArityNode<2, TFLNodeImpl<TFLOpcode::SUB>> +{ +public: + TFLSub() = default; + +public: + loco::Node *x(void) const { return at(0)->node(); } + void x(loco::Node *node) { at(0)->node(node); } + + loco::Node *y(void) const { return at(1)->node(); } + void y(loco::Node *node) { at(1)->node(node); } +}; // TODO TFLTanh |