未验证 提交 b57a0b9e 编写于 作者: D deqiangc 提交者: GitHub

Allow different variant of ADD kernel (#1139)

This corresponds to cl/9257066. This allows some variant
ADD kernel optimization.

BUG=https://b/231368244Co-authored-by: Nmergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
上级 8184347d
......@@ -120,8 +120,8 @@ class MicroMutableOpResolver : public MicroOpResolver {
ParseAbs);
}
TfLiteStatus AddAdd() {
return AddBuiltin(BuiltinOperator_ADD, tflite::Register_ADD(), ParseAdd);
TfLiteStatus AddAdd(const TfLiteRegistration& registration = Register_ADD()) {
return AddBuiltin(BuiltinOperator_ADD, registration, ParseAdd);
}
TfLiteStatus AddAddN() {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册