Skip to content

Commit 5bc82fc

Browse files
majiddadashicopybara-github
authored andcommitted
Update the version of SIGN operator in reference resolver.
Version 1 and 2 are supported by the kernel used by the reference resolver. PiperOrigin-RevId: 775468100
1 parent 04de34d commit 5bc82fc

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tflite/kernels/register_ref.cc

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -539,7 +539,9 @@ BuiltinRefOpResolver::BuiltinRefOpResolver() {
539539
AddBuiltin(BuiltinOperator_UNSORTED_SEGMENT_SUM,
540540
Register_UNSORTED_SEGMENT_SUM());
541541
AddBuiltin(BuiltinOperator_ATAN2, Register_ATAN2());
542-
AddBuiltin(BuiltinOperator_SIGN, Register_SIGN());
542+
AddBuiltin(BuiltinOperator_SIGN, Register_SIGN(),
543+
/* min_version = */ 1,
544+
/* max_version = */ 2);
543545
AddBuiltin(BuiltinOperator_CALL_ONCE,
544546
tflite::ops::builtin::Register_CALL_ONCE());
545547
AddBuiltin(BuiltinOperator_VAR_HANDLE, Register_VAR_HANDLE());

0 commit comments

Comments
 (0)