_graph.addOperation(std::move(new_op));
}
+void Loader::loadSquaredDifference(const tflite::Operator *op)
+{
+ model::OperandIndexSequence inputs;
+ model::OperandIndexSequence outputs;
+
+ loadOperationIO(op, inputs, outputs);
+
+ std::unique_ptr<model::Operation> new_op(
+ new model::operation::SquaredDifferenceNode(inputs, outputs));
+ _graph.addOperation(std::move(new_op));
+}
+
void Loader::loadTanh(const tflite::Operator *op)
{
model::OperandIndexSequence inputs;
case BuiltinOperator_RELU6:
loadRelu6(op);
return;
+ case BuiltinOperator_RSQRT:
+ loadRsqrt(op);
+ return;
+ case BuiltinOperator_SQUARED_DIFFERENCE:
+ loadSquaredDifference(op);
+ return;
case BuiltinOperator_TANH:
loadTanh(op);
return;
case BuiltinOperator_TRANSPOSE:
loadTranspose(op);
return;
- case BuiltinOperator_RSQRT:
- loadRsqrt(op);
- return;
default:
auto *names = EnumNamesBuiltinOperator();
int enum_value = static_cast<int>(_op_code_to_builtin_op[op->opcode_index()]);
void loadRelu(const tflite::Operator *op);
void loadRelu6(const tflite::Operator *op);
void loadRsqrt(const tflite::Operator *op);
- void loadTranspose(const tflite::Operator *op);
+ void loadSquaredDifference(const tflite::Operator *op);
void loadTanh(const tflite::Operator *op);
+ void loadTranspose(const tflite::Operator *op);
private:
// Buffer for loading (if needed)