_graph.addOperation(std::move(new_op));
}
+void Loader::loadTransposeConv(const tflite::Operator *op)
+{
+ model::OperandIndexSequence inputs;
+ model::OperandIndexSequence outputs;
+
+ loadOperationIO(op, inputs, outputs);
+
+ model::operation::TransposeConvNode::Param param;
+ const auto *options = op->builtin_options_as_TransposeConvOptions();
+ loadStridesAndPaddings(param, options);
+ std::unique_ptr<model::Operation> new_op(
+ new model::operation::TransposeConvNode(inputs, outputs, param));
+ _graph.addOperation(std::move(new_op));
+}
+
void Loader::loadAvgPool2D(const tflite::Operator *op)
{
model::OperandIndexSequence inputs;
case BuiltinOperator_DEPTHWISE_CONV_2D:
loadDepthwiseConv2D(op);
return;
+ case BuiltinOperator_TRANSPOSE_CONV:
+ loadTransposeConv(op);
+ return;
case BuiltinOperator_RESHAPE:
loadReshape(op);
return;
// Operations
void loadConv2D(const tflite::Operator *op);
void loadDepthwiseConv2D(const tflite::Operator *op);
+ void loadTransposeConv(const tflite::Operator *op);
void loadAvgPool2D(const tflite::Operator *op);
void loadReshape(const tflite::Operator *op);
void loadSoftmax(const tflite::Operator *op);