_graph.addOperation(std::move(new_op));
}
+void Loader::loadTanh(const tflite::Operator *op)
+{
+ model::OperandIndexSequence inputs;
+ model::OperandIndexSequence outputs;
+
+ loadOperationIO(op, inputs, outputs);
+
+ std::unique_ptr<model::Operation> new_op(new model::operation::TanhNode(inputs, outputs));
+ _graph.addOperation(std::move(new_op));
+}
+
void Loader::loadTranspose(const tflite::Operator *op)
{
model::OperandIndexSequence inputs;
case BuiltinOperator_RELU6:
loadRelu6(op);
return;
+ case BuiltinOperator_TANH:
+ loadTanh(op);
+ return;
case BuiltinOperator_TRANSPOSE:
loadTranspose(op);
return;
void loadSub(const tflite::Operator *op);
void loadRelu(const tflite::Operator *op);
void loadRelu6(const tflite::Operator *op);
- void loadTranspose(const tflite::Operator *op);
void loadRsqrt(const tflite::Operator *op);
+ void loadTranspose(const tflite::Operator *op);
+ void loadTanh(const tflite::Operator *op);
private:
// Buffer for loading (if needed)