_graph.addOperation(std::move(new_op));
}
+void Loader::loadRelu6(const tflite::Operator *op)
+{
+ model::OperandIndexSequence inputs;
+ model::OperandIndexSequence outputs;
+
+ loadOperationIO(op, inputs, outputs);
+
+ std::unique_ptr<model::Operation> new_op(new model::operation::ReLU6Node(inputs, outputs));
+ _graph.addOperation(std::move(new_op));
+}
+
void Loader::loadTranspose(const tflite::Operator *op)
{
model::OperandIndexSequence inputs;
case BuiltinOperator_RELU:
loadRelu(op);
return;
+ case BuiltinOperator_RELU6:
+ loadRelu6(op);
+ return;
case BuiltinOperator_TRANSPOSE:
loadTranspose(op);
return;
void loadFC(const tflite::Operator *op);
void loadAdd(const tflite::Operator *op);
void loadRelu(const tflite::Operator *op);
+ void loadRelu6(const tflite::Operator *op);
void loadTranspose(const tflite::Operator *op);
private: