Skip to content

Commit

Permalink
fix: fix tf version for cond function
Browse files Browse the repository at this point in the history
  • Loading branch information
Ishticode committed Feb 18, 2024
1 parent 1084760 commit b967bc8
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -186,7 +186,7 @@ def multi_dot(
return dot_out


@with_unsupported_dtypes({"1.25.0 and below": ("float16", "bfloat16")}, backend_version)
@with_unsupported_dtypes({"2.15.0 and below": ("float16", "bfloat16")}, backend_version)
def cond(
x: Union[tf.Tensor, tf.Variable],
/,
Expand Down

0 comments on commit b967bc8

Please sign in to comment.