Unsupported data type.
model is converted nevertheless and runs fine, error doesn't stop conversion.
### 1. System information

- OS Platform and Distribution (e.g., Linux Ubuntu 16.04): ChomeOS linux container (fully updated) Linux penguin 6.1.60-08594-g03a802b9a072  SMP PREEMPT Fri Jan 12 18:19:39 PST 2024 aarch64 GNU/Linux (rpi5 debian stable tested also, same result)
- TensorFlow installation (pip package or built from source): pip3 Python 3.11.2 (main, Mar 13 2023, 12:18:29) [GCC 12.2.0] on linux
- TensorFlow library (version, if pip package or github SHA, if built from source): 2.15.0

### 2. Code

Provide code to help us reproduce your issues using one of the following options:

#### Option A: Reference colab notebooks

1)  Reference [TensorFlow Model Colab](https://colab.research.google.com/gist/ymodak/e96a4270b953201d5362c61c1e8b78aa/tensorflow-datasets.ipynb?authuser=1): Demonstrate how to build your TF model.
https://colab.research.google.com/github/tensorflow/tensorflow/blob/master/tensorflow/lite/g3doc/examples/on_device_training/overview.ipynb

### 3. Failure after conversion
Using Colab, notebook runs as expected, running under jupyter in ChromeOS coversion fails, see section 5.

### 4. (optional) RNN conversion support
---
### 5. (optional) Any other info / logs

> INFO:tensorflow:Assets written to: saved_model/assets
INFO:tensorflow:Assets written to: saved_model/assets
WARNING:absl:Importing a function (__inference_internal_grad_fn_55651) with ops with unsaved custom gradients. Will likely fail if a gradient is requested.
WARNING:absl:Importing a function (__inference_internal_grad_fn_55678) with ops with unsaved custom gradients. Will likely fail if a gradient is requested.
2024-01-25 15:21:06.557730: W tensorflow/compiler/mlir/lite/python/tf_tfl_flatbuffer_helpers.cc:378] Ignored output_format.
2024-01-25 15:21:06.557979: W tensorflow/compiler/mlir/lite/python/tf_tfl_flatbuffer_helpers.cc:381] Ignored drop_control_dependency.
2024-01-25 15:21:06.558694: I tensorflow/cc/saved_model/reader.cc:83] Reading SavedModel from: saved_model
2024-01-25 15:21:06.564590: I tensorflow/cc/saved_model/reader.cc:51] Reading meta graph with tags { serve }
2024-01-25 15:21:06.564844: I tensorflow/cc/saved_model/reader.cc:146] Reading SavedModel debug info (if present) from: saved_model
2024-01-25 15:21:06.580002: I tensorflow/cc/saved_model/loader.cc:233] Restoring SavedModel bundle.
2024-01-25 15:21:06.698809: I tensorflow/cc/saved_model/loader.cc:217] Running initialization op on SavedModel bundle at path: saved_model
2024-01-25 15:21:06.774244: I tensorflow/cc/saved_model/loader.cc:316] SavedModel load for tags { serve }; Status: success: OK. Took 215566 microseconds.
2024-01-25 15:21:07.459350: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.459685: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.459769: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.459855: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.459911: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.459953: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.460003: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.460042: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.460088: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.460150: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.460200: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.460247: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.460293: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.460421: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.460486: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.460595: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.460646: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.460686: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.460732: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.460769: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.460846: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.460930: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.460981: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.461021: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.461066: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.461103: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.461175: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.461250: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.461293: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.461337: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.461374: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.461416: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.461451: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.461492: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
2024-01-25 15:21:07.461577: E tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc:119] Unsupported data type.
Summary on the non-converted ops:
---------------------------------
> * Accepted dialects: tfl, builtin, func
> * Non-Converted Ops: 33, Total Ops 198, % non-converted = 16.67 %
> * 26 ARITH ops, 6 TF ops, 1 TF_SAVED_MODEL ops
>- arith.constant:   26 occurrences  (: 5, i64: 2, f32: 11, i32: 8)
>- tf.ReluGrad:    1 occurrences  (f32: 1)
>- tf.Restore:    4 occurrences  (f32: 4)
>- tf.Save:    1 occurrences
>- tf_saved_model.session_initializer:    1 occurrences
>  (i64: 1, f32: 12)
>  (f32: 6)
>  (f32: 1)
>  (i32: 2)
>  (f32: 2)
>  (i1: 2)
>  (f32: 1)
>  (f32: 1)
>  (f32: 1)
>  (f32: 10)
>  (f32: 5)
>  (i64: 1, f32: 24)
>  (i32: 1)
>  (f32: 8)
>  (f32: 2)
>  (f32: 1)
>  (i32: 4)
>  (i32: 3)
>  (f32: 2)
>  (f32: 1)
>  (f32: 4)
>  (f32: 2)
>  (: 30)
>2024-01-25 15:21:07.492189: W tensorflow/compiler/mlir/lite/flatbuffer_export.cc:2921] TFLite interpreter needs to link Flex >delegate in order to run the model since it contains the following Select TFop(s):
>Flex ops: FlexReluGrad, FlexRestore, FlexSave
>Details:
>	tf.ReluGrad(tensor<?x128xf32>, tensor<?x128xf32>) -> (tensor<?x128xf32>) : {device = ""}
>	tf.Restore(tensor<!tf_type.string>, tensor<!tf_type.string>) -> (tensor<10xf32>) : {device = "", preferred_shard = -1 : i64}
>	tf.Restore(tensor<!tf_type.string>, tensor<!tf_type.string>) -> (tensor<128x10xf32>) : {device = "", preferred_shard = -1 : i64}
>	tf.Restore(tensor<!tf_type.string>, tensor<!tf_type.string>) -> (tensor<128xf32>) : {device = "", preferred_shard = -1 : i64}
>	tf.Restore(tensor<!tf_type.string>, tensor<!tf_type.string>) -> (tensor<784x128xf32>) : {device = "", preferred_shard = -1 : i64}
>	tf.Save(tensor<!tf_type.string>, tensor<4x!tf_type.string>, tensor<784x128xf32>, tensor<128xf32>, tensor<128x10xf32>, >tensor<10xf32>) -> () : {device = ""}
>See instructions: https://www.tensorflow.org/lite/guide/ops_select