Compute LSTM and GRU via cuDNN for RaggedTensors.

This cherry-picked commit fixes a bug preventing computing
LSTM and GRU for RaggedTensors via cuDNN, resulting in
a large speedup (easily 10 times).

A TF2-only test comparing ragged and masked tensor LSTM and GRU
on CPU and GPU is also provided.

Compute LSTM and GRU via cuDNN for RaggedTensors.

This cherry-picked commit fixes a bug preventing computing
LSTM and GRU for RaggedTensors via cuDNN, resulting in
a large speedup (easily 10 times).

A TF2-only test comparing ragged and masked tensor LSTM and GRU
on CPU and GPU is also provided.

Fix RandomCrop layer called on integer input

We had a tf.cond which called tf.image.resize on one path (returning float32) and
cropping the input dtype on the other, which lead to a error. Instead we cast the
inputs to the layer's compute dtype and do our computation with that type.

Fixes keras-team#15681

PiperOrigin-RevId: 417686461

Disable the failing tests due to numpy 1.20 change (keras-team#15552)

The test has been disabled on master branch as well, we need to cherrypick this to unblock the release build.

original cl in cl/400819173.

Disable the failing tests due to numpy 1.20 change (keras-team#15552)

The test has been disabled on master branch as well, we need to cherrypick this to unblock the release build.

original cl in cl/400819173.

Disable the failing tests due to numpy 1.20 change (keras-team#15552)

The test has been disabled on master branch as well, we need to cherrypick this to unblock the release build.

original cl in cl/400819173.

Update Keras Functional model docstring for the new feature of submod…

…el slicing.

PiperOrigin-RevId: 399303217

Remove temporary monitoring now that underlying perf issue is resolved

PiperOrigin-RevId: 398533606

Use the safer `safe_load` function instead of `unsafe_load` when poss…

…ible

There is no need to open ourselves up to arbitrary code execution, especially since this is not in a performance critical loop, so we can take the slowdown due to safety.

PiperOrigin-RevId: 388501098

Use the safer `safe_load` function instead of `unsafe_load` when poss…

…ible

There is no need to open ourselves up to arbitrary code execution, especially since this is not in a performance critical loop, so we can take the slowdown due to safety.

PiperOrigin-RevId: 388501098