未验证 提交 8b9d4692 编写于 作者: M Mikael Johansson 提交者: GitHub

Added SQUARED_DIFFERENCE to ops resolver

Add support for the SQUARED_DIFFERENCE op in the ops resolvers

BUG=b/238912370
上级 fed37847
...@@ -102,6 +102,7 @@ AllOpsResolver::AllOpsResolver() { ...@@ -102,6 +102,7 @@ AllOpsResolver::AllOpsResolver() {
AddSplitV(); AddSplitV();
AddSqrt(); AddSqrt();
AddSquare(); AddSquare();
AddSquaredDifference();
AddSqueeze(); AddSqueeze();
AddStridedSlice(); AddStridedSlice();
AddSub(); AddSub();
......
...@@ -519,6 +519,12 @@ class MicroMutableOpResolver : public MicroOpResolver { ...@@ -519,6 +519,12 @@ class MicroMutableOpResolver : public MicroOpResolver {
tflite::ops::micro::Register_SQUARE(), ParseSquare); tflite::ops::micro::Register_SQUARE(), ParseSquare);
} }
TfLiteStatus AddSquaredDifference() {
return AddBuiltin(BuiltinOperator_SQUARED_DIFFERENCE,
tflite::Register_SQUARED_DIFFERENCE(),
ParseSquaredDifference);
}
TfLiteStatus AddStridedSlice() { TfLiteStatus AddStridedSlice() {
return AddBuiltin(BuiltinOperator_STRIDED_SLICE, return AddBuiltin(BuiltinOperator_STRIDED_SLICE,
tflite::ops::micro::Register_STRIDED_SLICE(), tflite::ops::micro::Register_STRIDED_SLICE(),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册