[ WARN:0@1.791] global loadsave.cpp:848 imwrite_ Unsupported depth image for selected encoder is fallbacked to CV_8U.
running
Good looks like: tensor([[0., 0., 1.]], grad_fn=<ViewBackward0>)
Wrong looks like: tensor([[ 21.,  21., -41.]], grad_fn=<ViewBackward0>)
saving
Converting to ONNX with quantization float32
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:283: UserWarning: 
    Found GPU0 NVIDIA GeForce GTX 1050 Ti which is of cuda capability 6.1.
    Minimum and Maximum cuda capability supported by this version of PyTorch is
    (7.0) - (12.0)
    
  warnings.warn(
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:304: UserWarning: 
    Please install PyTorch with a following CUDA
    configurations:  12.6 following instructions at
    https://pytorch.org/get-started/locally/
    
  warnings.warn(matched_cuda_warn.format(matched_arches))
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:326: UserWarning: 
NVIDIA GeForce GTX 1050 Ti with CUDA capability sm_61 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_70 sm_75 sm_80 sm_86 sm_90 sm_100 sm_120.
If you want to use the NVIDIA GeForce GTX 1050 Ti GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

  warnings.warn(
[0;93m2025-12-20 21:39:07.576759303 [W:onnxruntime:Default, device_discovery.cc:164 DiscoverDevicesForPlatform] GPU device discovery failed: device_discovery.cc:89 ReadFileContents Failed to open file: "/sys/class/drm/card0/device/vendor"[m
[torch.onnx] Obtain model graph for `ChannelSum([...]` with `torch.export.export(..., strict=False)`...
[torch.onnx] Obtain model graph for `ChannelSum([...]` with `torch.export.export(..., strict=False)`... ✅
[torch.onnx] Run decomposition...
[torch.onnx] Run decomposition... ✅
[torch.onnx] Translate the graph into ONNX...
[torch.onnx] Translate the graph into ONNX... ✅
[torch.onnx] Check the ONNX model...
[torch.onnx] Check the ONNX model... ✅
[torch.onnx] Execute the model with ONNX Runtime...
[torch.onnx] Execute the model with ONNX Runtime... ✅
[torch.onnx] Verify output accuracy...
[torch.onnx] Verify output accuracy... ✅
[torch.onnx] Export report has been saved to 'build/onnx_export_2025-12-20_21-39-05-820126_success.md'.
Applied 1 of general pattern rewrite rules.
Done

Converting to ONNX with quantization float16
[0;93m2025-12-20 21:39:09.677928817 [W:onnxruntime:Default, device_discovery.cc:164 DiscoverDevicesForPlatform] GPU device discovery failed: device_discovery.cc:89 ReadFileContents Failed to open file: "/sys/class/drm/card0/device/vendor"[m
Done

Converting to ONNX with quantization int8 --calibration-dataset Random
[0;93m2025-12-20 21:39:10.372375035 [W:onnxruntime:Default, device_discovery.cc:164 DiscoverDevicesForPlatform] GPU device discovery failed: device_discovery.cc:89 ReadFileContents Failed to open file: "/sys/class/drm/card0/device/vendor"[m
Done

Converting to tflite with quantization float32
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/keras/src/export/tf2onnx_lib.py:8: FutureWarning: In the future `np.object` will be defined as the corresponding NumPy scalar.
  if not hasattr(np, "object"):
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/distributed/distributed_c10d.py:351: UserWarning: Device capability of jax unspecified, assuming `cpu` and `cuda`. Please specify it via the `devices` argument of `register_backend`.
  warnings.warn(
WARNING:2025-12-20 21:39:20,845:jax._src.xla_bridge:852: An NVIDIA GPU may be present on this machine, but a CUDA-enabled jaxlib is not installed. Falling back to cpu.
[WARNING  21:39:20] An NVIDIA GPU may be present on this machine, but a CUDA-enabled jaxlib is not installed. Falling back to cpu.
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:283: UserWarning: 
    Found GPU0 NVIDIA GeForce GTX 1050 Ti which is of cuda capability 6.1.
    Minimum and Maximum cuda capability supported by this version of PyTorch is
    (7.0) - (12.0)
    
  warnings.warn(
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:304: UserWarning: 
    Please install PyTorch with a following CUDA
    configurations:  12.6 following instructions at
    https://pytorch.org/get-started/locally/
    
  warnings.warn(matched_cuda_warn.format(matched_arches))
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:326: UserWarning: 
NVIDIA GeForce GTX 1050 Ti with CUDA capability sm_61 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_70 sm_75 sm_80 sm_86 sm_90 sm_100 sm_120.
If you want to use the NVIDIA GeForce GTX 1050 Ti GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

  warnings.warn(
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
W0000 00:00:1766266763.734035  108995 tf_tfl_flatbuffer_helpers.cc:364] Ignored output_format.
W0000 00:00:1766266763.734082  108995 tf_tfl_flatbuffer_helpers.cc:367] Ignored drop_control_dependency.
I0000 00:00:1766266763.734732  108995 reader.cc:83] Reading SavedModel from: /tmp/tmpt6g3pyme
I0000 00:00:1766266763.735023  108995 reader.cc:52] Reading meta graph with tags { serve }
I0000 00:00:1766266763.735039  108995 reader.cc:147] Reading SavedModel debug info (if present) from: /tmp/tmpt6g3pyme
I0000 00:00:1766266763.736758  108995 mlir_graph_optimization_pass.cc:437] MLIR V1 optimization pass is not enabled
I0000 00:00:1766266763.737068  108995 loader.cc:236] Restoring SavedModel bundle.
I0000 00:00:1766266763.750954  108995 loader.cc:220] Running initialization op on SavedModel bundle at path: /tmp/tmpt6g3pyme
I0000 00:00:1766266763.755389  108995 loader.cc:471] SavedModel load for tags { serve }; Status: success: OK. Took 20665 microseconds.
I0000 00:00:1766266763.763223  108995 dump_mlir_util.cc:269] disabling MLIR crash reproducer, set env var `MLIR_CRASH_REPRODUCER_DIRECTORY` to enable.
I0000 00:00:1766266763.818345  108995 flatbuffer_export.cc:4160] Estimated count of arithmetic ops: 1155  ops, equivalently 577  MACs
Done

Converting to tflite with quantization float16
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/keras/src/export/tf2onnx_lib.py:8: FutureWarning: In the future `np.object` will be defined as the corresponding NumPy scalar.
  if not hasattr(np, "object"):
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/distributed/distributed_c10d.py:351: UserWarning: Device capability of jax unspecified, assuming `cpu` and `cuda`. Please specify it via the `devices` argument of `register_backend`.
  warnings.warn(
WARNING:2025-12-20 21:39:32,667:jax._src.xla_bridge:852: An NVIDIA GPU may be present on this machine, but a CUDA-enabled jaxlib is not installed. Falling back to cpu.
[WARNING  21:39:32] An NVIDIA GPU may be present on this machine, but a CUDA-enabled jaxlib is not installed. Falling back to cpu.
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:283: UserWarning: 
    Found GPU0 NVIDIA GeForce GTX 1050 Ti which is of cuda capability 6.1.
    Minimum and Maximum cuda capability supported by this version of PyTorch is
    (7.0) - (12.0)
    
  warnings.warn(
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:304: UserWarning: 
    Please install PyTorch with a following CUDA
    configurations:  12.6 following instructions at
    https://pytorch.org/get-started/locally/
    
  warnings.warn(matched_cuda_warn.format(matched_arches))
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:326: UserWarning: 
NVIDIA GeForce GTX 1050 Ti with CUDA capability sm_61 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_70 sm_75 sm_80 sm_86 sm_90 sm_100 sm_120.
If you want to use the NVIDIA GeForce GTX 1050 Ti GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

  warnings.warn(
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
W0000 00:00:1766266775.462499  109092 tf_tfl_flatbuffer_helpers.cc:364] Ignored output_format.
W0000 00:00:1766266775.462633  109092 tf_tfl_flatbuffer_helpers.cc:367] Ignored drop_control_dependency.
I0000 00:00:1766266775.463313  109092 reader.cc:83] Reading SavedModel from: /tmp/tmp3ykp77vo
I0000 00:00:1766266775.463612  109092 reader.cc:52] Reading meta graph with tags { serve }
I0000 00:00:1766266775.463630  109092 reader.cc:147] Reading SavedModel debug info (if present) from: /tmp/tmp3ykp77vo
I0000 00:00:1766266775.465420  109092 mlir_graph_optimization_pass.cc:437] MLIR V1 optimization pass is not enabled
I0000 00:00:1766266775.465754  109092 loader.cc:236] Restoring SavedModel bundle.
I0000 00:00:1766266775.480010  109092 loader.cc:220] Running initialization op on SavedModel bundle at path: /tmp/tmp3ykp77vo
I0000 00:00:1766266775.488210  109092 loader.cc:471] SavedModel load for tags { serve }; Status: success: OK. Took 24908 microseconds.
I0000 00:00:1766266775.503995  109092 dump_mlir_util.cc:269] disabling MLIR crash reproducer, set env var `MLIR_CRASH_REPRODUCER_DIRECTORY` to enable.
loc("main.ChannelSum/torch.nn.modules.container.Sequential_model/torch.nn.modules.conv.Conv2d_0;"("conv2d"("/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/main.py":18:0))): error: failed to legalize operation 'tfl.transpose' that was explicitly marked illegal
[ERROR    21:39:35] None
╭──────────────────────────────────────────────────────────────────────── Traceback (most recent call last) ────────────────────────────────────────────────────────────────────────╮
│ /data/ai-parade/toolkit/src/ai_parade/_cli/ai_parade/main.py:437 in convert                                                                                                       │
│                                                                                                                                                                                   │
│   434 │   │   parsed_verify_dataset = None                                                                                                                                        │
│   435 │                                                                                                                                                                           │
│   436 │   try:                                                                                                                                                                    │
│ ❱ 437 │   │   results = convert_impl(                                                                                                                                             │
│   438 │   │   │   QuantizedModelFormat(metadata.format),                                                                                                                          │
│   439 │   │   │   QuantizedModelFormat(model_format, quantization),                                                                                                               │
│   440 │   │   │   metadata,                                                                                                                                                       │
│                                                                                                                                                                                   │
│ ╭─────────────────────────────────────────────────────────────────────────────────── locals ────────────────────────────────────────────────────────────────────────────────────╮ │
│ │                  artifacts = PosixPath('build')                                                                                                                               │ │
│ │        calibration_dataset = None                                                                                                                                             │ │
│ │   calibration_dataset_path = None                                                                                                                                             │ │
│ │            hw_acceleration = <HardwareAcceleration.CPU: 'CPU'>                                                                                                                │ │
│ │                   metadata = ModelMetadata(                                                                                                                                   │ │
│ │                              │   name='ChannelSum',                                                                                                                           │ │
│ │                              │   family=None,                                                                                                                                 │ │
│ │                              │   size=1.0,                                                                                                                                    │ │
│ │                              │   format=<ModelFormat.PyTorch: 'pytorch'>,                                                                                                     │ │
│ │                              │   task=<ModelTasks.Classification: 'classification'>,                                                                                          │ │
│ │                              │   task_properties={},                                                                                                                          │ │
│ │                              │   citation=None,                                                                                                                               │ │
│ │                              │   citation_link=None,                                                                                                                          │ │
│ │                              │   download_link=None,                                                                                                                          │ │
│ │                              │   license=None,                                                                                                                                │ │
│ │                              │   install=InstallOptions(pip_dependencies=None, conda_dependencies=None, build_system_dependencies=['setuptools'], python='3.12.3'),           │ │
│ │                              │   image_input=ImageInput(                                                                                                                      │ │
│ │                              │   │   batchSize=1,                                                                                                                             │ │
│ │                              │   │   height=8,                                                                                                                                │ │
│ │                              │   │   width=8,                                                                                                                                 │ │
│ │                              │   │   channelOrder=<ChannelOrder.RGB: 'RGB'>,                                                                                                  │ │
│ │                              │   │   dataOrder=<DataOrder.NCHW: 'NCHW'>,                                                                                                      │ │
│ │                              │   │   dataType=<DataType.float32: 'float32'>,                                                                                                  │ │
│ │                              │   │   means=None,                                                                                                                              │ │
│ │                              │   │   stds=None                                                                                                                                │ │
│ │                              │   ),                                                                                                                                           │ │
│ │                              │   output_mapping=None,                                                                                                                         │ │
│ │                              │   runner_name=None,                                                                                                                            │ │
│ │                              │   dataset=None,                                                                                                                                │ │
│ │                              │   backbone=None,                                                                                                                               │ │
│ │                              │   architecture=None,                                                                                                                           │ │
│ │                              │   tags=(),                                                                                                                                     │ │
│ │                              │   pytorch=PyTorchOptions(                                                                                                                      │ │
│ │                              │   │   model_import='main',                                                                                                                     │ │
│ │                              │   │   model_init_expression=None,                                                                                                              │ │
│ │                              │   │   model_init_fx='ChannelSum',                                                                                                              │ │
│ │                              │   │   yacs_configs=None,                                                                                                                       │ │
│ │                              │   │   load_mapping=None                                                                                                                        │ │
│ │                              │   ),                                                                                                                                           │ │
│ │                              │   get_installer=<function choose_installer.<locals>.uv_installer at 0x7076abf70ae0>,                                                           │ │
│ │                              │   get_runner=<function _api_to_final.<locals>.<lambda> at 0x7076abf70b80>,                                                                     │ │
│ │                              │   repository_path=PosixPath('.')                                                                                                               │ │
│ │                              )                                                                                                                                                │ │
│ │              metadata_list = [                                                                                                                                                │ │
│ │                              │   ModelMetadata(                                                                                                                               │ │
│ │                              │   │   name='ChannelSum',                                                                                                                       │ │
│ │                              │   │   family=None,                                                                                                                             │ │
│ │                              │   │   size=1.0,                                                                                                                                │ │
│ │                              │   │   format=<ModelFormat.PyTorch: 'pytorch'>,                                                                                                 │ │
│ │                              │   │   task=<ModelTasks.Classification: 'classification'>,                                                                                      │ │
│ │                              │   │   task_properties={},                                                                                                                      │ │
│ │                              │   │   citation=None,                                                                                                                           │ │
│ │                              │   │   citation_link=None,                                                                                                                      │ │
│ │                              │   │   download_link=None,                                                                                                                      │ │
│ │                              │   │   license=None,                                                                                                                            │ │
│ │                              │   │   install=InstallOptions(pip_dependencies=None, conda_dependencies=None, build_system_dependencies=['setuptools'], python='3.12.3'),       │ │
│ │                              │   │   image_input=ImageInput(                                                                                                                  │ │
│ │                              │   │   │   batchSize=1,                                                                                                                         │ │
│ │                              │   │   │   height=8,                                                                                                                            │ │
│ │                              │   │   │   width=8,                                                                                                                             │ │
│ │                              │   │   │   channelOrder=<ChannelOrder.RGB: 'RGB'>,                                                                                              │ │
│ │                              │   │   │   dataOrder=<DataOrder.NCHW: 'NCHW'>,                                                                                                  │ │
│ │                              │   │   │   dataType=<DataType.float32: 'float32'>,                                                                                              │ │
│ │                              │   │   │   means=None,                                                                                                                          │ │
│ │                              │   │   │   stds=None                                                                                                                            │ │
│ │                              │   │   ),                                                                                                                                       │ │
│ │                              │   │   output_mapping=None,                                                                                                                     │ │
│ │                              │   │   runner_name=None,                                                                                                                        │ │
│ │                              │   │   dataset=None,                                                                                                                            │ │
│ │                              │   │   backbone=None,                                                                                                                           │ │
│ │                              │   │   architecture=None,                                                                                                                       │ │
│ │                              │   │   tags=(),                                                                                                                                 │ │
│ │                              │   │   pytorch=PyTorchOptions(                                                                                                                  │ │
│ │                              │   │   │   model_import='main',                                                                                                                 │ │
│ │                              │   │   │   model_init_expression=None,                                                                                                          │ │
│ │                              │   │   │   model_init_fx='ChannelSum',                                                                                                          │ │
│ │                              │   │   │   yacs_configs=None,                                                                                                                   │ │
│ │                              │   │   │   load_mapping=None                                                                                                                    │ │
│ │                              │   │   ),                                                                                                                                       │ │
│ │                              │   │   get_installer=<function choose_installer.<locals>.uv_installer at 0x7076abf70ae0>,                                                       │ │
│ │                              │   │   get_runner=<function _api_to_final.<locals>.<lambda> at 0x7076abf70b80>,                                                                 │ │
│ │                              │   │   repository_path=PosixPath('.')                                                                                                           │ │
│ │                              │   )                                                                                                                                            │ │
│ │                              ]                                                                                                                                                │ │
│ │               model_format = <ModelFormat.LiteRT: 'tflite'>                                                                                                                   │ │
│ │                 model_name = 'ChannelSum'                                                                                                                                     │ │
│ │                     output = PosixPath('models')                                                                                                                              │ │
│ │                   override = False                                                                                                                                            │ │
│ │ parsed_calibration_dataset = None                                                                                                                                             │ │
│ │      parsed_verify_dataset = None                                                                                                                                             │ │
│ │               quantization = <Quantization.float16: 'float16'>                                                                                                                │ │
│ │            repository_path = PosixPath('.')                                                                                                                                   │ │
│ │                 use_remote = False                                                                                                                                            │ │
│ │                    verbose = []                                                                                                                                               │ │
│ │            verbosity_level = 0                                                                                                                                                │ │
│ │                     verify = False                                                                                                                                            │ │
│ │             verify_dataset = None                                                                                                                                             │ │
│ │        verify_dataset_path = None                                                                                                                                             │ │
│ │               weights_path = PosixPath('weights.pth')                                                                                                                         │ │
│ ╰───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯ │
│                                                                                                                                                                                   │
│ /data/ai-parade/toolkit/src/ai_parade/_cli/ai_parade/main.py:288 in convert_impl                                                                                                  │
│                                                                                                                                                                                   │
│   285 │   │   │   break                                                                                                                                                           │
│   286 │   │   except Exception as e:                                                                                                                                              │
│   287 │   │   │   logger.error(e.__cause__)                                                                                                                                       │
│ ❱ 288 │   │   │   raise e                                                                                                                                                         │
│   289 │                                                                                                                                                                           │
│   290 │   if verify_dataset is not None:                                                                                                                                          │
│   291 │   │   with get_runner(                                                                                                                                                    │
│                                                                                                                                                                                   │
│ ╭─────────────────────────────────────────────────────────────────────────────────── locals ───────────────────────────────────────────────────────────────────────────────────╮  │
│ │                   artifacts = PosixPath('build')                                                                                                                             │  │
│ │                base_results = None                                                                                                                                           │  │
│ │         calibration_dataset = None                                                                                                                                           │  │
│ │                   converter = <ai_parade._toolkit.converters.ai_edge_torch.AIEdgeConverterEntry object at 0x7076abf671a0>                                                    │  │
│ │             converter_chain = [<ai_parade._toolkit.converters.ai_edge_torch.AIEdgeConverterEntry object at 0x7076abf671a0>]                                                  │  │
│ │            converter_chains = [[<ai_parade._toolkit.converters.ai_edge_torch.AIEdgeConverterEntry object at 0x7076abf671a0>]]                                                │  │
│ │     current_hw_acceleration = <HardwareAcceleration.CPU: 'CPU'>                                                                                                              │  │
│ │       current_source_format = QuantizedModelFormat(format=<ModelFormat.PyTorch: 'pytorch'>, quantization=<Quantization.none: 'float32'>)                                     │  │
│ │ current_source_weights_path = PosixPath('weights.pth')                                                                                                                       │  │
│ │       current_target_format = QuantizedModelFormat(format=<ModelFormat.LiteRT: 'tflite'>, quantization=<Quantization.float16: 'float16'>)                                    │  │
│ │ current_target_weights_path = PosixPath('models/tflite.float16.all.tflite')                                                                                                  │  │
│ │             hw_acceleration = <HardwareAcceleration.CPU: 'CPU'>                                                                                                              │  │
│ │                    metadata = ModelMetadata(                                                                                                                                 │  │
│ │                               │   name='ChannelSum',                                                                                                                         │  │
│ │                               │   family=None,                                                                                                                               │  │
│ │                               │   size=1.0,                                                                                                                                  │  │
│ │                               │   format=<ModelFormat.PyTorch: 'pytorch'>,                                                                                                   │  │
│ │                               │   task=<ModelTasks.Classification: 'classification'>,                                                                                        │  │
│ │                               │   task_properties={},                                                                                                                        │  │
│ │                               │   citation=None,                                                                                                                             │  │
│ │                               │   citation_link=None,                                                                                                                        │  │
│ │                               │   download_link=None,                                                                                                                        │  │
│ │                               │   license=None,                                                                                                                              │  │
│ │                               │   install=InstallOptions(pip_dependencies=None, conda_dependencies=None, build_system_dependencies=['setuptools'], python='3.12.3'),         │  │
│ │                               │   image_input=ImageInput(                                                                                                                    │  │
│ │                               │   │   batchSize=1,                                                                                                                           │  │
│ │                               │   │   height=8,                                                                                                                              │  │
│ │                               │   │   width=8,                                                                                                                               │  │
│ │                               │   │   channelOrder=<ChannelOrder.RGB: 'RGB'>,                                                                                                │  │
│ │                               │   │   dataOrder=<DataOrder.NCHW: 'NCHW'>,                                                                                                    │  │
│ │                               │   │   dataType=<DataType.float32: 'float32'>,                                                                                                │  │
│ │                               │   │   means=None,                                                                                                                            │  │
│ │                               │   │   stds=None                                                                                                                              │  │
│ │                               │   ),                                                                                                                                         │  │
│ │                               │   output_mapping=None,                                                                                                                       │  │
│ │                               │   runner_name=None,                                                                                                                          │  │
│ │                               │   dataset=None,                                                                                                                              │  │
│ │                               │   backbone=None,                                                                                                                             │  │
│ │                               │   architecture=None,                                                                                                                         │  │
│ │                               │   tags=(),                                                                                                                                   │  │
│ │                               │   pytorch=PyTorchOptions(                                                                                                                    │  │
│ │                               │   │   model_import='main',                                                                                                                   │  │
│ │                               │   │   model_init_expression=None,                                                                                                            │  │
│ │                               │   │   model_init_fx='ChannelSum',                                                                                                            │  │
│ │                               │   │   yacs_configs=None,                                                                                                                     │  │
│ │                               │   │   load_mapping=None                                                                                                                      │  │
│ │                               │   ),                                                                                                                                         │  │
│ │                               │   get_installer=<function choose_installer.<locals>.uv_installer at 0x7076abf70ae0>,                                                         │  │
│ │                               │   get_runner=<function _api_to_final.<locals>.<lambda> at 0x7076abf70b80>,                                                                   │  │
│ │                               │   repository_path=PosixPath('.')                                                                                                             │  │
│ │                               )                                                                                                                                              │  │
│ │                      output = PosixPath('models')                                                                                                                            │  │
│ │                    override = False                                                                                                                                          │  │
│ │                      runner = <ai_parade._toolkit.run.PyTorch.PyTorchRunner object at 0x7076ac1ea780>                                                                        │  │
│ │               source_format = QuantizedModelFormat(format=<ModelFormat.PyTorch: 'pytorch'>, quantization=<Quantization.none: 'float32'>)                                     │  │
│ │               target_format = QuantizedModelFormat(format=<ModelFormat.LiteRT: 'tflite'>, quantization=<Quantization.float16: 'float16'>)                                    │  │
│ │                  use_remote = False                                                                                                                                          │  │
│ │              verify_dataset = None                                                                                                                                           │  │
│ │                weights_path = PosixPath('weights.pth')                                                                                                                       │  │
│ ╰──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯  │
│                                                                                                                                                                                   │
│ /data/ai-parade/toolkit/src/ai_parade/_cli/ai_parade/main.py:274 in convert_impl                                                                                                  │
│                                                                                                                                                                                   │
│   271 │   │   │   │   │   │   use_remote=use_remote,                                                                                                                              │
│   272 │   │   │   │   │   ) as runner:                                                                                                                                            │
│   273 │   │   │   │   │   │   runner.set_output_directory(artifacts)                                                                                                              │
│ ❱ 274 │   │   │   │   │   │   converted_weights = runner.run(                                                                                                                     │
│   275 │   │   │   │   │   │   │   converter.create(                                                                                                                               │
│   276 │   │   │   │   │   │   │   │   ConverterOptions(                                                                                                                           │
│   277 │   │   │   │   │   │   │   │   │   calibration_data=calibration_dataset,                                                                                                   │
│                                                                                                                                                                                   │
│ ╭─────────────────────────────────────────────────────────────────────────────────── locals ───────────────────────────────────────────────────────────────────────────────────╮  │
│ │                   artifacts = PosixPath('build')                                                                                                                             │  │
│ │                base_results = None                                                                                                                                           │  │
│ │         calibration_dataset = None                                                                                                                                           │  │
│ │                   converter = <ai_parade._toolkit.converters.ai_edge_torch.AIEdgeConverterEntry object at 0x7076abf671a0>                                                    │  │
│ │             converter_chain = [<ai_parade._toolkit.converters.ai_edge_torch.AIEdgeConverterEntry object at 0x7076abf671a0>]                                                  │  │
│ │            converter_chains = [[<ai_parade._toolkit.converters.ai_edge_torch.AIEdgeConverterEntry object at 0x7076abf671a0>]]                                                │  │
│ │     current_hw_acceleration = <HardwareAcceleration.CPU: 'CPU'>                                                                                                              │  │
│ │       current_source_format = QuantizedModelFormat(format=<ModelFormat.PyTorch: 'pytorch'>, quantization=<Quantization.none: 'float32'>)                                     │  │
│ │ current_source_weights_path = PosixPath('weights.pth')                                                                                                                       │  │
│ │       current_target_format = QuantizedModelFormat(format=<ModelFormat.LiteRT: 'tflite'>, quantization=<Quantization.float16: 'float16'>)                                    │  │
│ │ current_target_weights_path = PosixPath('models/tflite.float16.all.tflite')                                                                                                  │  │
│ │             hw_acceleration = <HardwareAcceleration.CPU: 'CPU'>                                                                                                              │  │
│ │                    metadata = ModelMetadata(                                                                                                                                 │  │
│ │                               │   name='ChannelSum',                                                                                                                         │  │
│ │                               │   family=None,                                                                                                                               │  │
│ │                               │   size=1.0,                                                                                                                                  │  │
│ │                               │   format=<ModelFormat.PyTorch: 'pytorch'>,                                                                                                   │  │
│ │                               │   task=<ModelTasks.Classification: 'classification'>,                                                                                        │  │
│ │                               │   task_properties={},                                                                                                                        │  │
│ │                               │   citation=None,                                                                                                                             │  │
│ │                               │   citation_link=None,                                                                                                                        │  │
│ │                               │   download_link=None,                                                                                                                        │  │
│ │                               │   license=None,                                                                                                                              │  │
│ │                               │   install=InstallOptions(pip_dependencies=None, conda_dependencies=None, build_system_dependencies=['setuptools'], python='3.12.3'),         │  │
│ │                               │   image_input=ImageInput(                                                                                                                    │  │
│ │                               │   │   batchSize=1,                                                                                                                           │  │
│ │                               │   │   height=8,                                                                                                                              │  │
│ │                               │   │   width=8,                                                                                                                               │  │
│ │                               │   │   channelOrder=<ChannelOrder.RGB: 'RGB'>,                                                                                                │  │
│ │                               │   │   dataOrder=<DataOrder.NCHW: 'NCHW'>,                                                                                                    │  │
│ │                               │   │   dataType=<DataType.float32: 'float32'>,                                                                                                │  │
│ │                               │   │   means=None,                                                                                                                            │  │
│ │                               │   │   stds=None                                                                                                                              │  │
│ │                               │   ),                                                                                                                                         │  │
│ │                               │   output_mapping=None,                                                                                                                       │  │
│ │                               │   runner_name=None,                                                                                                                          │  │
│ │                               │   dataset=None,                                                                                                                              │  │
│ │                               │   backbone=None,                                                                                                                             │  │
│ │                               │   architecture=None,                                                                                                                         │  │
│ │                               │   tags=(),                                                                                                                                   │  │
│ │                               │   pytorch=PyTorchOptions(                                                                                                                    │  │
│ │                               │   │   model_import='main',                                                                                                                   │  │
│ │                               │   │   model_init_expression=None,                                                                                                            │  │
│ │                               │   │   model_init_fx='ChannelSum',                                                                                                            │  │
│ │                               │   │   yacs_configs=None,                                                                                                                     │  │
│ │                               │   │   load_mapping=None                                                                                                                      │  │
│ │                               │   ),                                                                                                                                         │  │
│ │                               │   get_installer=<function choose_installer.<locals>.uv_installer at 0x7076abf70ae0>,                                                         │  │
│ │                               │   get_runner=<function _api_to_final.<locals>.<lambda> at 0x7076abf70b80>,                                                                   │  │
│ │                               │   repository_path=PosixPath('.')                                                                                                             │  │
│ │                               )                                                                                                                                              │  │
│ │                      output = PosixPath('models')                                                                                                                            │  │
│ │                    override = False                                                                                                                                          │  │
│ │                      runner = <ai_parade._toolkit.run.PyTorch.PyTorchRunner object at 0x7076ac1ea780>                                                                        │  │
│ │               source_format = QuantizedModelFormat(format=<ModelFormat.PyTorch: 'pytorch'>, quantization=<Quantization.none: 'float32'>)                                     │  │
│ │               target_format = QuantizedModelFormat(format=<ModelFormat.LiteRT: 'tflite'>, quantization=<Quantization.float16: 'float16'>)                                    │  │
│ │                  use_remote = False                                                                                                                                          │  │
│ │              verify_dataset = None                                                                                                                                           │  │
│ │                weights_path = PosixPath('weights.pth')                                                                                                                       │  │
│ ╰──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯  │
│                                                                                                                                                                                   │
│ /data/ai-parade/toolkit/src/ai_parade/_toolkit/run/ModelRunnerBase.py:49 in run                                                                                                   │
│                                                                                                                                                                                   │
│   46 │   @override                                                                                                                                                                │
│   47 │   def run(self, what: Callable[[TChild], TReturn]) -> TReturn:                                                                                                             │
│   48 │   │   logger.debug("Running %s", what)                                                                                                                                     │
│ ❱ 49 │   │   return what(self)  # type: ignore                                                                                                                                    │
│   50                                                                                                                                                                              │
│                                                                                                                                                                                   │
│ ╭────────────────────────────────────────────────────── locals ──────────────────────────────────────────────────────╮                                                            │
│ │ self = <ai_parade._toolkit.run.PyTorch.PyTorchRunner object at 0x7076ac1ea780>                                     │                                                            │
│ │ what = <ai_parade._toolkit.converters.ai_edge_torch.AIEdgeConverterEntry.AIEdgeConverter object at 0x7075a042c800> │                                                            │
│ ╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯                                                            │
│                                                                                                                                                                                   │
│ /data/ai-parade/toolkit/src/ai_parade/_toolkit/converters/_pytorch_base.py:67 in __call__                                                                                         │
│                                                                                                                                                                                   │
│   64 │   │   │   │   runner.model.to("cuda")                                                                                                                                      │
│   65 │   │   │   │   return run(sample_inputs.to("cuda"))                                                                                                                         │
│   66 │   │   │   else:                                                                                                                                                            │
│ ❱ 67 │   │   │   │   raise e                                                                                                                                                      │
│   68 │                                                                                                                                                                            │
│   69 │   @property                                                                                                                                                                │
│   70 │   def output_weights(self) -> Path:                                                                                                                                        │
│                                                                                                                                                                                   │
│ ╭──────────────────────────────────────────────────────────────────────── locals ────────────────────────────────────────────────────────────────────────╮                        │
│ │         dtype = torch.float32                                                                                                                          │                        │
│ │           msg = '/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/main.py:18:0: error: f'+79                                                  │                        │
│ │        runner = <ai_parade._toolkit.run.PyTorch.PyTorchRunner object at 0x7076ac1ea780>                                                                │                        │
│ │ sample_inputs = tensor([[[[0.2243, 0.5706, 0.4411, 0.5186, 0.8343, 0.3273, 0.3929, 0.3115],                                                            │                        │
│ │                 │   │     [0.5358, 0.1130, 0.5575, 0.1142, 0.5208, 0.3452, 0.8331, 0.7768],                                                            │                        │
│ │                 │   │     [0.9230, 0.2600, 0.0228, 0.0102, 0.2000, 0.6298, 0.4463, 0.1573],                                                            │                        │
│ │                 │   │     [0.9576, 0.0464, 0.9526, 0.7909, 0.8643, 0.1527, 0.7745, 0.1739],                                                            │                        │
│ │                 │   │     [0.4960, 0.0108, 0.3155, 0.8872, 0.0312, 0.6835, 0.8785, 0.0626],                                                            │                        │
│ │                 │   │     [0.4290, 0.7619, 0.8068, 0.9489, 0.6325, 0.9095, 0.1314, 0.5626],                                                            │                        │
│ │                 │   │     [0.4634, 0.0064, 0.7873, 0.0856, 0.5606, 0.9311, 0.4043, 0.9641],                                                            │                        │
│ │                 │   │     [0.0401, 0.8528, 0.5467, 0.4164, 0.2891, 0.4738, 0.7247, 0.4493]],                                                           │                        │
│ │                 │   │                                                                                                                                  │                        │
│ │                 │   │    [[0.1454, 0.3370, 0.6570, 0.1092, 0.1587, 0.7846, 0.0925, 0.6129],                                                            │                        │
│ │                 │   │     [0.8820, 0.6546, 0.1865, 0.0245, 0.7940, 0.6163, 0.1662, 0.2517],                                                            │                        │
│ │                 │   │     [0.0661, 0.4475, 0.2710, 0.7076, 0.1103, 0.0815, 0.7107, 0.2665],                                                            │                        │
│ │                 │   │     [0.2242, 0.3436, 0.5217, 0.6179, 0.6369, 0.3429, 0.4038, 0.6882],                                                            │                        │
│ │                 │   │     [0.7371, 0.8753, 0.8248, 0.7301, 0.0628, 0.5966, 0.4842, 0.1891],                                                            │                        │
│ │                 │   │     [0.4859, 0.4627, 0.0817, 0.1769, 0.2592, 0.9534, 0.0531, 0.0929],                                                            │                        │
│ │                 │   │     [0.5974, 0.1196, 0.8604, 0.2376, 0.5087, 0.6193, 0.2005, 0.4860],                                                            │                        │
│ │                 │   │     [0.7827, 0.4663, 0.7164, 0.9615, 0.3566, 0.2026, 0.5709, 0.0399]],                                                           │                        │
│ │                 │   │                                                                                                                                  │                        │
│ │                 │   │    [[0.7285, 0.8982, 0.5800, 0.0434, 0.6374, 0.1529, 0.2090, 0.8310],                                                            │                        │
│ │                 │   │     [0.4847, 0.1665, 0.3721, 0.1957, 0.4235, 0.8747, 0.1119, 0.8707],                                                            │                        │
│ │                 │   │     [0.1417, 0.3594, 0.9178, 0.6520, 0.4281, 0.6576, 0.7592, 0.4688],                                                            │                        │
│ │                 │   │     [0.6966, 0.4346, 0.1372, 0.6827, 0.9358, 0.1997, 0.2630, 0.6783],                                                            │                        │
│ │                 │   │     [0.9864, 0.0658, 0.0928, 0.1555, 0.3301, 0.4784, 0.7591, 0.7454],                                                            │                        │
│ │                 │   │     [0.3877, 0.6185, 0.5025, 0.9012, 0.9694, 0.3627, 0.1598, 0.0952],                                                            │                        │
│ │                 │   │     [0.6490, 0.9418, 0.8828, 0.7886, 0.1780, 0.3415, 0.2222, 0.4112],                                                            │                        │
│ │                 │   │     [0.9276, 0.2684, 0.9631, 0.5696, 0.7507, 0.4182, 0.6742, 0.9049]]]])                                                         │                        │
│ │          self = <ai_parade._toolkit.converters.ai_edge_torch.AIEdgeConverterEntry.AIEdgeConverter object at 0x7075a042c800>                            │                        │
│ │         torch = <module 'torch' from '/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/__init__.py'> │                        │
│ ╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯                        │
│                                                                                                                                                                                   │
│ /data/ai-parade/toolkit/src/ai_parade/_toolkit/converters/_pytorch_base.py:57 in __call__                                                                                         │
│                                                                                                                                                                                   │
│   54 │   │                                                                                                                                                                        │
│   55 │   │   try:                                                                                                                                                                 │
│   56 │   │   │   runner.model.to("cpu")                                                                                                                                           │
│ ❱ 57 │   │   │   return run(sample_inputs.to("cpu"))                                                                                                                              │
│   58 │   │   except Exception as e:                                                                                                                                               │
│   59 │   │   │   msg = str(e)                                                                                                                                                     │
│   60 │   │   │   if "cpu" in msg or "gpu" in msg or "device" in msg:                                                                                                              │
│                                                                                                                                                                                   │
│ ╭──────────────────────────────────────────────────────────────────────── locals ────────────────────────────────────────────────────────────────────────╮                        │
│ │         dtype = torch.float32                                                                                                                          │                        │
│ │           msg = '/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/main.py:18:0: error: f'+79                                                  │                        │
│ │        runner = <ai_parade._toolkit.run.PyTorch.PyTorchRunner object at 0x7076ac1ea780>                                                                │                        │
│ │ sample_inputs = tensor([[[[0.2243, 0.5706, 0.4411, 0.5186, 0.8343, 0.3273, 0.3929, 0.3115],                                                            │                        │
│ │                 │   │     [0.5358, 0.1130, 0.5575, 0.1142, 0.5208, 0.3452, 0.8331, 0.7768],                                                            │                        │
│ │                 │   │     [0.9230, 0.2600, 0.0228, 0.0102, 0.2000, 0.6298, 0.4463, 0.1573],                                                            │                        │
│ │                 │   │     [0.9576, 0.0464, 0.9526, 0.7909, 0.8643, 0.1527, 0.7745, 0.1739],                                                            │                        │
│ │                 │   │     [0.4960, 0.0108, 0.3155, 0.8872, 0.0312, 0.6835, 0.8785, 0.0626],                                                            │                        │
│ │                 │   │     [0.4290, 0.7619, 0.8068, 0.9489, 0.6325, 0.9095, 0.1314, 0.5626],                                                            │                        │
│ │                 │   │     [0.4634, 0.0064, 0.7873, 0.0856, 0.5606, 0.9311, 0.4043, 0.9641],                                                            │                        │
│ │                 │   │     [0.0401, 0.8528, 0.5467, 0.4164, 0.2891, 0.4738, 0.7247, 0.4493]],                                                           │                        │
│ │                 │   │                                                                                                                                  │                        │
│ │                 │   │    [[0.1454, 0.3370, 0.6570, 0.1092, 0.1587, 0.7846, 0.0925, 0.6129],                                                            │                        │
│ │                 │   │     [0.8820, 0.6546, 0.1865, 0.0245, 0.7940, 0.6163, 0.1662, 0.2517],                                                            │                        │
│ │                 │   │     [0.0661, 0.4475, 0.2710, 0.7076, 0.1103, 0.0815, 0.7107, 0.2665],                                                            │                        │
│ │                 │   │     [0.2242, 0.3436, 0.5217, 0.6179, 0.6369, 0.3429, 0.4038, 0.6882],                                                            │                        │
│ │                 │   │     [0.7371, 0.8753, 0.8248, 0.7301, 0.0628, 0.5966, 0.4842, 0.1891],                                                            │                        │
│ │                 │   │     [0.4859, 0.4627, 0.0817, 0.1769, 0.2592, 0.9534, 0.0531, 0.0929],                                                            │                        │
│ │                 │   │     [0.5974, 0.1196, 0.8604, 0.2376, 0.5087, 0.6193, 0.2005, 0.4860],                                                            │                        │
│ │                 │   │     [0.7827, 0.4663, 0.7164, 0.9615, 0.3566, 0.2026, 0.5709, 0.0399]],                                                           │                        │
│ │                 │   │                                                                                                                                  │                        │
│ │                 │   │    [[0.7285, 0.8982, 0.5800, 0.0434, 0.6374, 0.1529, 0.2090, 0.8310],                                                            │                        │
│ │                 │   │     [0.4847, 0.1665, 0.3721, 0.1957, 0.4235, 0.8747, 0.1119, 0.8707],                                                            │                        │
│ │                 │   │     [0.1417, 0.3594, 0.9178, 0.6520, 0.4281, 0.6576, 0.7592, 0.4688],                                                            │                        │
│ │                 │   │     [0.6966, 0.4346, 0.1372, 0.6827, 0.9358, 0.1997, 0.2630, 0.6783],                                                            │                        │
│ │                 │   │     [0.9864, 0.0658, 0.0928, 0.1555, 0.3301, 0.4784, 0.7591, 0.7454],                                                            │                        │
│ │                 │   │     [0.3877, 0.6185, 0.5025, 0.9012, 0.9694, 0.3627, 0.1598, 0.0952],                                                            │                        │
│ │                 │   │     [0.6490, 0.9418, 0.8828, 0.7886, 0.1780, 0.3415, 0.2222, 0.4112],                                                            │                        │
│ │                 │   │     [0.9276, 0.2684, 0.9631, 0.5696, 0.7507, 0.4182, 0.6742, 0.9049]]]])                                                         │                        │
│ │          self = <ai_parade._toolkit.converters.ai_edge_torch.AIEdgeConverterEntry.AIEdgeConverter object at 0x7075a042c800>                            │                        │
│ │         torch = <module 'torch' from '/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/__init__.py'> │                        │
│ ╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯                        │
│                                                                                                                                                                                   │
│ /data/ai-parade/toolkit/src/ai_parade/_toolkit/converters/_pytorch_base.py:52 in run                                                                                              │
│                                                                                                                                                                                   │
│   49 │   │   │   # run the model to trigger potential lazy initialization                                                                                                         │
│   50 │   │   │   runner.model(sample_inputs)                                                                                                                                      │
│   51 │   │   │                                                                                                                                                                    │
│ ❱ 52 │   │   │   self._export_call(runner.model, (sample_inputs,), runner.output_directory)                                                                                       │
│   53 │   │   │   return self.output_weights                                                                                                                                       │
│   54 │   │                                                                                                                                                                        │
│   55 │   │   try:                                                                                                                                                                 │
│                                                                                                                                                                                   │
│ ╭────────────────────────────────────────────────────────── locals ───────────────────────────────────────────────────────────╮                                                   │
│ │        runner = <ai_parade._toolkit.run.PyTorch.PyTorchRunner object at 0x7076ac1ea780>                                     │                                                   │
│ │ sample_inputs = tensor([[[[0.2243, 0.5706, 0.4411, 0.5186, 0.8343, 0.3273, 0.3929, 0.3115],                                 │                                                   │
│ │                 │   │     [0.5358, 0.1130, 0.5575, 0.1142, 0.5208, 0.3452, 0.8331, 0.7768],                                 │                                                   │
│ │                 │   │     [0.9230, 0.2600, 0.0228, 0.0102, 0.2000, 0.6298, 0.4463, 0.1573],                                 │                                                   │
│ │                 │   │     [0.9576, 0.0464, 0.9526, 0.7909, 0.8643, 0.1527, 0.7745, 0.1739],                                 │                                                   │
│ │                 │   │     [0.4960, 0.0108, 0.3155, 0.8872, 0.0312, 0.6835, 0.8785, 0.0626],                                 │                                                   │
│ │                 │   │     [0.4290, 0.7619, 0.8068, 0.9489, 0.6325, 0.9095, 0.1314, 0.5626],                                 │                                                   │
│ │                 │   │     [0.4634, 0.0064, 0.7873, 0.0856, 0.5606, 0.9311, 0.4043, 0.9641],                                 │                                                   │
│ │                 │   │     [0.0401, 0.8528, 0.5467, 0.4164, 0.2891, 0.4738, 0.7247, 0.4493]],                                │                                                   │
│ │                 │   │                                                                                                       │                                                   │
│ │                 │   │    [[0.1454, 0.3370, 0.6570, 0.1092, 0.1587, 0.7846, 0.0925, 0.6129],                                 │                                                   │
│ │                 │   │     [0.8820, 0.6546, 0.1865, 0.0245, 0.7940, 0.6163, 0.1662, 0.2517],                                 │                                                   │
│ │                 │   │     [0.0661, 0.4475, 0.2710, 0.7076, 0.1103, 0.0815, 0.7107, 0.2665],                                 │                                                   │
│ │                 │   │     [0.2242, 0.3436, 0.5217, 0.6179, 0.6369, 0.3429, 0.4038, 0.6882],                                 │                                                   │
│ │                 │   │     [0.7371, 0.8753, 0.8248, 0.7301, 0.0628, 0.5966, 0.4842, 0.1891],                                 │                                                   │
│ │                 │   │     [0.4859, 0.4627, 0.0817, 0.1769, 0.2592, 0.9534, 0.0531, 0.0929],                                 │                                                   │
│ │                 │   │     [0.5974, 0.1196, 0.8604, 0.2376, 0.5087, 0.6193, 0.2005, 0.4860],                                 │                                                   │
│ │                 │   │     [0.7827, 0.4663, 0.7164, 0.9615, 0.3566, 0.2026, 0.5709, 0.0399]],                                │                                                   │
│ │                 │   │                                                                                                       │                                                   │
│ │                 │   │    [[0.7285, 0.8982, 0.5800, 0.0434, 0.6374, 0.1529, 0.2090, 0.8310],                                 │                                                   │
│ │                 │   │     [0.4847, 0.1665, 0.3721, 0.1957, 0.4235, 0.8747, 0.1119, 0.8707],                                 │                                                   │
│ │                 │   │     [0.1417, 0.3594, 0.9178, 0.6520, 0.4281, 0.6576, 0.7592, 0.4688],                                 │                                                   │
│ │                 │   │     [0.6966, 0.4346, 0.1372, 0.6827, 0.9358, 0.1997, 0.2630, 0.6783],                                 │                                                   │
│ │                 │   │     [0.9864, 0.0658, 0.0928, 0.1555, 0.3301, 0.4784, 0.7591, 0.7454],                                 │                                                   │
│ │                 │   │     [0.3877, 0.6185, 0.5025, 0.9012, 0.9694, 0.3627, 0.1598, 0.0952],                                 │                                                   │
│ │                 │   │     [0.6490, 0.9418, 0.8828, 0.7886, 0.1780, 0.3415, 0.2222, 0.4112],                                 │                                                   │
│ │                 │   │     [0.9276, 0.2684, 0.9631, 0.5696, 0.7507, 0.4182, 0.6742, 0.9049]]]])                              │                                                   │
│ │          self = <ai_parade._toolkit.converters.ai_edge_torch.AIEdgeConverterEntry.AIEdgeConverter object at 0x7075a042c800> │                                                   │
│ ╰─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯                                                   │
│                                                                                                                                                                                   │
│ /data/ai-parade/toolkit/src/ai_parade/_toolkit/converters/ai_edge_torch.py:75 in _export_call                                                                                     │
│                                                                                                                                                                                   │
│   72 │   │   │   │   │   │   model, sample_inputs, _ai_edge_converter_flags=converter_flags                                                                                       │
│   73 │   │   │   │   │   )                                                                                                                                                        │
│   74 │   │   │   │   case Quantization.float16:                                                                                                                                   │
│ ❱ 75 │   │   │   │   │   edge_model = ai_edge_torch.convert(                                                                                                                      │
│   76 │   │   │   │   │   │   model.half(), tuple([x.half() for x in sample_inputs])                                                                                               │
│   77 │   │   │   │   │   )                                                                                                                                                        │
│   78 │   │   │   │   case _:                                                                                                                                                      │
│                                                                                                                                                                                   │
│ ╭───────────────────────────────────────────────────────────────────────────────── locals ──────────────────────────────────────────────────────────────────────────────────╮     │
│ │    ai_edge_torch = <module 'ai_edge_torch' from '/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/ai_edge_torch/__init__.py'> │     │
│ │  converter_flags = {}                                                                                                                                                     │     │
│ │           kvargs = {}                                                                                                                                                     │     │
│ │            model = ChannelSum(                                                                                                                                            │     │
│ │                      (model): Sequential(                                                                                                                                 │     │
│ │                    │   (0): Conv2d(3, 3, kernel_size=(8, 8), stride=(1, 1))                                                                                               │     │
│ │                    │   (1): Flatten(start_dim=1, end_dim=-1)                                                                                                              │     │
│ │                      )                                                                                                                                                    │     │
│ │                    )                                                                                                                                                      │     │
│ │ output_directory = PosixPath('build')                                                                                                                                     │     │
│ │    sample_inputs = (                                                                                                                                                      │     │
│ │                    │   tensor([[[[0.2243, 0.5706, 0.4411, 0.5186, 0.8343, 0.3273, 0.3929, 0.3115],                                                                        │     │
│ │                    │   │     [0.5358, 0.1130, 0.5575, 0.1142, 0.5208, 0.3452, 0.8331, 0.7768],                                                                            │     │
│ │                    │   │     [0.9230, 0.2600, 0.0228, 0.0102, 0.2000, 0.6298, 0.4463, 0.1573],                                                                            │     │
│ │                    │   │     [0.9576, 0.0464, 0.9526, 0.7909, 0.8643, 0.1527, 0.7745, 0.1739],                                                                            │     │
│ │                    │   │     [0.4960, 0.0108, 0.3155, 0.8872, 0.0312, 0.6835, 0.8785, 0.0626],                                                                            │     │
│ │                    │   │     [0.4290, 0.7619, 0.8068, 0.9489, 0.6325, 0.9095, 0.1314, 0.5626],                                                                            │     │
│ │                    │   │     [0.4634, 0.0064, 0.7873, 0.0856, 0.5606, 0.9311, 0.4043, 0.9641],                                                                            │     │
│ │                    │   │     [0.0401, 0.8528, 0.5467, 0.4164, 0.2891, 0.4738, 0.7247, 0.4493]],                                                                           │     │
│ │                    │   │                                                                                                                                                  │     │
│ │                    │   │    [[0.1454, 0.3370, 0.6570, 0.1092, 0.1587, 0.7846, 0.0925, 0.6129],                                                                            │     │
│ │                    │   │     [0.8820, 0.6546, 0.1865, 0.0245, 0.7940, 0.6163, 0.1662, 0.2517],                                                                            │     │
│ │                    │   │     [0.0661, 0.4475, 0.2710, 0.7076, 0.1103, 0.0815, 0.7107, 0.2665],                                                                            │     │
│ │                    │   │     [0.2242, 0.3436, 0.5217, 0.6179, 0.6369, 0.3429, 0.4038, 0.6882],                                                                            │     │
│ │                    │   │     [0.7371, 0.8753, 0.8248, 0.7301, 0.0628, 0.5966, 0.4842, 0.1891],                                                                            │     │
│ │                    │   │     [0.4859, 0.4627, 0.0817, 0.1769, 0.2592, 0.9534, 0.0531, 0.0929],                                                                            │     │
│ │                    │   │     [0.5974, 0.1196, 0.8604, 0.2376, 0.5087, 0.6193, 0.2005, 0.4860],                                                                            │     │
│ │                    │   │     [0.7827, 0.4663, 0.7164, 0.9615, 0.3566, 0.2026, 0.5709, 0.0399]],                                                                           │     │
│ │                    │   │                                                                                                                                                  │     │
│ │                    │   │    [[0.7285, 0.8982, 0.5800, 0.0434, 0.6374, 0.1529, 0.2090, 0.8310],                                                                            │     │
│ │                    │   │     [0.4847, 0.1665, 0.3721, 0.1957, 0.4235, 0.8747, 0.1119, 0.8707],                                                                            │     │
│ │                    │   │     [0.1417, 0.3594, 0.9178, 0.6520, 0.4281, 0.6576, 0.7592, 0.4688],                                                                            │     │
│ │                    │   │     [0.6966, 0.4346, 0.1372, 0.6827, 0.9358, 0.1997, 0.2630, 0.6783],                                                                            │     │
│ │                    │   │     [0.9864, 0.0658, 0.0928, 0.1555, 0.3301, 0.4784, 0.7591, 0.7454],                                                                            │     │
│ │                    │   │     [0.3877, 0.6185, 0.5025, 0.9012, 0.9694, 0.3627, 0.1598, 0.0952],                                                                            │     │
│ │                    │   │     [0.6490, 0.9418, 0.8828, 0.7886, 0.1780, 0.3415, 0.2222, 0.4112],                                                                            │     │
│ │                    │   │     [0.9276, 0.2684, 0.9631, 0.5696, 0.7507, 0.4182, 0.6742, 0.9049]]]]),                                                                        │     │
│ │                    )                                                                                                                                                      │     │
│ │             self = <ai_parade._toolkit.converters.ai_edge_torch.AIEdgeConverterEntry.AIEdgeConverter object at 0x7075a042c800>                                            │     │
│ │               tf = <module 'tensorflow' from '/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/tensorflow/__init__.py'>       │     │
│ ╰───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯     │
│                                                                                                                                                                                   │
│ /data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/ai_edge_torch/_convert/converter.py:315 in convert                                   │
│                                                                                                                                                                                   │
│   312   if _ai_edge_converter_flags is None:                                                                                                                                      │
│   313 │   _ai_edge_converter_flags = {}                                                                                                                                           │
│   314                                                                                                                                                                             │
│ ❱ 315   return Converter().convert(                                                                                                                                               │
│   316 │     module,                                                                                                                                                               │
│   317 │     sample_args,                                                                                                                                                          │
│   318 │     sample_kwargs,                                                                                                                                                        │
│                                                                                                                                                                                   │
│ ╭────────────────────────────────────────────────── locals ──────────────────────────────────────────────────╮                                                                    │
│ │ _ai_edge_converter_flags = {}                                                                              │                                                                    │
│ │         _saved_model_dir = None                                                                            │                                                                    │
│ │           dynamic_shapes = None                                                                            │                                                                    │
│ │                   module = ChannelSum(                                                                     │                                                                    │
│ │                              (model): Sequential(                                                          │                                                                    │
│ │                            │   (0): Conv2d(3, 3, kernel_size=(8, 8), stride=(1, 1))                        │                                                                    │
│ │                            │   (1): Flatten(start_dim=1, end_dim=-1)                                       │                                                                    │
│ │                              )                                                                             │                                                                    │
│ │                            )                                                                               │                                                                    │
│ │             quant_config = None                                                                            │                                                                    │
│ │              sample_args = (                                                                               │                                                                    │
│ │                            │   tensor([[[[0.2242, 0.5708, 0.4412, 0.5186, 0.8345, 0.3274, 0.3928, 0.3115], │                                                                    │
│ │                            │   │     [0.5356, 0.1130, 0.5576, 0.1142, 0.5210, 0.3452, 0.8330, 0.7769],     │                                                                    │
│ │                            │   │     [0.9229, 0.2600, 0.0228, 0.0102, 0.2001, 0.6299, 0.4463, 0.1573],     │                                                                    │
│ │                            │   │     [0.9575, 0.0464, 0.9526, 0.7910, 0.8643, 0.1527, 0.7744, 0.1738],     │                                                                    │
│ │                            │   │     [0.4961, 0.0108, 0.3154, 0.8872, 0.0312, 0.6836, 0.8784, 0.0626],     │                                                                    │
│ │                            │   │     [0.4290, 0.7617, 0.8066, 0.9487, 0.6323, 0.9097, 0.1315, 0.5625],     │                                                                    │
│ │                            │   │     [0.4634, 0.0064, 0.7871, 0.0856, 0.5605, 0.9312, 0.4043, 0.9639],     │                                                                    │
│ │                            │   │     [0.0401, 0.8530, 0.5469, 0.4165, 0.2891, 0.4739, 0.7246, 0.4492]],    │                                                                    │
│ │                            │   │                                                                           │                                                                    │
│ │                            │   │    [[0.1454, 0.3369, 0.6572, 0.1093, 0.1587, 0.7847, 0.0925, 0.6128],     │                                                                    │
│ │                            │   │     [0.8818, 0.6548, 0.1865, 0.0245, 0.7939, 0.6162, 0.1663, 0.2517],     │                                                                    │
│ │                            │   │     [0.0661, 0.4475, 0.2710, 0.7075, 0.1103, 0.0815, 0.7104, 0.2666],     │                                                                    │
│ │                            │   │     [0.2242, 0.3438, 0.5215, 0.6177, 0.6367, 0.3428, 0.4038, 0.6880],     │                                                                    │
│ │                            │   │     [0.7373, 0.8755, 0.8247, 0.7300, 0.0629, 0.5967, 0.4841, 0.1891],     │                                                                    │
│ │                            │   │     [0.4858, 0.4626, 0.0817, 0.1769, 0.2593, 0.9536, 0.0531, 0.0929],     │                                                                    │
│ │                            │   │     [0.5977, 0.1196, 0.8604, 0.2375, 0.5088, 0.6191, 0.2006, 0.4861],     │                                                                    │
│ │                            │   │     [0.7827, 0.4663, 0.7163, 0.9614, 0.3567, 0.2026, 0.5708, 0.0399]],    │                                                                    │
│ │                            │   │                                                                           │                                                                    │
│ │                            │   │    [[0.7285, 0.8984, 0.5801, 0.0434, 0.6372, 0.1530, 0.2090, 0.8311],     │                                                                    │
│ │                            │   │     [0.4846, 0.1665, 0.3721, 0.1958, 0.4236, 0.8745, 0.1119, 0.8706],     │                                                                    │
│ │                            │   │     [0.1417, 0.3594, 0.9180, 0.6519, 0.4282, 0.6577, 0.7593, 0.4688],     │                                                                    │
│ │                            │   │     [0.6968, 0.4346, 0.1372, 0.6826, 0.9360, 0.1997, 0.2629, 0.6782],     │                                                                    │
│ │                            │   │     [0.9863, 0.0658, 0.0928, 0.1555, 0.3301, 0.4783, 0.7593, 0.7456],     │                                                                    │
│ │                            │   │     [0.3877, 0.6187, 0.5024, 0.9014, 0.9692, 0.3625, 0.1598, 0.0953],     │                                                                    │
│ │                            │   │     [0.6489, 0.9419, 0.8828, 0.7886, 0.1780, 0.3416, 0.2223, 0.4111],     │                                                                    │
│ │                            │   │     [0.9277, 0.2686, 0.9629, 0.5698, 0.7510, 0.4182, 0.6743, 0.9048]]]],  │                                                                    │
│ │                            │      dtype=torch.float16),                                                    │                                                                    │
│ │                            )                                                                               │                                                                    │
│ │            sample_kwargs = None                                                                            │                                                                    │
│ │            strict_export = True                                                                            │                                                                    │
│ ╰────────────────────────────────────────────────────────────────────────────────────────────────────────────╯                                                                    │
│                                                                                                                                                                                   │
│ /data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/ai_edge_torch/_convert/converter.py:203 in convert                                   │
│                                                                                                                                                                                   │
│   200 │   │   │   "sample_args or sample_kwargs must be provided if a module is"                                                                                                  │
│   201 │   │   │   " specified."                                                                                                                                                   │
│   202 │   │   )                                                                                                                                                                   │
│ ❱ 203 │   converted_model = conversion.convert_signatures(                                                                                                                        │
│   204 │   │   self._signatures,                                                                                                                                                   │
│   205 │   │   strict_export=strict_export,                                                                                                                                        │
│   206 │   │   quant_config=quant_config,                                                                                                                                          │
│                                                                                                                                                                                   │
│ ╭────────────────────────────────────────────────── locals ──────────────────────────────────────────────────╮                                                                    │
│ │ _ai_edge_converter_flags = {}                                                                              │                                                                    │
│ │         _saved_model_dir = None                                                                            │                                                                    │
│ │           dynamic_shapes = None                                                                            │                                                                    │
│ │                   module = ChannelSum(                                                                     │                                                                    │
│ │                              (model): Sequential(                                                          │                                                                    │
│ │                            │   (0): Conv2d(3, 3, kernel_size=(8, 8), stride=(1, 1))                        │                                                                    │
│ │                            │   (1): Flatten(start_dim=1, end_dim=-1)                                       │                                                                    │
│ │                              )                                                                             │                                                                    │
│ │                            )                                                                               │                                                                    │
│ │             quant_config = None                                                                            │                                                                    │
│ │              sample_args = (                                                                               │                                                                    │
│ │                            │   tensor([[[[0.2242, 0.5708, 0.4412, 0.5186, 0.8345, 0.3274, 0.3928, 0.3115], │                                                                    │
│ │                            │   │     [0.5356, 0.1130, 0.5576, 0.1142, 0.5210, 0.3452, 0.8330, 0.7769],     │                                                                    │
│ │                            │   │     [0.9229, 0.2600, 0.0228, 0.0102, 0.2001, 0.6299, 0.4463, 0.1573],     │                                                                    │
│ │                            │   │     [0.9575, 0.0464, 0.9526, 0.7910, 0.8643, 0.1527, 0.7744, 0.1738],     │                                                                    │
│ │                            │   │     [0.4961, 0.0108, 0.3154, 0.8872, 0.0312, 0.6836, 0.8784, 0.0626],     │                                                                    │
│ │                            │   │     [0.4290, 0.7617, 0.8066, 0.9487, 0.6323, 0.9097, 0.1315, 0.5625],     │                                                                    │
│ │                            │   │     [0.4634, 0.0064, 0.7871, 0.0856, 0.5605, 0.9312, 0.4043, 0.9639],     │                                                                    │
│ │                            │   │     [0.0401, 0.8530, 0.5469, 0.4165, 0.2891, 0.4739, 0.7246, 0.4492]],    │                                                                    │
│ │                            │   │                                                                           │                                                                    │
│ │                            │   │    [[0.1454, 0.3369, 0.6572, 0.1093, 0.1587, 0.7847, 0.0925, 0.6128],     │                                                                    │
│ │                            │   │     [0.8818, 0.6548, 0.1865, 0.0245, 0.7939, 0.6162, 0.1663, 0.2517],     │                                                                    │
│ │                            │   │     [0.0661, 0.4475, 0.2710, 0.7075, 0.1103, 0.0815, 0.7104, 0.2666],     │                                                                    │
│ │                            │   │     [0.2242, 0.3438, 0.5215, 0.6177, 0.6367, 0.3428, 0.4038, 0.6880],     │                                                                    │
│ │                            │   │     [0.7373, 0.8755, 0.8247, 0.7300, 0.0629, 0.5967, 0.4841, 0.1891],     │                                                                    │
│ │                            │   │     [0.4858, 0.4626, 0.0817, 0.1769, 0.2593, 0.9536, 0.0531, 0.0929],     │                                                                    │
│ │                            │   │     [0.5977, 0.1196, 0.8604, 0.2375, 0.5088, 0.6191, 0.2006, 0.4861],     │                                                                    │
│ │                            │   │     [0.7827, 0.4663, 0.7163, 0.9614, 0.3567, 0.2026, 0.5708, 0.0399]],    │                                                                    │
│ │                            │   │                                                                           │                                                                    │
│ │                            │   │    [[0.7285, 0.8984, 0.5801, 0.0434, 0.6372, 0.1530, 0.2090, 0.8311],     │                                                                    │
│ │                            │   │     [0.4846, 0.1665, 0.3721, 0.1958, 0.4236, 0.8745, 0.1119, 0.8706],     │                                                                    │
│ │                            │   │     [0.1417, 0.3594, 0.9180, 0.6519, 0.4282, 0.6577, 0.7593, 0.4688],     │                                                                    │
│ │                            │   │     [0.6968, 0.4346, 0.1372, 0.6826, 0.9360, 0.1997, 0.2629, 0.6782],     │                                                                    │
│ │                            │   │     [0.9863, 0.0658, 0.0928, 0.1555, 0.3301, 0.4783, 0.7593, 0.7456],     │                                                                    │
│ │                            │   │     [0.3877, 0.6187, 0.5024, 0.9014, 0.9692, 0.3625, 0.1598, 0.0953],     │                                                                    │
│ │                            │   │     [0.6489, 0.9419, 0.8828, 0.7886, 0.1780, 0.3416, 0.2223, 0.4111],     │                                                                    │
│ │                            │   │     [0.9277, 0.2686, 0.9629, 0.5698, 0.7510, 0.4182, 0.6743, 0.9048]]]],  │                                                                    │
│ │                            │      dtype=torch.float16),                                                    │                                                                    │
│ │                            )                                                                               │                                                                    │
│ │            sample_kwargs = None                                                                            │                                                                    │
│ │                     self = <ai_edge_torch._convert.converter.Converter object at 0x70753452bc80>           │                                                                    │
│ │            strict_export = True                                                                            │                                                                    │
│ ╰────────────────────────────────────────────────────────────────────────────────────────────────────────────╯                                                                    │
│                                                                                                                                                                                   │
│ /data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/ai_edge_torch/_convert/conversion.py:151 in convert_signatures                       │
│                                                                                                                                                                                   │
│   148                                                                                                                                                                             │
│   149   # Apply default fx passes                                                                                                                                                 │
│   150   exported_programs = list(map(_run_convert_passes, exported_programs))                                                                                                     │
│ ❱ 151   tflite_model = lowertools.exported_programs_to_tflite(                                                                                                                    │
│   152 │     exported_programs,                                                                                                                                                    │
│   153 │     signatures,                                                                                                                                                           │
│   154 │     quant_config=quant_config,                                                                                                                                            │
│                                                                                                                                                                                   │
│ ╭──────────────────────────────────────────────────── locals ────────────────────────────────────────────────────╮                                                                │
│ │     _saved_model_dir = None                                                                                    │                                                                │
│ │ _tfl_converter_flags = {}                                                                                      │                                                                │
│ │    exported_programs = [<torch.export.exported_program.ExportedProgram object at 0x70758f016ed0>]              │                                                                │
│ │         quant_config = None                                                                                    │                                                                │
│ │           signatures = [                                                                                       │                                                                │
│ │                        │   Signature(                                                                          │                                                                │
│ │                        │   │   name='serving_default',                                                         │                                                                │
│ │                        │   │   module=ChannelSum(                                                              │                                                                │
│ │                          (model): Sequential(                                                                  │                                                                │
│ │                        │   (0): Conv2d(3, 3, kernel_size=(8, 8), stride=(1, 1))                                │                                                                │
│ │                        │   (1): Flatten(start_dim=1, end_dim=-1)                                               │                                                                │
│ │                          )                                                                                     │                                                                │
│ │                        ),                                                                                      │                                                                │
│ │                        │   │   sample_args=(                                                                   │                                                                │
│ │                        │   │   │   tensor([[[[0.2242, 0.5708, 0.4412, 0.5186, 0.8345, 0.3274, 0.3928, 0.3115], │                                                                │
│ │                        │   │     [0.5356, 0.1130, 0.5576, 0.1142, 0.5210, 0.3452, 0.8330, 0.7769],             │                                                                │
│ │                        │   │     [0.9229, 0.2600, 0.0228, 0.0102, 0.2001, 0.6299, 0.4463, 0.1573],             │                                                                │
│ │                        │   │     [0.9575, 0.0464, 0.9526, 0.7910, 0.8643, 0.1527, 0.7744, 0.1738],             │                                                                │
│ │                        │   │     [0.4961, 0.0108, 0.3154, 0.8872, 0.0312, 0.6836, 0.8784, 0.0626],             │                                                                │
│ │                        │   │     [0.4290, 0.7617, 0.8066, 0.9487, 0.6323, 0.9097, 0.1315, 0.5625],             │                                                                │
│ │                        │   │     [0.4634, 0.0064, 0.7871, 0.0856, 0.5605, 0.9312, 0.4043, 0.9639],             │                                                                │
│ │                        │   │     [0.0401, 0.8530, 0.5469, 0.4165, 0.2891, 0.4739, 0.7246, 0.4492]],            │                                                                │
│ │                        │   │                                                                                   │                                                                │
│ │                        │   │    [[0.1454, 0.3369, 0.6572, 0.1093, 0.1587, 0.7847, 0.0925, 0.6128],             │                                                                │
│ │                        │   │     [0.8818, 0.6548, 0.1865, 0.0245, 0.7939, 0.6162, 0.1663, 0.2517],             │                                                                │
│ │                        │   │     [0.0661, 0.4475, 0.2710, 0.7075, 0.1103, 0.0815, 0.7104, 0.2666],             │                                                                │
│ │                        │   │     [0.2242, 0.3438, 0.5215, 0.6177, 0.6367, 0.3428, 0.4038, 0.6880],             │                                                                │
│ │                        │   │     [0.7373, 0.8755, 0.8247, 0.7300, 0.0629, 0.5967, 0.4841, 0.1891],             │                                                                │
│ │                        │   │     [0.4858, 0.4626, 0.0817, 0.1769, 0.2593, 0.9536, 0.0531, 0.0929],             │                                                                │
│ │                        │   │     [0.5977, 0.1196, 0.8604, 0.2375, 0.5088, 0.6191, 0.2006, 0.4861],             │                                                                │
│ │                        │   │     [0.7827, 0.4663, 0.7163, 0.9614, 0.3567, 0.2026, 0.5708, 0.0399]],            │                                                                │
│ │                        │   │                                                                                   │                                                                │
│ │                        │   │    [[0.7285, 0.8984, 0.5801, 0.0434, 0.6372, 0.1530, 0.2090, 0.8311],             │                                                                │
│ │                        │   │     [0.4846, 0.1665, 0.3721, 0.1958, 0.4236, 0.8745, 0.1119, 0.8706],             │                                                                │
│ │                        │   │     [0.1417, 0.3594, 0.9180, 0.6519, 0.4282, 0.6577, 0.7593, 0.4688],             │                                                                │
│ │                        │   │     [0.6968, 0.4346, 0.1372, 0.6826, 0.9360, 0.1997, 0.2629, 0.6782],             │                                                                │
│ │                        │   │     [0.9863, 0.0658, 0.0928, 0.1555, 0.3301, 0.4783, 0.7593, 0.7456],             │                                                                │
│ │                        │   │     [0.3877, 0.6187, 0.5024, 0.9014, 0.9692, 0.3625, 0.1598, 0.0953],             │                                                                │
│ │                        │   │     [0.6489, 0.9419, 0.8828, 0.7886, 0.1780, 0.3416, 0.2223, 0.4111],             │                                                                │
│ │                        │   │     [0.9277, 0.2686, 0.9629, 0.5698, 0.7510, 0.4182, 0.6743, 0.9048]]]],          │                                                                │
│ │                        │      dtype=torch.float16),                                                            │                                                                │
│ │                        │   │   ),                                                                              │                                                                │
│ │                        │   │   sample_kwargs=None,                                                             │                                                                │
│ │                        │   │   dynamic_shapes=None                                                             │                                                                │
│ │                        │   )                                                                                   │                                                                │
│ │                        ]                                                                                       │                                                                │
│ │        strict_export = True                                                                                    │                                                                │
│ ╰────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯                                                                │
│                                                                                                                                                                                   │
│ /data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/ai_edge_torch/lowertools/_shim.py:80 in exported_programs_to_tflite                  │
│                                                                                                                                                                                   │
│   77 │     bundles, signatures, exported_programs                                                                                                                                 │
│   78   )                                                                                                                                                                          │
│   79                                                                                                                                                                              │
│ ❱ 80   return utils.merged_bundle_to_tfl_model(                                                                                                                                   │
│   81 │     merged_bundle,                                                                                                                                                         │
│   82 │     signatures,                                                                                                                                                            │
│   83 │     quant_config=quant_config,                                                                                                                                             │
│                                                                                                                                                                                   │
│ ╭─────────────────────────────────────────────────────────────────────────────────── locals ────────────────────────────────────────────────────────────────────────────────────╮ │
│ │     _saved_model_dir = None                                                                                                                                                   │ │
│ │ _tfl_converter_flags = {}                                                                                                                                                     │ │
│ │              bundles = [                                                                                                                                                      │ │
│ │                        │   module {                                                                                                                                           │ │
│ │                          func.func public @main(%arg0: tensor<3x3x8x8xf16>, %arg1: tensor<3xf16>, %arg2: tensor<1x3x8x8xf16>) -> tensor<1x3xf16> {                            │ │
│ │                        │   %0 = call @permute_48a1a91b(%arg2) : (tensor<1x3x8x8xf16>) -> tensor<1x8x8x3xf16>                                                                  │ │
│ │                        │   %1 = call @_aten_clone_9c7bb3b1(%0) : (tensor<1x8x8x3xf16>) -> tensor<1x8x8x3xf16>                                                                 │ │
│ │                        │   %2 = call @permute_1965d0a5(%1) : (tensor<1x8x8x3xf16>) -> tensor<1x3x8x8xf16>                                                                     │ │
│ │                        │   %3 = call @_aten_clone_576ef666(%2) : (tensor<1x3x8x8xf16>) -> tensor<1x3x8x8xf16>                                                                 │ │
│ │                        │   %4 = stablehlo.convolution(%3, %arg0) dim_numbers = [b, f, 0, 1]x[o, i, 0, 1]->[b, f, 0, 1], window = {stride = [1, 1], pad = [[0, 0], [0, 0]],    │ │
│ │                        rhs_dilate = [1, 1]} {batch_group_count = 1 : i64, feature_group_count = 1 : i64} : (tensor<1x3x8x8xf16>, tensor<3x3x8x8xf16>) -> tensor<1x3x1x1xf16>  │ │
│ │                        │   %5 = call @permute_f718d144(%4) : (tensor<1x3x1x1xf16>) -> tensor<1x1x1x3xf16>                                                                     │ │
│ │                        │   %6 = call @jax_lowering_ee7b84fe(%5, %arg1) : (tensor<1x1x1x3xf16>, tensor<3xf16>) -> tensor<1x1x1x3xf16>                                          │ │
│ │                        │   %7 = call @permute_ceca6e02(%6) : (tensor<1x1x1x3xf16>) -> tensor<1x3x1x1xf16>                                                                     │ │
│ │                        │   %8 = call @permute_71657670(%7) : (tensor<1x3x1x1xf16>) -> tensor<1x1x1x3xf16>                                                                     │ │
│ │                        │   %9 = call @permute_e7832658(%8) : (tensor<1x1x1x3xf16>) -> tensor<1x3x1x1xf16>                                                                     │ │
│ │                        │   %10 = call @_aten_unsafe_view_bee6c7a7(%9) : (tensor<1x3x1x1xf16>) -> tensor<1x3xf16>                                                              │ │
│ │                        │   return %10 : tensor<1x3xf16>                                                                                                                       │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @_aten_unsafe_view_bee6c7a7(%arg0: tensor<1x3x1x1xf16>) -> (tensor<1x3xf16> {jax.result_info = "result"}) {                        │ │
│ │                        │   %0 = stablehlo.reshape %arg0 : (tensor<1x3x1x1xf16>) -> tensor<1x3xf16>                                                                            │ │
│ │                        │   return %0 : tensor<1x3xf16>                                                                                                                        │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @permute_e7832658(%arg0: tensor<1x1x1x3xf16>) -> (tensor<1x3x1x1xf16> {jax.result_info = "result"}) {                              │ │
│ │                        │   %0 = stablehlo.transpose %arg0, dims = [0, 3, 1, 2] : (tensor<1x1x1x3xf16>) -> tensor<1x3x1x1xf16>                                                 │ │
│ │                        │   return %0 : tensor<1x3x1x1xf16>                                                                                                                    │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @permute_71657670(%arg0: tensor<1x3x1x1xf16>) -> (tensor<1x1x1x3xf16> {jax.result_info = "result"}) {                              │ │
│ │                        │   %0 = stablehlo.transpose %arg0, dims = [0, 2, 3, 1] : (tensor<1x3x1x1xf16>) -> tensor<1x1x1x3xf16>                                                 │ │
│ │                        │   return %0 : tensor<1x1x1x3xf16>                                                                                                                    │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @permute_ceca6e02(%arg0: tensor<1x1x1x3xf16>) -> (tensor<1x3x1x1xf16> {jax.result_info = "result"}) {                              │ │
│ │                        │   %0 = stablehlo.transpose %arg0, dims = [0, 3, 1, 2] : (tensor<1x1x1x3xf16>) -> tensor<1x3x1x1xf16>                                                 │ │
│ │                        │   return %0 : tensor<1x3x1x1xf16>                                                                                                                    │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @jax_lowering_ee7b84fe(%arg0: tensor<1x1x1x3xf16>, %arg1: tensor<3xf16>) -> (tensor<1x1x1x3xf16> {jax.result_info = "result"}) {   │ │
│ │                        │   %0 = stablehlo.broadcast_in_dim %arg1, dims = [3] : (tensor<3xf16>) -> tensor<1x1x1x3xf16>                                                         │ │
│ │                        │   %1 = stablehlo.add %arg0, %0 : tensor<1x1x1x3xf16>                                                                                                 │ │
│ │                        │   return %1 : tensor<1x1x1x3xf16>                                                                                                                    │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @permute_f718d144(%arg0: tensor<1x3x1x1xf16>) -> (tensor<1x1x1x3xf16> {jax.result_info = "result"}) {                              │ │
│ │                        │   %0 = stablehlo.transpose %arg0, dims = [0, 2, 3, 1] : (tensor<1x3x1x1xf16>) -> tensor<1x1x1x3xf16>                                                 │ │
│ │                        │   return %0 : tensor<1x1x1x3xf16>                                                                                                                    │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @_aten_clone_576ef666(%arg0: tensor<1x3x8x8xf16>) -> (tensor<1x3x8x8xf16> {jax.result_info = "result"}) {                          │ │
│ │                        │   return %arg0 : tensor<1x3x8x8xf16>                                                                                                                 │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @permute_1965d0a5(%arg0: tensor<1x8x8x3xf16>) -> (tensor<1x3x8x8xf16> {jax.result_info = "result"}) {                              │ │
│ │                        │   %0 = stablehlo.transpose %arg0, dims = [0, 3, 1, 2] : (tensor<1x8x8x3xf16>) -> tensor<1x3x8x8xf16>                                                 │ │
│ │                        │   return %0 : tensor<1x3x8x8xf16>                                                                                                                    │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @_aten_clone_9c7bb3b1(%arg0: tensor<1x8x8x3xf16>) -> (tensor<1x8x8x3xf16> {jax.result_info = "result"}) {                          │ │
│ │                        │   return %arg0 : tensor<1x8x8x3xf16>                                                                                                                 │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @permute_48a1a91b(%arg0: tensor<1x3x8x8xf16>) -> (tensor<1x8x8x3xf16> {jax.result_info = "result"}) {                              │ │
│ │                        │   %0 = stablehlo.transpose %arg0, dims = [0, 2, 3, 1] : (tensor<1x3x8x8xf16>) -> tensor<1x8x8x3xf16>                                                 │ │
│ │                        │   return %0 : tensor<1x8x8x3xf16>                                                                                                                    │ │
│ │                          }                                                                                                                                                    │ │
│ │                        }                                                                                                                                                      │ │
│ │                                                                                                                                                                               │ │
│ │                        ]                                                                                                                                                      │ │
│ │    exported_programs = [<torch.export.exported_program.ExportedProgram object at 0x70758f016ed0>]                                                                             │ │
│ │        merged_bundle = MergedBundle(                                                                                                                                          │ │
│ │                        │   bundles=[                                                                                                                                          │ │
│ │                        │   │   module {                                                                                                                                       │ │
│ │                          func.func public @main(%arg0: tensor<3x3x8x8xf16>, %arg1: tensor<3xf16>, %arg2: tensor<1x3x8x8xf16>) -> tensor<1x3xf16> {                            │ │
│ │                        │   %0 = call @permute_48a1a91b(%arg2) : (tensor<1x3x8x8xf16>) -> tensor<1x8x8x3xf16>                                                                  │ │
│ │                        │   %1 = call @_aten_clone_9c7bb3b1(%0) : (tensor<1x8x8x3xf16>) -> tensor<1x8x8x3xf16>                                                                 │ │
│ │                        │   %2 = call @permute_1965d0a5(%1) : (tensor<1x8x8x3xf16>) -> tensor<1x3x8x8xf16>                                                                     │ │
│ │                        │   %3 = call @_aten_clone_576ef666(%2) : (tensor<1x3x8x8xf16>) -> tensor<1x3x8x8xf16>                                                                 │ │
│ │                        │   %4 = stablehlo.convolution(%3, %arg0) dim_numbers = [b, f, 0, 1]x[o, i, 0, 1]->[b, f, 0, 1], window = {stride = [1, 1], pad = [[0, 0], [0, 0]],    │ │
│ │                        rhs_dilate = [1, 1]} {batch_group_count = 1 : i64, feature_group_count = 1 : i64} : (tensor<1x3x8x8xf16>, tensor<3x3x8x8xf16>) -> tensor<1x3x1x1xf16>  │ │
│ │                        │   %5 = call @permute_f718d144(%4) : (tensor<1x3x1x1xf16>) -> tensor<1x1x1x3xf16>                                                                     │ │
│ │                        │   %6 = call @jax_lowering_ee7b84fe(%5, %arg1) : (tensor<1x1x1x3xf16>, tensor<3xf16>) -> tensor<1x1x1x3xf16>                                          │ │
│ │                        │   %7 = call @permute_ceca6e02(%6) : (tensor<1x1x1x3xf16>) -> tensor<1x3x1x1xf16>                                                                     │ │
│ │                        │   %8 = call @permute_71657670(%7) : (tensor<1x3x1x1xf16>) -> tensor<1x1x1x3xf16>                                                                     │ │
│ │                        │   %9 = call @permute_e7832658(%8) : (tensor<1x1x1x3xf16>) -> tensor<1x3x1x1xf16>                                                                     │ │
│ │                        │   %10 = call @_aten_unsafe_view_bee6c7a7(%9) : (tensor<1x3x1x1xf16>) -> tensor<1x3xf16>                                                              │ │
│ │                        │   return %10 : tensor<1x3xf16>                                                                                                                       │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @_aten_unsafe_view_bee6c7a7(%arg0: tensor<1x3x1x1xf16>) -> (tensor<1x3xf16> {jax.result_info = "result"}) {                        │ │
│ │                        │   %0 = stablehlo.reshape %arg0 : (tensor<1x3x1x1xf16>) -> tensor<1x3xf16>                                                                            │ │
│ │                        │   return %0 : tensor<1x3xf16>                                                                                                                        │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @permute_e7832658(%arg0: tensor<1x1x1x3xf16>) -> (tensor<1x3x1x1xf16> {jax.result_info = "result"}) {                              │ │
│ │                        │   %0 = stablehlo.transpose %arg0, dims = [0, 3, 1, 2] : (tensor<1x1x1x3xf16>) -> tensor<1x3x1x1xf16>                                                 │ │
│ │                        │   return %0 : tensor<1x3x1x1xf16>                                                                                                                    │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @permute_71657670(%arg0: tensor<1x3x1x1xf16>) -> (tensor<1x1x1x3xf16> {jax.result_info = "result"}) {                              │ │
│ │                        │   %0 = stablehlo.transpose %arg0, dims = [0, 2, 3, 1] : (tensor<1x3x1x1xf16>) -> tensor<1x1x1x3xf16>                                                 │ │
│ │                        │   return %0 : tensor<1x1x1x3xf16>                                                                                                                    │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @permute_ceca6e02(%arg0: tensor<1x1x1x3xf16>) -> (tensor<1x3x1x1xf16> {jax.result_info = "result"}) {                              │ │
│ │                        │   %0 = stablehlo.transpose %arg0, dims = [0, 3, 1, 2] : (tensor<1x1x1x3xf16>) -> tensor<1x3x1x1xf16>                                                 │ │
│ │                        │   return %0 : tensor<1x3x1x1xf16>                                                                                                                    │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @jax_lowering_ee7b84fe(%arg0: tensor<1x1x1x3xf16>, %arg1: tensor<3xf16>) -> (tensor<1x1x1x3xf16> {jax.result_info = "result"}) {   │ │
│ │                        │   %0 = stablehlo.broadcast_in_dim %arg1, dims = [3] : (tensor<3xf16>) -> tensor<1x1x1x3xf16>                                                         │ │
│ │                        │   %1 = stablehlo.add %arg0, %0 : tensor<1x1x1x3xf16>                                                                                                 │ │
│ │                        │   return %1 : tensor<1x1x1x3xf16>                                                                                                                    │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @permute_f718d144(%arg0: tensor<1x3x1x1xf16>) -> (tensor<1x1x1x3xf16> {jax.result_info = "result"}) {                              │ │
│ │                        │   %0 = stablehlo.transpose %arg0, dims = [0, 2, 3, 1] : (tensor<1x3x1x1xf16>) -> tensor<1x1x1x3xf16>                                                 │ │
│ │                        │   return %0 : tensor<1x1x1x3xf16>                                                                                                                    │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @_aten_clone_576ef666(%arg0: tensor<1x3x8x8xf16>) -> (tensor<1x3x8x8xf16> {jax.result_info = "result"}) {                          │ │
│ │                        │   return %arg0 : tensor<1x3x8x8xf16>                                                                                                                 │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @permute_1965d0a5(%arg0: tensor<1x8x8x3xf16>) -> (tensor<1x3x8x8xf16> {jax.result_info = "result"}) {                              │ │
│ │                        │   %0 = stablehlo.transpose %arg0, dims = [0, 3, 1, 2] : (tensor<1x8x8x3xf16>) -> tensor<1x3x8x8xf16>                                                 │ │
│ │                        │   return %0 : tensor<1x3x8x8xf16>                                                                                                                    │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @_aten_clone_9c7bb3b1(%arg0: tensor<1x8x8x3xf16>) -> (tensor<1x8x8x3xf16> {jax.result_info = "result"}) {                          │ │
│ │                        │   return %arg0 : tensor<1x8x8x3xf16>                                                                                                                 │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @permute_48a1a91b(%arg0: tensor<1x3x8x8xf16>) -> (tensor<1x8x8x3xf16> {jax.result_info = "result"}) {                              │ │
│ │                        │   %0 = stablehlo.transpose %arg0, dims = [0, 2, 3, 1] : (tensor<1x3x8x8xf16>) -> tensor<1x8x8x3xf16>                                                 │ │
│ │                        │   return %0 : tensor<1x8x8x3xf16>                                                                                                                    │ │
│ │                          }                                                                                                                                                    │ │
│ │                        }                                                                                                                                                      │ │
│ │                        │                                                                                                                                                      │ │
│ │                        │   ],                                                                                                                                                 │ │
│ │                        │   exported_programs=[<torch.export.exported_program.ExportedProgram object at 0x70758f016ed0>],                                                      │ │
│ │                        │   deduped_tf_vars=[                                                                                                                                  │ │
│ │                        │   │   <tf.Variable 'Variable:0' shape=(3, 3, 8, 8) dtype=float16, numpy=                                                                             │ │
│ │                        array([[[[1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.]],                                                                                                            │ │
│ │                        │   │                                                                                                                                                  │ │
│ │                        │   │   [[0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.]],                                                                                                            │ │
│ │                        │   │                                                                                                                                                  │ │
│ │                        │   │   [[0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.]]],                                                                                                           │ │
│ │                        │                                                                                                                                                      │ │
│ │                        │                                                                                                                                                      │ │
│ │                        │      [[[0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.]],                                                                                                            │ │
│ │                        │   │                                                                                                                                                  │ │
│ │                        │   │   [[1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.]],                                                                                                            │ │
│ │                        │   │                                                                                                                                                  │ │
│ │                        │   │   [[0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.]]],                                                                                                           │ │
│ │                        │                                                                                                                                                      │ │
│ │                        │                                                                                                                                                      │ │
│ │                        │      [[[0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.]],                                                                                                            │ │
│ │                        │   │                                                                                                                                                  │ │
│ │                        │   │   [[0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.]],                                                                                                            │ │
│ │                        │   │                                                                                                                                                  │ │
│ │                        │   │   [[1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.]]]], dtype=float16)>,                                                                                         │ │
│ │                        │   │   <tf.Variable 'Variable:0' shape=(3,) dtype=float16, numpy=array([  0.,   0., -63.], dtype=float16)>                                            │ │
│ │                        │   ]                                                                                                                                                  │ │
│ │                        )                                                                                                                                                      │ │
│ │         quant_config = None                                                                                                                                                   │ │
│ │           signatures = [                                                                                                                                                      │ │
│ │                        │   Signature(                                                                                                                                         │ │
│ │                        │   │   name='serving_default',                                                                                                                        │ │
│ │                        │   │   module=ChannelSum(                                                                                                                             │ │
│ │                          (model): Sequential(                                                                                                                                 │ │
│ │                        │   (0): Conv2d(3, 3, kernel_size=(8, 8), stride=(1, 1))                                                                                               │ │
│ │                        │   (1): Flatten(start_dim=1, end_dim=-1)                                                                                                              │ │
│ │                          )                                                                                                                                                    │ │
│ │                        ),                                                                                                                                                     │ │
│ │                        │   │   sample_args=(                                                                                                                                  │ │
│ │                        │   │   │   tensor([[[[0.2242, 0.5708, 0.4412, 0.5186, 0.8345, 0.3274, 0.3928, 0.3115],                                                                │ │
│ │                        │   │     [0.5356, 0.1130, 0.5576, 0.1142, 0.5210, 0.3452, 0.8330, 0.7769],                                                                            │ │
│ │                        │   │     [0.9229, 0.2600, 0.0228, 0.0102, 0.2001, 0.6299, 0.4463, 0.1573],                                                                            │ │
│ │                        │   │     [0.9575, 0.0464, 0.9526, 0.7910, 0.8643, 0.1527, 0.7744, 0.1738],                                                                            │ │
│ │                        │   │     [0.4961, 0.0108, 0.3154, 0.8872, 0.0312, 0.6836, 0.8784, 0.0626],                                                                            │ │
│ │                        │   │     [0.4290, 0.7617, 0.8066, 0.9487, 0.6323, 0.9097, 0.1315, 0.5625],                                                                            │ │
│ │                        │   │     [0.4634, 0.0064, 0.7871, 0.0856, 0.5605, 0.9312, 0.4043, 0.9639],                                                                            │ │
│ │                        │   │     [0.0401, 0.8530, 0.5469, 0.4165, 0.2891, 0.4739, 0.7246, 0.4492]],                                                                           │ │
│ │                        │   │                                                                                                                                                  │ │
│ │                        │   │    [[0.1454, 0.3369, 0.6572, 0.1093, 0.1587, 0.7847, 0.0925, 0.6128],                                                                            │ │
│ │                        │   │     [0.8818, 0.6548, 0.1865, 0.0245, 0.7939, 0.6162, 0.1663, 0.2517],                                                                            │ │
│ │                        │   │     [0.0661, 0.4475, 0.2710, 0.7075, 0.1103, 0.0815, 0.7104, 0.2666],                                                                            │ │
│ │                        │   │     [0.2242, 0.3438, 0.5215, 0.6177, 0.6367, 0.3428, 0.4038, 0.6880],                                                                            │ │
│ │                        │   │     [0.7373, 0.8755, 0.8247, 0.7300, 0.0629, 0.5967, 0.4841, 0.1891],                                                                            │ │
│ │                        │   │     [0.4858, 0.4626, 0.0817, 0.1769, 0.2593, 0.9536, 0.0531, 0.0929],                                                                            │ │
│ │                        │   │     [0.5977, 0.1196, 0.8604, 0.2375, 0.5088, 0.6191, 0.2006, 0.4861],                                                                            │ │
│ │                        │   │     [0.7827, 0.4663, 0.7163, 0.9614, 0.3567, 0.2026, 0.5708, 0.0399]],                                                                           │ │
│ │                        │   │                                                                                                                                                  │ │
│ │                        │   │    [[0.7285, 0.8984, 0.5801, 0.0434, 0.6372, 0.1530, 0.2090, 0.8311],                                                                            │ │
│ │                        │   │     [0.4846, 0.1665, 0.3721, 0.1958, 0.4236, 0.8745, 0.1119, 0.8706],                                                                            │ │
│ │                        │   │     [0.1417, 0.3594, 0.9180, 0.6519, 0.4282, 0.6577, 0.7593, 0.4688],                                                                            │ │
│ │                        │   │     [0.6968, 0.4346, 0.1372, 0.6826, 0.9360, 0.1997, 0.2629, 0.6782],                                                                            │ │
│ │                        │   │     [0.9863, 0.0658, 0.0928, 0.1555, 0.3301, 0.4783, 0.7593, 0.7456],                                                                            │ │
│ │                        │   │     [0.3877, 0.6187, 0.5024, 0.9014, 0.9692, 0.3625, 0.1598, 0.0953],                                                                            │ │
│ │                        │   │     [0.6489, 0.9419, 0.8828, 0.7886, 0.1780, 0.3416, 0.2223, 0.4111],                                                                            │ │
│ │                        │   │     [0.9277, 0.2686, 0.9629, 0.5698, 0.7510, 0.4182, 0.6743, 0.9048]]]],                                                                         │ │
│ │                        │      dtype=torch.float16),                                                                                                                           │ │
│ │                        │   │   ),                                                                                                                                             │ │
│ │                        │   │   sample_kwargs=None,                                                                                                                            │ │
│ │                        │   │   dynamic_shapes=None                                                                                                                            │ │
│ │                        │   )                                                                                                                                                  │ │
│ │                        ]                                                                                                                                                      │ │
│ ╰───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯ │
│                                                                                                                                                                                   │
│ /data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/ai_edge_torch/lowertools/odml_torch_utils.py:208 in merged_bundle_to_tfl_model       │
│                                                                                                                                                                                   │
│   205 │                                                                                                                                                                           │
│   206 │   conversion_utils.apply_tfl_converter_flags(converter, _tfl_converter_flags)                                                                                             │
│   207 │                                                                                                                                                                           │
│ ❱ 208 │   tflite_model = converter.convert()                                                                                                                                      │
│   209 │                                                                                                                                                                           │
│   210 │   if (                                                                                                                                                                    │
│   211 │   │   quant_config is not None                                                                                                                                            │
│                                                                                                                                                                                   │
│ ╭─────────────────────────────────────────────────────────────────────────────────── locals ────────────────────────────────────────────────────────────────────────────────────╮ │
│ │     _saved_model_dir = None                                                                                                                                                   │ │
│ │ _tfl_converter_flags = {}                                                                                                                                                     │ │
│ │            converter = <tensorflow.lite.python.lite.TFLiteSavedModelConverterV2 object at 0x70751c76eba0>                                                                     │ │
│ │        merged_bundle = MergedBundle(                                                                                                                                          │ │
│ │                        │   bundles=[                                                                                                                                          │ │
│ │                        │   │   module {                                                                                                                                       │ │
│ │                          func.func public @main(%arg0: tensor<3x3x8x8xf16>, %arg1: tensor<3xf16>, %arg2: tensor<1x3x8x8xf16>) -> tensor<1x3xf16> {                            │ │
│ │                        │   %0 = call @permute_48a1a91b(%arg2) : (tensor<1x3x8x8xf16>) -> tensor<1x8x8x3xf16>                                                                  │ │
│ │                        │   %1 = call @_aten_clone_9c7bb3b1(%0) : (tensor<1x8x8x3xf16>) -> tensor<1x8x8x3xf16>                                                                 │ │
│ │                        │   %2 = call @permute_1965d0a5(%1) : (tensor<1x8x8x3xf16>) -> tensor<1x3x8x8xf16>                                                                     │ │
│ │                        │   %3 = call @_aten_clone_576ef666(%2) : (tensor<1x3x8x8xf16>) -> tensor<1x3x8x8xf16>                                                                 │ │
│ │                        │   %4 = stablehlo.convolution(%3, %arg0) dim_numbers = [b, f, 0, 1]x[o, i, 0, 1]->[b, f, 0, 1], window = {stride = [1, 1], pad = [[0, 0], [0, 0]],    │ │
│ │                        rhs_dilate = [1, 1]} {batch_group_count = 1 : i64, feature_group_count = 1 : i64} : (tensor<1x3x8x8xf16>, tensor<3x3x8x8xf16>) -> tensor<1x3x1x1xf16>  │ │
│ │                        │   %5 = call @permute_f718d144(%4) : (tensor<1x3x1x1xf16>) -> tensor<1x1x1x3xf16>                                                                     │ │
│ │                        │   %6 = call @jax_lowering_ee7b84fe(%5, %arg1) : (tensor<1x1x1x3xf16>, tensor<3xf16>) -> tensor<1x1x1x3xf16>                                          │ │
│ │                        │   %7 = call @permute_ceca6e02(%6) : (tensor<1x1x1x3xf16>) -> tensor<1x3x1x1xf16>                                                                     │ │
│ │                        │   %8 = call @permute_71657670(%7) : (tensor<1x3x1x1xf16>) -> tensor<1x1x1x3xf16>                                                                     │ │
│ │                        │   %9 = call @permute_e7832658(%8) : (tensor<1x1x1x3xf16>) -> tensor<1x3x1x1xf16>                                                                     │ │
│ │                        │   %10 = call @_aten_unsafe_view_bee6c7a7(%9) : (tensor<1x3x1x1xf16>) -> tensor<1x3xf16>                                                              │ │
│ │                        │   return %10 : tensor<1x3xf16>                                                                                                                       │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @_aten_unsafe_view_bee6c7a7(%arg0: tensor<1x3x1x1xf16>) -> (tensor<1x3xf16> {jax.result_info = "result"}) {                        │ │
│ │                        │   %0 = stablehlo.reshape %arg0 : (tensor<1x3x1x1xf16>) -> tensor<1x3xf16>                                                                            │ │
│ │                        │   return %0 : tensor<1x3xf16>                                                                                                                        │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @permute_e7832658(%arg0: tensor<1x1x1x3xf16>) -> (tensor<1x3x1x1xf16> {jax.result_info = "result"}) {                              │ │
│ │                        │   %0 = stablehlo.transpose %arg0, dims = [0, 3, 1, 2] : (tensor<1x1x1x3xf16>) -> tensor<1x3x1x1xf16>                                                 │ │
│ │                        │   return %0 : tensor<1x3x1x1xf16>                                                                                                                    │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @permute_71657670(%arg0: tensor<1x3x1x1xf16>) -> (tensor<1x1x1x3xf16> {jax.result_info = "result"}) {                              │ │
│ │                        │   %0 = stablehlo.transpose %arg0, dims = [0, 2, 3, 1] : (tensor<1x3x1x1xf16>) -> tensor<1x1x1x3xf16>                                                 │ │
│ │                        │   return %0 : tensor<1x1x1x3xf16>                                                                                                                    │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @permute_ceca6e02(%arg0: tensor<1x1x1x3xf16>) -> (tensor<1x3x1x1xf16> {jax.result_info = "result"}) {                              │ │
│ │                        │   %0 = stablehlo.transpose %arg0, dims = [0, 3, 1, 2] : (tensor<1x1x1x3xf16>) -> tensor<1x3x1x1xf16>                                                 │ │
│ │                        │   return %0 : tensor<1x3x1x1xf16>                                                                                                                    │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @jax_lowering_ee7b84fe(%arg0: tensor<1x1x1x3xf16>, %arg1: tensor<3xf16>) -> (tensor<1x1x1x3xf16> {jax.result_info = "result"}) {   │ │
│ │                        │   %0 = stablehlo.broadcast_in_dim %arg1, dims = [3] : (tensor<3xf16>) -> tensor<1x1x1x3xf16>                                                         │ │
│ │                        │   %1 = stablehlo.add %arg0, %0 : tensor<1x1x1x3xf16>                                                                                                 │ │
│ │                        │   return %1 : tensor<1x1x1x3xf16>                                                                                                                    │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @permute_f718d144(%arg0: tensor<1x3x1x1xf16>) -> (tensor<1x1x1x3xf16> {jax.result_info = "result"}) {                              │ │
│ │                        │   %0 = stablehlo.transpose %arg0, dims = [0, 2, 3, 1] : (tensor<1x3x1x1xf16>) -> tensor<1x1x1x3xf16>                                                 │ │
│ │                        │   return %0 : tensor<1x1x1x3xf16>                                                                                                                    │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @_aten_clone_576ef666(%arg0: tensor<1x3x8x8xf16>) -> (tensor<1x3x8x8xf16> {jax.result_info = "result"}) {                          │ │
│ │                        │   return %arg0 : tensor<1x3x8x8xf16>                                                                                                                 │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @permute_1965d0a5(%arg0: tensor<1x8x8x3xf16>) -> (tensor<1x3x8x8xf16> {jax.result_info = "result"}) {                              │ │
│ │                        │   %0 = stablehlo.transpose %arg0, dims = [0, 3, 1, 2] : (tensor<1x8x8x3xf16>) -> tensor<1x3x8x8xf16>                                                 │ │
│ │                        │   return %0 : tensor<1x3x8x8xf16>                                                                                                                    │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @_aten_clone_9c7bb3b1(%arg0: tensor<1x8x8x3xf16>) -> (tensor<1x8x8x3xf16> {jax.result_info = "result"}) {                          │ │
│ │                        │   return %arg0 : tensor<1x8x8x3xf16>                                                                                                                 │ │
│ │                          }                                                                                                                                                    │ │
│ │                          func.func private @permute_48a1a91b(%arg0: tensor<1x3x8x8xf16>) -> (tensor<1x8x8x3xf16> {jax.result_info = "result"}) {                              │ │
│ │                        │   %0 = stablehlo.transpose %arg0, dims = [0, 2, 3, 1] : (tensor<1x3x8x8xf16>) -> tensor<1x8x8x3xf16>                                                 │ │
│ │                        │   return %0 : tensor<1x8x8x3xf16>                                                                                                                    │ │
│ │                          }                                                                                                                                                    │ │
│ │                        }                                                                                                                                                      │ │
│ │                        │                                                                                                                                                      │ │
│ │                        │   ],                                                                                                                                                 │ │
│ │                        │   exported_programs=[<torch.export.exported_program.ExportedProgram object at 0x70758f016ed0>],                                                      │ │
│ │                        │   deduped_tf_vars=[                                                                                                                                  │ │
│ │                        │   │   <tf.Variable 'Variable:0' shape=(3, 3, 8, 8) dtype=float16, numpy=                                                                             │ │
│ │                        array([[[[1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.]],                                                                                                            │ │
│ │                        │   │                                                                                                                                                  │ │
│ │                        │   │   [[0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.]],                                                                                                            │ │
│ │                        │   │                                                                                                                                                  │ │
│ │                        │   │   [[0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.]]],                                                                                                           │ │
│ │                        │                                                                                                                                                      │ │
│ │                        │                                                                                                                                                      │ │
│ │                        │      [[[0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.]],                                                                                                            │ │
│ │                        │   │                                                                                                                                                  │ │
│ │                        │   │   [[1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.]],                                                                                                            │ │
│ │                        │   │                                                                                                                                                  │ │
│ │                        │   │   [[0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.]]],                                                                                                           │ │
│ │                        │                                                                                                                                                      │ │
│ │                        │                                                                                                                                                      │ │
│ │                        │      [[[0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.]],                                                                                                            │ │
│ │                        │   │                                                                                                                                                  │ │
│ │                        │   │   [[0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.]],                                                                                                            │ │
│ │                        │   │                                                                                                                                                  │ │
│ │                        │   │   [[1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.]]]], dtype=float16)>,                                                                                         │ │
│ │                        │   │   <tf.Variable 'Variable:0' shape=(3,) dtype=float16, numpy=array([  0.,   0., -63.], dtype=float16)>                                            │ │
│ │                        │   ]                                                                                                                                                  │ │
│ │                        )                                                                                                                                                      │ │
│ │         quant_config = None                                                                                                                                                   │ │
│ │           signatures = [                                                                                                                                                      │ │
│ │                        │   Signature(                                                                                                                                         │ │
│ │                        │   │   name='serving_default',                                                                                                                        │ │
│ │                        │   │   module=ChannelSum(                                                                                                                             │ │
│ │                          (model): Sequential(                                                                                                                                 │ │
│ │                        │   (0): Conv2d(3, 3, kernel_size=(8, 8), stride=(1, 1))                                                                                               │ │
│ │                        │   (1): Flatten(start_dim=1, end_dim=-1)                                                                                                              │ │
│ │                          )                                                                                                                                                    │ │
│ │                        ),                                                                                                                                                     │ │
│ │                        │   │   sample_args=(                                                                                                                                  │ │
│ │                        │   │   │   tensor([[[[0.2242, 0.5708, 0.4412, 0.5186, 0.8345, 0.3274, 0.3928, 0.3115],                                                                │ │
│ │                        │   │     [0.5356, 0.1130, 0.5576, 0.1142, 0.5210, 0.3452, 0.8330, 0.7769],                                                                            │ │
│ │                        │   │     [0.9229, 0.2600, 0.0228, 0.0102, 0.2001, 0.6299, 0.4463, 0.1573],                                                                            │ │
│ │                        │   │     [0.9575, 0.0464, 0.9526, 0.7910, 0.8643, 0.1527, 0.7744, 0.1738],                                                                            │ │
│ │                        │   │     [0.4961, 0.0108, 0.3154, 0.8872, 0.0312, 0.6836, 0.8784, 0.0626],                                                                            │ │
│ │                        │   │     [0.4290, 0.7617, 0.8066, 0.9487, 0.6323, 0.9097, 0.1315, 0.5625],                                                                            │ │
│ │                        │   │     [0.4634, 0.0064, 0.7871, 0.0856, 0.5605, 0.9312, 0.4043, 0.9639],                                                                            │ │
│ │                        │   │     [0.0401, 0.8530, 0.5469, 0.4165, 0.2891, 0.4739, 0.7246, 0.4492]],                                                                           │ │
│ │                        │   │                                                                                                                                                  │ │
│ │                        │   │    [[0.1454, 0.3369, 0.6572, 0.1093, 0.1587, 0.7847, 0.0925, 0.6128],                                                                            │ │
│ │                        │   │     [0.8818, 0.6548, 0.1865, 0.0245, 0.7939, 0.6162, 0.1663, 0.2517],                                                                            │ │
│ │                        │   │     [0.0661, 0.4475, 0.2710, 0.7075, 0.1103, 0.0815, 0.7104, 0.2666],                                                                            │ │
│ │                        │   │     [0.2242, 0.3438, 0.5215, 0.6177, 0.6367, 0.3428, 0.4038, 0.6880],                                                                            │ │
│ │                        │   │     [0.7373, 0.8755, 0.8247, 0.7300, 0.0629, 0.5967, 0.4841, 0.1891],                                                                            │ │
│ │                        │   │     [0.4858, 0.4626, 0.0817, 0.1769, 0.2593, 0.9536, 0.0531, 0.0929],                                                                            │ │
│ │                        │   │     [0.5977, 0.1196, 0.8604, 0.2375, 0.5088, 0.6191, 0.2006, 0.4861],                                                                            │ │
│ │                        │   │     [0.7827, 0.4663, 0.7163, 0.9614, 0.3567, 0.2026, 0.5708, 0.0399]],                                                                           │ │
│ │                        │   │                                                                                                                                                  │ │
│ │                        │   │    [[0.7285, 0.8984, 0.5801, 0.0434, 0.6372, 0.1530, 0.2090, 0.8311],                                                                            │ │
│ │                        │   │     [0.4846, 0.1665, 0.3721, 0.1958, 0.4236, 0.8745, 0.1119, 0.8706],                                                                            │ │
│ │                        │   │     [0.1417, 0.3594, 0.9180, 0.6519, 0.4282, 0.6577, 0.7593, 0.4688],                                                                            │ │
│ │                        │   │     [0.6968, 0.4346, 0.1372, 0.6826, 0.9360, 0.1997, 0.2629, 0.6782],                                                                            │ │
│ │                        │   │     [0.9863, 0.0658, 0.0928, 0.1555, 0.3301, 0.4783, 0.7593, 0.7456],                                                                            │ │
│ │                        │   │     [0.3877, 0.6187, 0.5024, 0.9014, 0.9692, 0.3625, 0.1598, 0.0953],                                                                            │ │
│ │                        │   │     [0.6489, 0.9419, 0.8828, 0.7886, 0.1780, 0.3416, 0.2223, 0.4111],                                                                            │ │
│ │                        │   │     [0.9277, 0.2686, 0.9629, 0.5698, 0.7510, 0.4182, 0.6743, 0.9048]]]],                                                                         │ │
│ │                        │      dtype=torch.float16),                                                                                                                           │ │
│ │                        │   │   ),                                                                                                                                             │ │
│ │                        │   │   sample_kwargs=None,                                                                                                                            │ │
│ │                        │   │   dynamic_shapes=None                                                                                                                            │ │
│ │                        │   )                                                                                                                                                  │ │
│ │                        ]                                                                                                                                                      │ │
│ │        temp_dir_path = '/tmp/tmp3ykp77vo'                                                                                                                                     │ │
│ │    tf_concrete_funcs = [                                                                                                                                                      │ │
│ │                        │   <ConcreteFunction (args_0: TensorSpec(shape=(1, 3, 8, 8), dtype=tf.float16, name='args_0'), /) -> List[TensorSpec(shape=(1, 3), dtype=tf.float16,  │ │
│ │                        name=None)] at 0x70751C738230>                                                                                                                         │ │
│ │                        ]                                                                                                                                                      │ │
│ │         tf_functions = [<function _wrap_as_tf_func.<locals>.inner at 0x70751c723240>]                                                                                         │ │
│ │            tf_module = <tensorflow.python.module.module.Module object at 0x70751c83a720>                                                                                      │ │
│ │               tf_sig = [TensorSpec(shape=(1, 3, 8, 8), dtype=tf.float16, name='args_0')]                                                                                      │ │
│ │        tf_signatures = [[TensorSpec(shape=(1, 3, 8, 8), dtype=tf.float16, name='args_0')]]                                                                                    │ │
│ │        tf_state_dict = {                                                                                                                                                      │ │
│ │                        │   'serving_default_model.0.weight': <tf.Variable 'Variable:0' shape=(3, 3, 8, 8) dtype=float16, numpy=                                               │ │
│ │                        array([[[[1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.]],                                                                                                            │ │
│ │                        │   │                                                                                                                                                  │ │
│ │                        │   │   [[0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.]],                                                                                                            │ │
│ │                        │   │                                                                                                                                                  │ │
│ │                        │   │   [[0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.]]],                                                                                                           │ │
│ │                        │                                                                                                                                                      │ │
│ │                        │                                                                                                                                                      │ │
│ │                        │      [[[0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.]],                                                                                                            │ │
│ │                        │   │                                                                                                                                                  │ │
│ │                        │   │   [[1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.]],                                                                                                            │ │
│ │                        │   │                                                                                                                                                  │ │
│ │                        │   │   [[0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.]]],                                                                                                           │ │
│ │                        │                                                                                                                                                      │ │
│ │                        │                                                                                                                                                      │ │
│ │                        │      [[[0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.]],                                                                                                            │ │
│ │                        │   │                                                                                                                                                  │ │
│ │                        │   │   [[0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.],                                                                                                             │ │
│ │                        │   │    [0., 0., 0., 0., 0., 0., 0., 0.]],                                                                                                            │ │
│ │                        │   │                                                                                                                                                  │ │
│ │                        │   │   [[1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.],                                                                                                             │ │
│ │                        │   │    [1., 1., 1., 1., 1., 1., 1., 1.]]]], dtype=float16)>,                                                                                         │ │
│ │                        │   'serving_default_model.0.bias': <tf.Variable 'Variable:0' shape=(3,) dtype=float16, numpy=array([  0.,   0., -63.], dtype=float16)>                │ │
│ │                        }                                                                                                                                                      │ │
│ ╰───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯ │
│                                                                                                                                                                                   │
│ /data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/tensorflow/lite/python/lite.py:1254 in wrapper                                       │
│                                                                                                                                                                                   │
│   1251   @functools.wraps(convert_func)                                                                                                                                           │
│   1252   def wrapper(self, *args, **kwargs):                                                                                                                                      │
│   1253 │   # pylint: disable=protected-access                                                                                                                                     │
│ ❱ 1254 │   return self._convert_and_export_metrics(convert_func, *args, **kwargs)                                                                                                 │
│   1255 │   # pylint: enable=protected-access                                                                                                                                      │
│   1256                                                                                                                                                                            │
│   1257   return wrapper                                                                                                                                                           │
│                                                                                                                                                                                   │
│ ╭────────────────────────────────────────── locals ───────────────────────────────────────────╮                                                                                   │
│ │   args = ()                                                                                 │                                                                                   │
│ │ kwargs = {}                                                                                 │                                                                                   │
│ │   self = <tensorflow.lite.python.lite.TFLiteSavedModelConverterV2 object at 0x70751c76eba0> │                                                                                   │
│ ╰─────────────────────────────────────────────────────────────────────────────────────────────╯                                                                                   │
│                                                                                                                                                                                   │
│ /data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/tensorflow/lite/python/lite.py:1206 in _convert_and_export_metrics                   │
│                                                                                                                                                                                   │
│   1203 │   self._increase_conversion_attempt_metric()                                                                                                                             │
│   1204 │   self._save_conversion_params_metric()                                                                                                                                  │
│   1205 │   start_time = time.process_time()                                                                                                                                       │
│ ❱ 1206 │   result = convert_func(self, *args, **kwargs)                                                                                                                           │
│   1207 │   elapsed_time_ms = (time.process_time() - start_time) * 1000                                                                                                            │
│   1208 │   if result:                                                                                                                                                             │
│   1209 │     self._increase_conversion_success_metric()                                                                                                                           │
│                                                                                                                                                                                   │
│ ╭──────────────────────────────────────────── locals ─────────────────────────────────────────────╮                                                                               │
│ │       args = ()                                                                                 │                                                                               │
│ │     kwargs = {}                                                                                 │                                                                               │
│ │       self = <tensorflow.lite.python.lite.TFLiteSavedModelConverterV2 object at 0x70751c76eba0> │                                                                               │
│ │ start_time = 10.589068047                                                                       │                                                                               │
│ ╰─────────────────────────────────────────────────────────────────────────────────────────────────╯                                                                               │
│                                                                                                                                                                                   │
│ /data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/tensorflow/lite/python/lite.py:1590 in convert                                       │
│                                                                                                                                                                                   │
│   1587 │                                                                                                                                                                          │
│   1588 │   del trackable_obj                                                                                                                                                      │
│   1589 │   gc.collect()                                                                                                                                                           │
│ ❱ 1590 │   return self._convert_from_saved_model(graph_def)                                                                                                                       │
│   1591                                                                                                                                                                            │
│   1592                                                                                                                                                                            │
│   1593 class TFLiteKerasModelConverterV2(TFLiteConverterBaseV2):                                                                                                                  │
│                                                                                                                                                                                   │
│ ╭─────────────────────────────────────────────────────────────────────────────────── locals ────────────────────────────────────────────────────────────────────────────────────╮ │
│ │      graph_def = node {                                                                                                                                                       │ │
│ │                    name: "Variable"                                                                                                                                           │ │
│ │                    op: "VarHandleOp"                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "shared_name"                                                                                                                                       │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     s: "Variable"                                                                                                                                          │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "shape"                                                                                                                                             │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     shape {                                                                                                                                                │ │
│ │                  │   │   dim {                                                                                                                                                │ │
│ │                  │   │     size: 3                                                                                                                                            │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "dtype"                                                                                                                                             │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     type: DT_HALF                                                                                                                                          │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "debug_name"                                                                                                                                        │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     s: "Variable/"                                                                                                                                         │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "container"                                                                                                                                         │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     s: ""                                                                                                                                                  │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "allowed_devices"                                                                                                                                   │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     list {                                                                                                                                                 │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "_output_shapes"                                                                                                                                    │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     list {                                                                                                                                                 │ │
│ │                  │   │   shape {                                                                                                                                              │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                  }                                                                                                                                                            │ │
│ │                  node {                                                                                                                                                       │ │
│ │                    name: "Variable/Read/ReadVariableOp"                                                                                                                       │ │
│ │                    op: "ReadVariableOp"                                                                                                                                       │ │
│ │                    input: "Variable"                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "dtype"                                                                                                                                             │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     type: DT_HALF                                                                                                                                          │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "_output_shapes"                                                                                                                                    │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     list {                                                                                                                                                 │ │
│ │                  │   │   shape {                                                                                                                                              │ │
│ │                  │   │     dim {                                                                                                                                              │ │
│ │                  │   │   │   size: 3                                                                                                                                          │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                  }                                                                                                                                                            │ │
│ │                  node {                                                                                                                                                       │ │
│ │                    name: "Variable_1"                                                                                                                                         │ │
│ │                    op: "VarHandleOp"                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "shared_name"                                                                                                                                       │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     s: "Variable_1"                                                                                                                                        │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "shape"                                                                                                                                             │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     shape {                                                                                                                                                │ │
│ │                  │   │   dim {                                                                                                                                                │ │
│ │                  │   │     size: 3                                                                                                                                            │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │   │   dim {                                                                                                                                                │ │
│ │                  │   │     size: 3                                                                                                                                            │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │   │   dim {                                                                                                                                                │ │
│ │                  │   │     size: 8                                                                                                                                            │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │   │   dim {                                                                                                                                                │ │
│ │                  │   │     size: 8                                                                                                                                            │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "dtype"                                                                                                                                             │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     type: DT_HALF                                                                                                                                          │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "debug_name"                                                                                                                                        │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     s: "Variable_1/"                                                                                                                                       │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "container"                                                                                                                                         │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     s: ""                                                                                                                                                  │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "allowed_devices"                                                                                                                                   │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     list {                                                                                                                                                 │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "_output_shapes"                                                                                                                                    │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     list {                                                                                                                                                 │ │
│ │                  │   │   shape {                                                                                                                                              │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                  }                                                                                                                                                            │ │
│ │                  node {                                                                                                                                                       │ │
│ │                    name: "Variable_1/Read/ReadVariableOp"                                                                                                                     │ │
│ │                    op: "ReadVariableOp"                                                                                                                                       │ │
│ │                    input: "Variable_1"                                                                                                                                        │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "dtype"                                                                                                                                             │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     type: DT_HALF                                                                                                                                          │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "_output_shapes"                                                                                                                                    │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     list {                                                                                                                                                 │ │
│ │                  │   │   shape {                                                                                                                                              │ │
│ │                  │   │     dim {                                                                                                                                              │ │
│ │                  │   │   │   size: 3                                                                                                                                          │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │     dim {                                                                                                                                              │ │
│ │                  │   │   │   size: 3                                                                                                                                          │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │     dim {                                                                                                                                              │ │
│ │                  │   │   │   size: 8                                                                                                                                          │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │     dim {                                                                                                                                              │ │
│ │                  │   │   │   size: 8                                                                                                                                          │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                  }                                                                                                                                                            │ │
│ │                  node {                                                                                                                                                       │ │
│ │                    name: "serving_default_args_0"                                                                                                                             │ │
│ │                    op: "Placeholder"                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "shape"                                                                                                                                             │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     shape {                                                                                                                                                │ │
│ │                  │   │   dim {                                                                                                                                                │ │
│ │                  │   │     size: 1                                                                                                                                            │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │   │   dim {                                                                                                                                                │ │
│ │                  │   │     size: 3                                                                                                                                            │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │   │   dim {                                                                                                                                                │ │
│ │                  │   │     size: 8                                                                                                                                            │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │   │   dim {                                                                                                                                                │ │
│ │                  │   │     size: 8                                                                                                                                            │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "dtype"                                                                                                                                             │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     type: DT_HALF                                                                                                                                          │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "_output_shapes"                                                                                                                                    │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     list {                                                                                                                                                 │ │
│ │                  │   │   shape {                                                                                                                                              │ │
│ │                  │   │     dim {                                                                                                                                              │ │
│ │                  │   │   │   size: 1                                                                                                                                          │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │     dim {                                                                                                                                              │ │
│ │                  │   │   │   size: 3                                                                                                                                          │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │     dim {                                                                                                                                              │ │
│ │                  │   │   │   size: 8                                                                                                                                          │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │     dim {                                                                                                                                              │ │
│ │                  │   │   │   size: 8                                                                                                                                          │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                  }                                                                                                                                                            │ │
│ │                  node {                                                                                                                                                       │ │
│ │                    name: "StatefulPartitionedCall"                                                                                                                            │ │
│ │                    op: "StatefulPartitionedCall"                                                                                                                              │ │
│ │                    input: "serving_default_args_0"                                                                                                                            │ │
│ │                    input: "Variable_1"                                                                                                                                        │ │
│ │                    input: "Variable"                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "f"                                                                                                                                                 │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     func {                                                                                                                                                 │ │
│ │                  │   │   name: "__inference_signature_wrapper_25"                                                                                                             │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "executor_type"                                                                                                                                     │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     s: ""                                                                                                                                                  │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "config"                                                                                                                                            │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     s: ""                                                                                                                                                  │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "config_proto"                                                                                                                                      │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     s: "\n\007\n\003CPU\020\001\n\007\n\003GPU\020\0002\002J\0008\001\202\001\000\222\001\002J\000"                                                        │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "_read_only_resource_inputs"                                                                                                                        │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     list {                                                                                                                                                 │ │
│ │                  │   │   i: 1                                                                                                                                                 │ │
│ │                  │   │   i: 2                                                                                                                                                 │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "_output_shapes"                                                                                                                                    │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     list {                                                                                                                                                 │ │
│ │                  │   │   shape {                                                                                                                                              │ │
│ │                  │   │     dim {                                                                                                                                              │ │
│ │                  │   │   │   size: 1                                                                                                                                          │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │     dim {                                                                                                                                              │ │
│ │                  │   │   │   size: 3                                                                                                                                          │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "_collective_manager_ids"                                                                                                                           │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     list {                                                                                                                                                 │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "Tout"                                                                                                                                              │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     list {                                                                                                                                                 │ │
│ │                  │   │   type: DT_HALF                                                                                                                                        │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "Tin"                                                                                                                                               │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     list {                                                                                                                                                 │ │
│ │                  │   │   type: DT_HALF                                                                                                                                        │ │
│ │                  │   │   type: DT_RESOURCE                                                                                                                                    │ │
│ │                  │   │   type: DT_RESOURCE                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                  }                                                                                                                                                            │ │
│ │                  node {                                                                                                                                                       │ │
│ │                    name: "NoOp"                                                                                                                                               │ │
│ │                    op: "NoOp"                                                                                                                                                 │ │
│ │                  }                                                                                                                                                            │ │
│ │                  node {                                                                                                                                                       │ │
│ │                    name: "Const"                                                                                                                                              │ │
│ │                    op: "Const"                                                                                                                                                │ │
│ │                    device: "/device:CPU:0"                                                                                                                                    │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "value"                                                                                                                                             │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     tensor {                                                                                                                                               │ │
│ │                  │   │   dtype: DT_STRING                                                                                                                                     │ │
│ │                  │   │   tensor_shape {                                                                                                                                       │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │   │   string_val:                                                                                                                                          │ │
│ │                  "\n+\n\005\010\001\022\001f\n\016\010\002\022\n_variables\n\016\010\003\022\nsignatures*\002\010\001\n\t\n\005\010\004\022\0010*\000\n\022\n\005\010\005\02… │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "dtype"                                                                                                                                             │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     type: DT_STRING                                                                                                                                        │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "_output_shapes"                                                                                                                                    │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     list {                                                                                                                                                 │ │
│ │                  │   │   shape {                                                                                                                                              │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                  }                                                                                                                                                            │ │
│ │                  node {                                                                                                                                                       │ │
│ │                    name: "saver_filename"                                                                                                                                     │ │
│ │                    op: "Placeholder"                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "shape"                                                                                                                                             │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     shape {                                                                                                                                                │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "dtype"                                                                                                                                             │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     type: DT_STRING                                                                                                                                        │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "_output_shapes"                                                                                                                                    │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     list {                                                                                                                                                 │ │
│ │                  │   │   shape {                                                                                                                                              │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                  }                                                                                                                                                            │ │
│ │                  node {                                                                                                                                                       │ │
│ │                    name: "StatefulPartitionedCall_1"                                                                                                                          │ │
│ │                    op: "StatefulPartitionedCall"                                                                                                                              │ │
│ │                    input: "saver_filename"                                                                                                                                    │ │
│ │                    input: "Variable_1"                                                                                                                                        │ │
│ │                    input: "Variable"                                                                                                                                          │ │
│ │                    input: "Const"                                                                                                                                             │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "f"                                                                                                                                                 │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     func {                                                                                                                                                 │ │
│ │                  │   │   name: "__inference__traced_save_68"                                                                                                                  │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "executor_type"                                                                                                                                     │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     s: ""                                                                                                                                                  │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "config"                                                                                                                                            │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     s: ""                                                                                                                                                  │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "config_proto"                                                                                                                                      │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     s: "\n\007\n\003CPU\020\001\n\007\n\003GPU\020\0002\002J\0008\001\202\001\000\222\001\002J\000"                                                        │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "_read_only_resource_inputs"                                                                                                                        │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     list {                                                                                                                                                 │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "_output_shapes"                                                                                                                                    │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     list {                                                                                                                                                 │ │
│ │                  │   │   shape {                                                                                                                                              │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "_collective_manager_ids"                                                                                                                           │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     list {                                                                                                                                                 │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "Tout"                                                                                                                                              │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     list {                                                                                                                                                 │ │
│ │                  │   │   type: DT_STRING                                                                                                                                      │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "Tin"                                                                                                                                               │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     list {                                                                                                                                                 │ │
│ │                  │   │   type: DT_STRING                                                                                                                                      │ │
│ │                  │   │   type: DT_RESOURCE                                                                                                                                    │ │
│ │                  │   │   type: DT_RESOURCE                                                                                                                                    │ │
│ │                  │   │   type: DT_STRING                                                                                                                                      │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                  }                                                                                                                                                            │ │
│ │                  node {                                                                                                                                                       │ │
│ │                    name: "StatefulPartitionedCall_2"                                                                                                                          │ │
│ │                    op: "StatefulPartitionedCall"                                                                                                                              │ │
│ │                    input: "saver_filename"                                                                                                                                    │ │
│ │                    input: "Variable_1"                                                                                                                                        │ │
│ │                    input: "Variable"                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "f"                                                                                                                                                 │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     func {                                                                                                                                                 │ │
│ │                  │   │   name: "__inference__traced_restore_83"                                                                                                               │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "executor_type"                                                                                                                                     │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     s: ""                                                                                                                                                  │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "config"                                                                                                                                            │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     s: ""                                                                                                                                                  │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "config_proto"                                                                                                                                      │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     s: "\n\007\n\003CPU\020\001\n\007\n\003GPU\020\0002\002J\0008\001\202\001\000\222\001\002J\000"                                                        │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "_read_only_resource_inputs"                                                                                                                        │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     list {                                                                                                                                                 │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "_output_shapes"                                                                                                                                    │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     list {                                                                                                                                                 │ │
│ │                  │   │   shape {                                                                                                                                              │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "_collective_manager_ids"                                                                                                                           │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     list {                                                                                                                                                 │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "Tout"                                                                                                                                              │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     list {                                                                                                                                                 │ │
│ │                  │   │   type: DT_STRING                                                                                                                                      │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    attr {                                                                                                                                                     │ │
│ │                  │   key: "Tin"                                                                                                                                               │ │
│ │                  │   value {                                                                                                                                                  │ │
│ │                  │     list {                                                                                                                                                 │ │
│ │                  │   │   type: DT_STRING                                                                                                                                      │ │
│ │                  │   │   type: DT_RESOURCE                                                                                                                                    │ │
│ │                  │   │   type: DT_RESOURCE                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                  }                                                                                                                                                            │ │
│ │                  library {                                                                                                                                                    │ │
│ │                    function {                                                                                                                                                 │ │
│ │                  │   signature {                                                                                                                                              │ │
│ │                  │     name: "__inference_signature_wrapper_25"                                                                                                               │ │
│ │                  │     input_arg {                                                                                                                                            │ │
│ │                  │   │   name: "args_0"                                                                                                                                       │ │
│ │                  │   │   type: DT_HALF                                                                                                                                        │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     input_arg {                                                                                                                                            │ │
│ │                  │   │   name: "unknown"                                                                                                                                      │ │
│ │                  │   │   type: DT_RESOURCE                                                                                                                                    │ │
│ │                  │   │   handle_data {                                                                                                                                        │ │
│ │                  │   │     dtype: DT_HALF                                                                                                                                     │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 3                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 3                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 8                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 8                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     input_arg {                                                                                                                                            │ │
│ │                  │   │   name: "unknown_0"                                                                                                                                    │ │
│ │                  │   │   type: DT_RESOURCE                                                                                                                                    │ │
│ │                  │   │   handle_data {                                                                                                                                        │ │
│ │                  │   │     dtype: DT_HALF                                                                                                                                     │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 3                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     output_arg {                                                                                                                                           │ │
│ │                  │   │   name: "identity"                                                                                                                                     │ │
│ │                  │   │   type: DT_HALF                                                                                                                                        │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     is_stateful: true                                                                                                                                      │ │
│ │                  │     control_output: "StatefulPartitionedCall"                                                                                                              │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "StatefulPartitionedCall"                                                                                                                        │ │
│ │                  │     op: "StatefulPartitionedCall"                                                                                                                          │ │
│ │                  │     input: "args_0"                                                                                                                                        │ │
│ │                  │     input: "unknown"                                                                                                                                       │ │
│ │                  │     input: "unknown_0"                                                                                                                                     │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "f"                                                                                                                                             │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     func {                                                                                                                                             │ │
│ │                  │   │   │   name: "__inference_inner_15"                                                                                                                     │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "config_proto"                                                                                                                                  │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     s: "\n\007\n\003CPU\020\001\n\007\n\003GPU\020\0002\002J\0008\001\202\001\000\222\001\002J\000"                                                    │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_read_only_resource_inputs"                                                                                                                    │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   i: 1                                                                                                                                             │ │
│ │                  │   │   │   i: 2                                                                                                                                             │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 1                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_collective_manager_ids"                                                                                                                       │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "Tout"                                                                                                                                          │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   type: DT_HALF                                                                                                                                    │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "Tin"                                                                                                                                           │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   type: DT_HALF                                                                                                                                    │ │
│ │                  │   │   │   type: DT_RESOURCE                                                                                                                                │ │
│ │                  │   │   │   type: DT_RESOURCE                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "Identity"                                                                                                                                       │ │
│ │                  │     op: "Identity"                                                                                                                                         │ │
│ │                  │     input: "StatefulPartitionedCall:output:0"                                                                                                              │ │
│ │                  │     input: "^NoOp"                                                                                                                                         │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 1                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "T"                                                                                                                                             │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_HALF                                                                                                                                      │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "NoOp"                                                                                                                                           │ │
│ │                  │     op: "NoOp"                                                                                                                                             │ │
│ │                  │     input: "^StatefulPartitionedCall"                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   ret {                                                                                                                                                    │ │
│ │                  │     key: "identity"                                                                                                                                        │ │
│ │                  │     value: "Identity:output:0"                                                                                                                             │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   attr {                                                                                                                                                   │ │
│ │                  │     key: "_input_shapes"                                                                                                                                   │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   list {                                                                                                                                               │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 1                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 3                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 8                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 8                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   attr {                                                                                                                                                   │ │
│ │                  │     key: "_construction_context"                                                                                                                           │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   s: "kEagerRuntime"                                                                                                                                   │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   control_ret {                                                                                                                                            │ │
│ │                  │     key: "StatefulPartitionedCall"                                                                                                                         │ │
│ │                  │     value: "StatefulPartitionedCall"                                                                                                                       │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   arg_attr {                                                                                                                                               │ │
│ │                  │     key: 0                                                                                                                                                 │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   attr {                                                                                                                                               │ │
│ │                  │   │     key: "_user_specified_name"                                                                                                                        │ │
│ │                  │   │     value {                                                                                                                                            │ │
│ │                  │   │   │   s: "args_0"                                                                                                                                      │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │   │   attr {                                                                                                                                               │ │
│ │                  │   │     key: "_output_shapes"                                                                                                                              │ │
│ │                  │   │     value {                                                                                                                                            │ │
│ │                  │   │   │   list {                                                                                                                                           │ │
│ │                  │   │   │     shape {                                                                                                                                        │ │
│ │                  │   │   │   │   dim {                                                                                                                                        │ │
│ │                  │   │   │   │     size: 1                                                                                                                                    │ │
│ │                  │   │   │   │   }                                                                                                                                            │ │
│ │                  │   │   │   │   dim {                                                                                                                                        │ │
│ │                  │   │   │   │     size: 3                                                                                                                                    │ │
│ │                  │   │   │   │   }                                                                                                                                            │ │
│ │                  │   │   │   │   dim {                                                                                                                                        │ │
│ │                  │   │   │   │     size: 8                                                                                                                                    │ │
│ │                  │   │   │   │   }                                                                                                                                            │ │
│ │                  │   │   │   │   dim {                                                                                                                                        │ │
│ │                  │   │   │   │     size: 8                                                                                                                                    │ │
│ │                  │   │   │   │   }                                                                                                                                            │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   arg_attr {                                                                                                                                               │ │
│ │                  │     key: 1                                                                                                                                                 │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   attr {                                                                                                                                               │ │
│ │                  │   │     key: "_user_specified_name"                                                                                                                        │ │
│ │                  │   │     value {                                                                                                                                            │ │
│ │                  │   │   │   s: "19"                                                                                                                                          │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   arg_attr {                                                                                                                                               │ │
│ │                  │     key: 2                                                                                                                                                 │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   attr {                                                                                                                                               │ │
│ │                  │   │     key: "_user_specified_name"                                                                                                                        │ │
│ │                  │   │     value {                                                                                                                                            │ │
│ │                  │   │   │   s: "21"                                                                                                                                          │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    function {                                                                                                                                                 │ │
│ │                  │   signature {                                                                                                                                              │ │
│ │                  │     name: "__inference_inner_15"                                                                                                                           │ │
│ │                  │     input_arg {                                                                                                                                            │ │
│ │                  │   │   name: "args_0"                                                                                                                                       │ │
│ │                  │   │   type: DT_HALF                                                                                                                                        │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     input_arg {                                                                                                                                            │ │
│ │                  │   │   name: "xlacallmodule_readvariableop_resource"                                                                                                        │ │
│ │                  │   │   type: DT_RESOURCE                                                                                                                                    │ │
│ │                  │   │   handle_data {                                                                                                                                        │ │
│ │                  │   │     dtype: DT_HALF                                                                                                                                     │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 3                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 3                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 8                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 8                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     input_arg {                                                                                                                                            │ │
│ │                  │   │   name: "xlacallmodule_readvariableop_1_resource"                                                                                                      │ │
│ │                  │   │   type: DT_RESOURCE                                                                                                                                    │ │
│ │                  │   │   handle_data {                                                                                                                                        │ │
│ │                  │   │     dtype: DT_HALF                                                                                                                                     │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 3                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     output_arg {                                                                                                                                           │ │
│ │                  │   │   name: "identity"                                                                                                                                     │ │
│ │                  │   │   type: DT_HALF                                                                                                                                        │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     is_stateful: true                                                                                                                                      │ │
│ │                  │     control_output: "XlaCallModule"                                                                                                                        │ │
│ │                  │     control_output: "XlaCallModule/ReadVariableOp"                                                                                                         │ │
│ │                  │     control_output: "XlaCallModule/ReadVariableOp_1"                                                                                                       │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "XlaCallModule/ReadVariableOp"                                                                                                                   │ │
│ │                  │     op: "ReadVariableOp"                                                                                                                                   │ │
│ │                  │     input: "xlacallmodule_readvariableop_resource"                                                                                                         │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "dtype"                                                                                                                                         │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_HALF                                                                                                                                      │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 8                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 8                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "XlaCallModule/ReadVariableOp_1"                                                                                                                 │ │
│ │                  │     op: "ReadVariableOp"                                                                                                                                   │ │
│ │                  │     input: "xlacallmodule_readvariableop_1_resource"                                                                                                       │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "dtype"                                                                                                                                         │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_HALF                                                                                                                                      │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "XlaCallModule"                                                                                                                                  │ │
│ │                  │     op: "XlaCallModule"                                                                                                                                    │ │
│ │                  │     input: "XlaCallModule/ReadVariableOp:value:0"                                                                                                          │ │
│ │                  │     input: "XlaCallModule/ReadVariableOp_1:value:0"                                                                                                        │ │
│ │                  │     input: "args_0"                                                                                                                                        │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "version"                                                                                                                                       │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     i: 5                                                                                                                                               │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "module"                                                                                                                                        │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     s:                                                                                                                                                 │ │
│ │                  "ML\357R\rStableHLO_v1.13.0\000\001#\005\001\005\023\001\003\013\003\021\017\023\027\033\037#\'+\003\257o1\001#\007\017\013\013\017\013\023\013\017\017\013… │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 1                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "Tout"                                                                                                                                          │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   type: DT_HALF                                                                                                                                    │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "Tin"                                                                                                                                           │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   type: DT_HALF                                                                                                                                    │ │
│ │                  │   │   │   type: DT_HALF                                                                                                                                    │ │
│ │                  │   │   │   type: DT_HALF                                                                                                                                    │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "Sout"                                                                                                                                          │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 1                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "Identity"                                                                                                                                       │ │
│ │                  │     op: "Identity"                                                                                                                                         │ │
│ │                  │     input: "XlaCallModule:output:0"                                                                                                                        │ │
│ │                  │     input: "^NoOp"                                                                                                                                         │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 1                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "T"                                                                                                                                             │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_HALF                                                                                                                                      │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "NoOp"                                                                                                                                           │ │
│ │                  │     op: "NoOp"                                                                                                                                             │ │
│ │                  │     input: "^XlaCallModule"                                                                                                                                │ │
│ │                  │     input: "^XlaCallModule/ReadVariableOp"                                                                                                                 │ │
│ │                  │     input: "^XlaCallModule/ReadVariableOp_1"                                                                                                               │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   ret {                                                                                                                                                    │ │
│ │                  │     key: "identity"                                                                                                                                        │ │
│ │                  │     value: "Identity:output:0"                                                                                                                             │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   attr {                                                                                                                                                   │ │
│ │                  │     key: "_input_shapes"                                                                                                                                   │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   list {                                                                                                                                               │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 1                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 3                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 8                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 8                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   attr {                                                                                                                                                   │ │
│ │                  │     key: "_construction_context"                                                                                                                           │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   s: "kEagerRuntime"                                                                                                                                   │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   control_ret {                                                                                                                                            │ │
│ │                  │     key: "XlaCallModule"                                                                                                                                   │ │
│ │                  │     value: "XlaCallModule"                                                                                                                                 │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   control_ret {                                                                                                                                            │ │
│ │                  │     key: "XlaCallModule/ReadVariableOp"                                                                                                                    │ │
│ │                  │     value: "XlaCallModule/ReadVariableOp"                                                                                                                  │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   control_ret {                                                                                                                                            │ │
│ │                  │     key: "XlaCallModule/ReadVariableOp_1"                                                                                                                  │ │
│ │                  │     value: "XlaCallModule/ReadVariableOp_1"                                                                                                                │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   arg_attr {                                                                                                                                               │ │
│ │                  │     key: 0                                                                                                                                                 │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   attr {                                                                                                                                               │ │
│ │                  │   │     key: "_user_specified_name"                                                                                                                        │ │
│ │                  │   │     value {                                                                                                                                            │ │
│ │                  │   │   │   s: "args_0"                                                                                                                                      │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │   │   attr {                                                                                                                                               │ │
│ │                  │   │     key: "_output_shapes"                                                                                                                              │ │
│ │                  │   │     value {                                                                                                                                            │ │
│ │                  │   │   │   list {                                                                                                                                           │ │
│ │                  │   │   │     shape {                                                                                                                                        │ │
│ │                  │   │   │   │   dim {                                                                                                                                        │ │
│ │                  │   │   │   │     size: 1                                                                                                                                    │ │
│ │                  │   │   │   │   }                                                                                                                                            │ │
│ │                  │   │   │   │   dim {                                                                                                                                        │ │
│ │                  │   │   │   │     size: 3                                                                                                                                    │ │
│ │                  │   │   │   │   }                                                                                                                                            │ │
│ │                  │   │   │   │   dim {                                                                                                                                        │ │
│ │                  │   │   │   │     size: 8                                                                                                                                    │ │
│ │                  │   │   │   │   }                                                                                                                                            │ │
│ │                  │   │   │   │   dim {                                                                                                                                        │ │
│ │                  │   │   │   │     size: 8                                                                                                                                    │ │
│ │                  │   │   │   │   }                                                                                                                                            │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   arg_attr {                                                                                                                                               │ │
│ │                  │     key: 1                                                                                                                                                 │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   attr {                                                                                                                                               │ │
│ │                  │   │     key: "_user_specified_name"                                                                                                                        │ │
│ │                  │   │     value {                                                                                                                                            │ │
│ │                  │   │   │   s: "resource"                                                                                                                                    │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   arg_attr {                                                                                                                                               │ │
│ │                  │     key: 2                                                                                                                                                 │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   attr {                                                                                                                                               │ │
│ │                  │   │     key: "_user_specified_name"                                                                                                                        │ │
│ │                  │   │     value {                                                                                                                                            │ │
│ │                  │   │   │   s: "resource"                                                                                                                                    │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    function {                                                                                                                                                 │ │
│ │                  │   signature {                                                                                                                                              │ │
│ │                  │     name: "__inference__traced_restore_83"                                                                                                                 │ │
│ │                  │     input_arg {                                                                                                                                            │ │
│ │                  │   │   name: "file_prefix"                                                                                                                                  │ │
│ │                  │   │   type: DT_STRING                                                                                                                                      │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     input_arg {                                                                                                                                            │ │
│ │                  │   │   name: "assignvariableop_variable_1"                                                                                                                  │ │
│ │                  │   │   type: DT_RESOURCE                                                                                                                                    │ │
│ │                  │   │   handle_data {                                                                                                                                        │ │
│ │                  │   │     dtype: DT_HALF                                                                                                                                     │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 3                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 3                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 8                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 8                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     input_arg {                                                                                                                                            │ │
│ │                  │   │   name: "assignvariableop_1_variable"                                                                                                                  │ │
│ │                  │   │   type: DT_RESOURCE                                                                                                                                    │ │
│ │                  │   │   handle_data {                                                                                                                                        │ │
│ │                  │   │     dtype: DT_HALF                                                                                                                                     │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 3                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     output_arg {                                                                                                                                           │ │
│ │                  │   │   name: "identity_3"                                                                                                                                   │ │
│ │                  │   │   type: DT_STRING                                                                                                                                      │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     is_stateful: true                                                                                                                                      │ │
│ │                  │     control_output: "AssignVariableOp"                                                                                                                     │ │
│ │                  │     control_output: "AssignVariableOp_1"                                                                                                                   │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "RestoreV2/tensor_names"                                                                                                                         │ │
│ │                  │     op: "Const"                                                                                                                                            │ │
│ │                  │     device: "/device:CPU:0"                                                                                                                                │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "value"                                                                                                                                         │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     tensor {                                                                                                                                           │ │
│ │                  │   │   │   dtype: DT_STRING                                                                                                                                 │ │
│ │                  │   │   │   tensor_shape {                                                                                                                                   │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   string_val: "_variables/0/.ATTRIBUTES/VARIABLE_VALUE"                                                                                            │ │
│ │                  │   │   │   string_val: "_variables/1/.ATTRIBUTES/VARIABLE_VALUE"                                                                                            │ │
│ │                  │   │   │   string_val: "_CHECKPOINTABLE_OBJECT_GRAPH"                                                                                                       │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "dtype"                                                                                                                                         │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_STRING                                                                                                                                    │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "RestoreV2/shape_and_slices"                                                                                                                     │ │
│ │                  │     op: "Const"                                                                                                                                            │ │
│ │                  │     device: "/device:CPU:0"                                                                                                                                │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "value"                                                                                                                                         │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     tensor {                                                                                                                                           │ │
│ │                  │   │   │   dtype: DT_STRING                                                                                                                                 │ │
│ │                  │   │   │   tensor_shape {                                                                                                                                   │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   string_val: ""                                                                                                                                   │ │
│ │                  │   │   │   string_val: ""                                                                                                                                   │ │
│ │                  │   │   │   string_val: ""                                                                                                                                   │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "dtype"                                                                                                                                         │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_STRING                                                                                                                                    │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "RestoreV2"                                                                                                                                      │ │
│ │                  │     op: "RestoreV2"                                                                                                                                        │ │
│ │                  │     input: "file_prefix"                                                                                                                                   │ │
│ │                  │     input: "RestoreV2/tensor_names:output:0"                                                                                                               │ │
│ │                  │     input: "RestoreV2/shape_and_slices:output:0"                                                                                                           │ │
│ │                  │     device: "/device:CPU:0"                                                                                                                                │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "dtypes"                                                                                                                                        │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   type: DT_HALF                                                                                                                                    │ │
│ │                  │   │   │   type: DT_HALF                                                                                                                                    │ │
│ │                  │   │   │   type: DT_STRING                                                                                                                                  │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     unknown_rank: true                                                                                                                             │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     unknown_rank: true                                                                                                                             │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     unknown_rank: true                                                                                                                             │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "Identity"                                                                                                                                       │ │
│ │                  │     op: "Identity"                                                                                                                                         │ │
│ │                  │     input: "RestoreV2:tensors:0"                                                                                                                           │ │
│ │                  │     device: "/device:CPU:0"                                                                                                                                │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     unknown_rank: true                                                                                                                             │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "T"                                                                                                                                             │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_HALF                                                                                                                                      │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "AssignVariableOp"                                                                                                                               │ │
│ │                  │     op: "AssignVariableOp"                                                                                                                                 │ │
│ │                  │     input: "assignvariableop_variable_1"                                                                                                                   │ │
│ │                  │     input: "Identity:output:0"                                                                                                                             │ │
│ │                  │     device: "/device:CPU:0"                                                                                                                                │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "dtype"                                                                                                                                         │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_HALF                                                                                                                                      │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_has_manual_control_dependencies"                                                                                                              │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     b: true                                                                                                                                            │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "Identity_1"                                                                                                                                     │ │
│ │                  │     op: "Identity"                                                                                                                                         │ │
│ │                  │     input: "RestoreV2:tensors:1"                                                                                                                           │ │
│ │                  │     device: "/device:CPU:0"                                                                                                                                │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     unknown_rank: true                                                                                                                             │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "T"                                                                                                                                             │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_HALF                                                                                                                                      │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "AssignVariableOp_1"                                                                                                                             │ │
│ │                  │     op: "AssignVariableOp"                                                                                                                                 │ │
│ │                  │     input: "assignvariableop_1_variable"                                                                                                                   │ │
│ │                  │     input: "Identity_1:output:0"                                                                                                                           │ │
│ │                  │     device: "/device:CPU:0"                                                                                                                                │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "dtype"                                                                                                                                         │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_HALF                                                                                                                                      │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_has_manual_control_dependencies"                                                                                                              │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     b: true                                                                                                                                            │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "NoOp"                                                                                                                                           │ │
│ │                  │     op: "NoOp"                                                                                                                                             │ │
│ │                  │     device: "/device:CPU:0"                                                                                                                                │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_has_manual_control_dependencies"                                                                                                              │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     b: true                                                                                                                                            │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "Identity_2"                                                                                                                                     │ │
│ │                  │     op: "Identity"                                                                                                                                         │ │
│ │                  │     input: "file_prefix"                                                                                                                                   │ │
│ │                  │     input: "^AssignVariableOp"                                                                                                                             │ │
│ │                  │     input: "^AssignVariableOp_1"                                                                                                                           │ │
│ │                  │     input: "^NoOp"                                                                                                                                         │ │
│ │                  │     device: "/device:CPU:0"                                                                                                                                │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "T"                                                                                                                                             │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_STRING                                                                                                                                    │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "Identity_3"                                                                                                                                     │ │
│ │                  │     op: "Identity"                                                                                                                                         │ │
│ │                  │     input: "Identity_2:output:0"                                                                                                                           │ │
│ │                  │     input: "^NoOp_1"                                                                                                                                       │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "T"                                                                                                                                             │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_STRING                                                                                                                                    │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "NoOp_1"                                                                                                                                         │ │
│ │                  │     op: "NoOp"                                                                                                                                             │ │
│ │                  │     input: "^AssignVariableOp"                                                                                                                             │ │
│ │                  │     input: "^AssignVariableOp_1"                                                                                                                           │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   ret {                                                                                                                                                    │ │
│ │                  │     key: "identity_3"                                                                                                                                      │ │
│ │                  │     value: "Identity_3:output:0"                                                                                                                           │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   attr {                                                                                                                                                   │ │
│ │                  │     key: "_input_shapes"                                                                                                                                   │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   list {                                                                                                                                               │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   attr {                                                                                                                                                   │ │
│ │                  │     key: "_construction_context"                                                                                                                           │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   s: "kEagerRuntime"                                                                                                                                   │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   control_ret {                                                                                                                                            │ │
│ │                  │     key: "AssignVariableOp"                                                                                                                                │ │
│ │                  │     value: "AssignVariableOp"                                                                                                                              │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   control_ret {                                                                                                                                            │ │
│ │                  │     key: "AssignVariableOp_1"                                                                                                                              │ │
│ │                  │     value: "AssignVariableOp_1"                                                                                                                            │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   arg_attr {                                                                                                                                               │ │
│ │                  │     key: 0                                                                                                                                                 │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   attr {                                                                                                                                               │ │
│ │                  │   │     key: "_user_specified_name"                                                                                                                        │ │
│ │                  │   │     value {                                                                                                                                            │ │
│ │                  │   │   │   s: "file_prefix"                                                                                                                                 │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │   │   attr {                                                                                                                                               │ │
│ │                  │   │     key: "_output_shapes"                                                                                                                              │ │
│ │                  │   │     value {                                                                                                                                            │ │
│ │                  │   │   │   list {                                                                                                                                           │ │
│ │                  │   │   │     shape {                                                                                                                                        │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   arg_attr {                                                                                                                                               │ │
│ │                  │     key: 1                                                                                                                                                 │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   attr {                                                                                                                                               │ │
│ │                  │   │     key: "_user_specified_name"                                                                                                                        │ │
│ │                  │   │     value {                                                                                                                                            │ │
│ │                  │   │   │   s: "Variable_1"                                                                                                                                  │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   arg_attr {                                                                                                                                               │ │
│ │                  │     key: 2                                                                                                                                                 │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   attr {                                                                                                                                               │ │
│ │                  │   │     key: "_user_specified_name"                                                                                                                        │ │
│ │                  │   │     value {                                                                                                                                            │ │
│ │                  │   │   │   s: "Variable"                                                                                                                                    │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    function {                                                                                                                                                 │ │
│ │                  │   signature {                                                                                                                                              │ │
│ │                  │     name: "__inference__traced_save_68"                                                                                                                    │ │
│ │                  │     input_arg {                                                                                                                                            │ │
│ │                  │   │   name: "file_prefix"                                                                                                                                  │ │
│ │                  │   │   type: DT_STRING                                                                                                                                      │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     input_arg {                                                                                                                                            │ │
│ │                  │   │   name: "read_disablecopyonread_variable_1"                                                                                                            │ │
│ │                  │   │   type: DT_RESOURCE                                                                                                                                    │ │
│ │                  │   │   handle_data {                                                                                                                                        │ │
│ │                  │   │     dtype: DT_HALF                                                                                                                                     │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 3                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 3                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 8                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 8                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     input_arg {                                                                                                                                            │ │
│ │                  │   │   name: "read_1_disablecopyonread_variable"                                                                                                            │ │
│ │                  │   │   type: DT_RESOURCE                                                                                                                                    │ │
│ │                  │   │   handle_data {                                                                                                                                        │ │
│ │                  │   │     dtype: DT_HALF                                                                                                                                     │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 3                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     input_arg {                                                                                                                                            │ │
│ │                  │   │   name: "savev2_const"                                                                                                                                 │ │
│ │                  │   │   type: DT_STRING                                                                                                                                      │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     output_arg {                                                                                                                                           │ │
│ │                  │   │   name: "identity_5"                                                                                                                                   │ │
│ │                  │   │   type: DT_STRING                                                                                                                                      │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     is_stateful: true                                                                                                                                      │ │
│ │                  │     control_output: "MergeV2Checkpoints"                                                                                                                   │ │
│ │                  │     control_output: "Read/DisableCopyOnRead"                                                                                                               │ │
│ │                  │     control_output: "Read/ReadVariableOp"                                                                                                                  │ │
│ │                  │     control_output: "Read_1/DisableCopyOnRead"                                                                                                             │ │
│ │                  │     control_output: "Read_1/ReadVariableOp"                                                                                                                │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "StaticRegexFullMatch"                                                                                                                           │ │
│ │                  │     op: "StaticRegexFullMatch"                                                                                                                             │ │
│ │                  │     input: "file_prefix"                                                                                                                                   │ │
│ │                  │     device: "/device:CPU:*"                                                                                                                                │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "pattern"                                                                                                                                       │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     s: "^s3://.*"                                                                                                                                      │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "Const"                                                                                                                                          │ │
│ │                  │     op: "Const"                                                                                                                                            │ │
│ │                  │     device: "/device:CPU:*"                                                                                                                                │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "value"                                                                                                                                         │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     tensor {                                                                                                                                           │ │
│ │                  │   │   │   dtype: DT_STRING                                                                                                                                 │ │
│ │                  │   │   │   tensor_shape {                                                                                                                                   │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   string_val: ".part"                                                                                                                              │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "dtype"                                                                                                                                         │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_STRING                                                                                                                                    │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "Const_1"                                                                                                                                        │ │
│ │                  │     op: "Const"                                                                                                                                            │ │
│ │                  │     device: "/device:CPU:*"                                                                                                                                │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "value"                                                                                                                                         │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     tensor {                                                                                                                                           │ │
│ │                  │   │   │   dtype: DT_STRING                                                                                                                                 │ │
│ │                  │   │   │   tensor_shape {                                                                                                                                   │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   string_val: "_temp/part"                                                                                                                         │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "dtype"                                                                                                                                         │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_STRING                                                                                                                                    │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "Select"                                                                                                                                         │ │
│ │                  │     op: "Select"                                                                                                                                           │ │
│ │                  │     input: "StaticRegexFullMatch:output:0"                                                                                                                 │ │
│ │                  │     input: "Const:output:0"                                                                                                                                │ │
│ │                  │     input: "Const_1:output:0"                                                                                                                              │ │
│ │                  │     device: "/device:CPU:*"                                                                                                                                │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "T"                                                                                                                                             │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_STRING                                                                                                                                    │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "StringJoin"                                                                                                                                     │ │
│ │                  │     op: "StringJoin"                                                                                                                                       │ │
│ │                  │     input: "file_prefix"                                                                                                                                   │ │
│ │                  │     input: "Select:output:0"                                                                                                                               │ │
│ │                  │     device: "/device:CPU:*"                                                                                                                                │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "N"                                                                                                                                             │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     i: 2                                                                                                                                               │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "Read/DisableCopyOnRead"                                                                                                                         │ │
│ │                  │     op: "DisableCopyOnRead"                                                                                                                                │ │
│ │                  │     input: "read_disablecopyonread_variable_1"                                                                                                             │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "Read/ReadVariableOp"                                                                                                                            │ │
│ │                  │     op: "ReadVariableOp"                                                                                                                                   │ │
│ │                  │     input: "read_disablecopyonread_variable_1"                                                                                                             │ │
│ │                  │     input: "^Read/DisableCopyOnRead"                                                                                                                       │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "dtype"                                                                                                                                         │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_HALF                                                                                                                                      │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 8                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 8                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "Identity"                                                                                                                                       │ │
│ │                  │     op: "Identity"                                                                                                                                         │ │
│ │                  │     input: "Read/ReadVariableOp:value:0"                                                                                                                   │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 8                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 8                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "T"                                                                                                                                             │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_HALF                                                                                                                                      │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "Identity_1"                                                                                                                                     │ │
│ │                  │     op: "Identity"                                                                                                                                         │ │
│ │                  │     input: "Identity:output:0"                                                                                                                             │ │
│ │                  │     device: "/device:CPU:0"                                                                                                                                │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 8                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 8                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "T"                                                                                                                                             │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_HALF                                                                                                                                      │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "Read_1/DisableCopyOnRead"                                                                                                                       │ │
│ │                  │     op: "DisableCopyOnRead"                                                                                                                                │ │
│ │                  │     input: "read_1_disablecopyonread_variable"                                                                                                             │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "Read_1/ReadVariableOp"                                                                                                                          │ │
│ │                  │     op: "ReadVariableOp"                                                                                                                                   │ │
│ │                  │     input: "read_1_disablecopyonread_variable"                                                                                                             │ │
│ │                  │     input: "^Read_1/DisableCopyOnRead"                                                                                                                     │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "dtype"                                                                                                                                         │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_HALF                                                                                                                                      │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "Identity_2"                                                                                                                                     │ │
│ │                  │     op: "Identity"                                                                                                                                         │ │
│ │                  │     input: "Read_1/ReadVariableOp:value:0"                                                                                                                 │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "T"                                                                                                                                             │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_HALF                                                                                                                                      │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "Identity_3"                                                                                                                                     │ │
│ │                  │     op: "Identity"                                                                                                                                         │ │
│ │                  │     input: "Identity_2:output:0"                                                                                                                           │ │
│ │                  │     device: "/device:CPU:0"                                                                                                                                │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "T"                                                                                                                                             │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_HALF                                                                                                                                      │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "num_shards"                                                                                                                                     │ │
│ │                  │     op: "Const"                                                                                                                                            │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "value"                                                                                                                                         │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     tensor {                                                                                                                                           │ │
│ │                  │   │   │   dtype: DT_INT32                                                                                                                                  │ │
│ │                  │   │   │   tensor_shape {                                                                                                                                   │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   int_val: 1                                                                                                                                       │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "dtype"                                                                                                                                         │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_INT32                                                                                                                                     │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "ShardedFilename/shard"                                                                                                                          │ │
│ │                  │     op: "Const"                                                                                                                                            │ │
│ │                  │     device: "/device:CPU:0"                                                                                                                                │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "value"                                                                                                                                         │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     tensor {                                                                                                                                           │ │
│ │                  │   │   │   dtype: DT_INT32                                                                                                                                  │ │
│ │                  │   │   │   tensor_shape {                                                                                                                                   │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   int_val: 0                                                                                                                                       │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "dtype"                                                                                                                                         │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_INT32                                                                                                                                     │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "ShardedFilename"                                                                                                                                │ │
│ │                  │     op: "ShardedFilename"                                                                                                                                  │ │
│ │                  │     input: "StringJoin:output:0"                                                                                                                           │ │
│ │                  │     input: "ShardedFilename/shard:output:0"                                                                                                                │ │
│ │                  │     input: "num_shards:output:0"                                                                                                                           │ │
│ │                  │     device: "/device:CPU:0"                                                                                                                                │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "SaveV2/tensor_names"                                                                                                                            │ │
│ │                  │     op: "Const"                                                                                                                                            │ │
│ │                  │     device: "/device:CPU:0"                                                                                                                                │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "value"                                                                                                                                         │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     tensor {                                                                                                                                           │ │
│ │                  │   │   │   dtype: DT_STRING                                                                                                                                 │ │
│ │                  │   │   │   tensor_shape {                                                                                                                                   │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   string_val: "_variables/0/.ATTRIBUTES/VARIABLE_VALUE"                                                                                            │ │
│ │                  │   │   │   string_val: "_variables/1/.ATTRIBUTES/VARIABLE_VALUE"                                                                                            │ │
│ │                  │   │   │   string_val: "_CHECKPOINTABLE_OBJECT_GRAPH"                                                                                                       │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "dtype"                                                                                                                                         │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_STRING                                                                                                                                    │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "SaveV2/shape_and_slices"                                                                                                                        │ │
│ │                  │     op: "Const"                                                                                                                                            │ │
│ │                  │     device: "/device:CPU:0"                                                                                                                                │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "value"                                                                                                                                         │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     tensor {                                                                                                                                           │ │
│ │                  │   │   │   dtype: DT_STRING                                                                                                                                 │ │
│ │                  │   │   │   tensor_shape {                                                                                                                                   │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   string_val: ""                                                                                                                                   │ │
│ │                  │   │   │   string_val: ""                                                                                                                                   │ │
│ │                  │   │   │   string_val: ""                                                                                                                                   │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "dtype"                                                                                                                                         │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_STRING                                                                                                                                    │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "SaveV2"                                                                                                                                         │ │
│ │                  │     op: "SaveV2"                                                                                                                                           │ │
│ │                  │     input: "ShardedFilename:filename:0"                                                                                                                    │ │
│ │                  │     input: "SaveV2/tensor_names:output:0"                                                                                                                  │ │
│ │                  │     input: "SaveV2/shape_and_slices:output:0"                                                                                                              │ │
│ │                  │     input: "Identity_1:output:0"                                                                                                                           │ │
│ │                  │     input: "Identity_3:output:0"                                                                                                                           │ │
│ │                  │     input: "savev2_const"                                                                                                                                  │ │
│ │                  │     device: "/device:CPU:0"                                                                                                                                │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "dtypes"                                                                                                                                        │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   type: DT_HALF                                                                                                                                    │ │
│ │                  │   │   │   type: DT_HALF                                                                                                                                    │ │
│ │                  │   │   │   type: DT_STRING                                                                                                                                  │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_has_manual_control_dependencies"                                                                                                              │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     b: true                                                                                                                                            │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "MergeV2Checkpoints/checkpoint_prefixes"                                                                                                         │ │
│ │                  │     op: "Pack"                                                                                                                                             │ │
│ │                  │     input: "ShardedFilename:filename:0"                                                                                                                    │ │
│ │                  │     input: "^SaveV2"                                                                                                                                       │ │
│ │                  │     device: "/device:CPU:0"                                                                                                                                │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 1                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "T"                                                                                                                                             │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_STRING                                                                                                                                    │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "N"                                                                                                                                             │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     i: 1                                                                                                                                               │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "MergeV2Checkpoints"                                                                                                                             │ │
│ │                  │     op: "MergeV2Checkpoints"                                                                                                                               │ │
│ │                  │     input: "MergeV2Checkpoints/checkpoint_prefixes:output:0"                                                                                               │ │
│ │                  │     input: "file_prefix"                                                                                                                                   │ │
│ │                  │     device: "/device:CPU:0"                                                                                                                                │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_has_manual_control_dependencies"                                                                                                              │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     b: true                                                                                                                                            │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "Identity_4"                                                                                                                                     │ │
│ │                  │     op: "Identity"                                                                                                                                         │ │
│ │                  │     input: "file_prefix"                                                                                                                                   │ │
│ │                  │     input: "^MergeV2Checkpoints"                                                                                                                           │ │
│ │                  │     device: "/device:CPU:0"                                                                                                                                │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "T"                                                                                                                                             │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_STRING                                                                                                                                    │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "Identity_5"                                                                                                                                     │ │
│ │                  │     op: "Identity"                                                                                                                                         │ │
│ │                  │     input: "Identity_4:output:0"                                                                                                                           │ │
│ │                  │     input: "^NoOp"                                                                                                                                         │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "T"                                                                                                                                             │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_STRING                                                                                                                                    │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "NoOp"                                                                                                                                           │ │
│ │                  │     op: "NoOp"                                                                                                                                             │ │
│ │                  │     input: "^MergeV2Checkpoints"                                                                                                                           │ │
│ │                  │     input: "^Read/DisableCopyOnRead"                                                                                                                       │ │
│ │                  │     input: "^Read/ReadVariableOp"                                                                                                                          │ │
│ │                  │     input: "^Read_1/DisableCopyOnRead"                                                                                                                     │ │
│ │                  │     input: "^Read_1/ReadVariableOp"                                                                                                                        │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   ret {                                                                                                                                                    │ │
│ │                  │     key: "identity_5"                                                                                                                                      │ │
│ │                  │     value: "Identity_5:output:0"                                                                                                                           │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   attr {                                                                                                                                                   │ │
│ │                  │     key: "_input_shapes"                                                                                                                                   │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   list {                                                                                                                                               │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   attr {                                                                                                                                                   │ │
│ │                  │     key: "_construction_context"                                                                                                                           │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   s: "kEagerRuntime"                                                                                                                                   │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   control_ret {                                                                                                                                            │ │
│ │                  │     key: "Read_1/ReadVariableOp"                                                                                                                           │ │
│ │                  │     value: "Read_1/ReadVariableOp"                                                                                                                         │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   control_ret {                                                                                                                                            │ │
│ │                  │     key: "Read_1/DisableCopyOnRead"                                                                                                                        │ │
│ │                  │     value: "Read_1/DisableCopyOnRead"                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   control_ret {                                                                                                                                            │ │
│ │                  │     key: "Read/ReadVariableOp"                                                                                                                             │ │
│ │                  │     value: "Read/ReadVariableOp"                                                                                                                           │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   control_ret {                                                                                                                                            │ │
│ │                  │     key: "Read/DisableCopyOnRead"                                                                                                                          │ │
│ │                  │     value: "Read/DisableCopyOnRead"                                                                                                                        │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   control_ret {                                                                                                                                            │ │
│ │                  │     key: "MergeV2Checkpoints"                                                                                                                              │ │
│ │                  │     value: "MergeV2Checkpoints"                                                                                                                            │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   arg_attr {                                                                                                                                               │ │
│ │                  │     key: 0                                                                                                                                                 │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   attr {                                                                                                                                               │ │
│ │                  │   │     key: "_user_specified_name"                                                                                                                        │ │
│ │                  │   │     value {                                                                                                                                            │ │
│ │                  │   │   │   s: "file_prefix"                                                                                                                                 │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │   │   attr {                                                                                                                                               │ │
│ │                  │   │     key: "_output_shapes"                                                                                                                              │ │
│ │                  │   │     value {                                                                                                                                            │ │
│ │                  │   │   │   list {                                                                                                                                           │ │
│ │                  │   │   │     shape {                                                                                                                                        │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   arg_attr {                                                                                                                                               │ │
│ │                  │     key: 1                                                                                                                                                 │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   attr {                                                                                                                                               │ │
│ │                  │   │     key: "_user_specified_name"                                                                                                                        │ │
│ │                  │   │     value {                                                                                                                                            │ │
│ │                  │   │   │   s: "Variable_1"                                                                                                                                  │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   arg_attr {                                                                                                                                               │ │
│ │                  │     key: 2                                                                                                                                                 │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   attr {                                                                                                                                               │ │
│ │                  │   │     key: "_user_specified_name"                                                                                                                        │ │
│ │                  │   │     value {                                                                                                                                            │ │
│ │                  │   │   │   s: "Variable"                                                                                                                                    │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   arg_attr {                                                                                                                                               │ │
│ │                  │     key: 3                                                                                                                                                 │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   attr {                                                                                                                                               │ │
│ │                  │   │     key: "_user_specified_name"                                                                                                                        │ │
│ │                  │   │     value {                                                                                                                                            │ │
│ │                  │   │   │   s: "Const"                                                                                                                                       │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │   │   attr {                                                                                                                                               │ │
│ │                  │   │     key: "_output_shapes"                                                                                                                              │ │
│ │                  │   │     value {                                                                                                                                            │ │
│ │                  │   │   │   list {                                                                                                                                           │ │
│ │                  │   │   │     shape {                                                                                                                                        │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    function {                                                                                                                                                 │ │
│ │                  │   signature {                                                                                                                                              │ │
│ │                  │     name: "__inference_inner_34"                                                                                                                           │ │
│ │                  │     input_arg {                                                                                                                                            │ │
│ │                  │   │   name: "args_0"                                                                                                                                       │ │
│ │                  │   │   type: DT_HALF                                                                                                                                        │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     input_arg {                                                                                                                                            │ │
│ │                  │   │   name: "xlacallmodule_readvariableop_resource"                                                                                                        │ │
│ │                  │   │   type: DT_RESOURCE                                                                                                                                    │ │
│ │                  │   │   handle_data {                                                                                                                                        │ │
│ │                  │   │     dtype: DT_HALF                                                                                                                                     │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 3                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 3                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 8                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 8                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     input_arg {                                                                                                                                            │ │
│ │                  │   │   name: "xlacallmodule_readvariableop_1_resource"                                                                                                      │ │
│ │                  │   │   type: DT_RESOURCE                                                                                                                                    │ │
│ │                  │   │   handle_data {                                                                                                                                        │ │
│ │                  │   │     dtype: DT_HALF                                                                                                                                     │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 3                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     output_arg {                                                                                                                                           │ │
│ │                  │   │   name: "identity"                                                                                                                                     │ │
│ │                  │   │   type: DT_HALF                                                                                                                                        │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     is_stateful: true                                                                                                                                      │ │
│ │                  │     control_output: "XlaCallModule"                                                                                                                        │ │
│ │                  │     control_output: "XlaCallModule/ReadVariableOp"                                                                                                         │ │
│ │                  │     control_output: "XlaCallModule/ReadVariableOp_1"                                                                                                       │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "XlaCallModule/ReadVariableOp"                                                                                                                   │ │
│ │                  │     op: "ReadVariableOp"                                                                                                                                   │ │
│ │                  │     input: "xlacallmodule_readvariableop_resource"                                                                                                         │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "dtype"                                                                                                                                         │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_HALF                                                                                                                                      │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 8                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 8                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "XlaCallModule/ReadVariableOp_1"                                                                                                                 │ │
│ │                  │     op: "ReadVariableOp"                                                                                                                                   │ │
│ │                  │     input: "xlacallmodule_readvariableop_1_resource"                                                                                                       │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "dtype"                                                                                                                                         │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_HALF                                                                                                                                      │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "XlaCallModule"                                                                                                                                  │ │
│ │                  │     op: "XlaCallModule"                                                                                                                                    │ │
│ │                  │     input: "XlaCallModule/ReadVariableOp:value:0"                                                                                                          │ │
│ │                  │     input: "XlaCallModule/ReadVariableOp_1:value:0"                                                                                                        │ │
│ │                  │     input: "args_0"                                                                                                                                        │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "version"                                                                                                                                       │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     i: 5                                                                                                                                               │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "module"                                                                                                                                        │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     s:                                                                                                                                                 │ │
│ │                  "ML\357R\rStableHLO_v1.13.0\000\001#\005\001\005\023\001\003\013\003\021\017\023\027\033\037#\'+\003\257o1\001#\007\017\013\013\017\013\023\013\017\017\013… │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 1                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "Tout"                                                                                                                                          │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   type: DT_HALF                                                                                                                                    │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "Tin"                                                                                                                                           │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   type: DT_HALF                                                                                                                                    │ │
│ │                  │   │   │   type: DT_HALF                                                                                                                                    │ │
│ │                  │   │   │   type: DT_HALF                                                                                                                                    │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "Sout"                                                                                                                                          │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 1                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "Identity"                                                                                                                                       │ │
│ │                  │     op: "Identity"                                                                                                                                         │ │
│ │                  │     input: "XlaCallModule:output:0"                                                                                                                        │ │
│ │                  │     input: "^NoOp"                                                                                                                                         │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │   │   shape {                                                                                                                                          │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 1                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │     dim {                                                                                                                                          │ │
│ │                  │   │   │   │   size: 3                                                                                                                                      │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "T"                                                                                                                                             │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     type: DT_HALF                                                                                                                                      │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   node_def {                                                                                                                                               │ │
│ │                  │     name: "NoOp"                                                                                                                                           │ │
│ │                  │     op: "NoOp"                                                                                                                                             │ │
│ │                  │     input: "^XlaCallModule"                                                                                                                                │ │
│ │                  │     input: "^XlaCallModule/ReadVariableOp"                                                                                                                 │ │
│ │                  │     input: "^XlaCallModule/ReadVariableOp_1"                                                                                                               │ │
│ │                  │     attr {                                                                                                                                                 │ │
│ │                  │   │   key: "_output_shapes"                                                                                                                                │ │
│ │                  │   │   value {                                                                                                                                              │ │
│ │                  │   │     list {                                                                                                                                             │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   ret {                                                                                                                                                    │ │
│ │                  │     key: "identity"                                                                                                                                        │ │
│ │                  │     value: "Identity:output:0"                                                                                                                             │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   attr {                                                                                                                                                   │ │
│ │                  │     key: "_input_shapes"                                                                                                                                   │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   list {                                                                                                                                               │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 1                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 3                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 8                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │   │   dim {                                                                                                                                            │ │
│ │                  │   │   │     size: 8                                                                                                                                        │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │     shape {                                                                                                                                            │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   attr {                                                                                                                                                   │ │
│ │                  │     key: "_construction_context"                                                                                                                           │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   s: "kEagerRuntime"                                                                                                                                   │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   control_ret {                                                                                                                                            │ │
│ │                  │     key: "XlaCallModule"                                                                                                                                   │ │
│ │                  │     value: "XlaCallModule"                                                                                                                                 │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   control_ret {                                                                                                                                            │ │
│ │                  │     key: "XlaCallModule/ReadVariableOp"                                                                                                                    │ │
│ │                  │     value: "XlaCallModule/ReadVariableOp"                                                                                                                  │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   control_ret {                                                                                                                                            │ │
│ │                  │     key: "XlaCallModule/ReadVariableOp_1"                                                                                                                  │ │
│ │                  │     value: "XlaCallModule/ReadVariableOp_1"                                                                                                                │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   arg_attr {                                                                                                                                               │ │
│ │                  │     key: 0                                                                                                                                                 │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   attr {                                                                                                                                               │ │
│ │                  │   │     key: "_user_specified_name"                                                                                                                        │ │
│ │                  │   │     value {                                                                                                                                            │ │
│ │                  │   │   │   s: "args_0"                                                                                                                                      │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │   │   attr {                                                                                                                                               │ │
│ │                  │   │     key: "_output_shapes"                                                                                                                              │ │
│ │                  │   │     value {                                                                                                                                            │ │
│ │                  │   │   │   list {                                                                                                                                           │ │
│ │                  │   │   │     shape {                                                                                                                                        │ │
│ │                  │   │   │   │   dim {                                                                                                                                        │ │
│ │                  │   │   │   │     size: 1                                                                                                                                    │ │
│ │                  │   │   │   │   }                                                                                                                                            │ │
│ │                  │   │   │   │   dim {                                                                                                                                        │ │
│ │                  │   │   │   │     size: 3                                                                                                                                    │ │
│ │                  │   │   │   │   }                                                                                                                                            │ │
│ │                  │   │   │   │   dim {                                                                                                                                        │ │
│ │                  │   │   │   │     size: 8                                                                                                                                    │ │
│ │                  │   │   │   │   }                                                                                                                                            │ │
│ │                  │   │   │   │   dim {                                                                                                                                        │ │
│ │                  │   │   │   │     size: 8                                                                                                                                    │ │
│ │                  │   │   │   │   }                                                                                                                                            │ │
│ │                  │   │   │     }                                                                                                                                              │ │
│ │                  │   │   │   }                                                                                                                                                │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   arg_attr {                                                                                                                                               │ │
│ │                  │     key: 1                                                                                                                                                 │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   attr {                                                                                                                                               │ │
│ │                  │   │     key: "_user_specified_name"                                                                                                                        │ │
│ │                  │   │     value {                                                                                                                                            │ │
│ │                  │   │   │   s: "resource"                                                                                                                                    │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                  │   arg_attr {                                                                                                                                               │ │
│ │                  │     key: 2                                                                                                                                                 │ │
│ │                  │     value {                                                                                                                                                │ │
│ │                  │   │   attr {                                                                                                                                               │ │
│ │                  │   │     key: "_user_specified_name"                                                                                                                        │ │
│ │                  │   │     value {                                                                                                                                            │ │
│ │                  │   │   │   s: "resource"                                                                                                                                    │ │
│ │                  │   │     }                                                                                                                                                  │ │
│ │                  │   │   }                                                                                                                                                    │ │
│ │                  │     }                                                                                                                                                      │ │
│ │                  │   }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                  }                                                                                                                                                            │ │
│ │                  versions {                                                                                                                                                   │ │
│ │                    producer: 2407                                                                                                                                             │ │
│ │                    min_consumer: 12                                                                                                                                           │ │
│ │                  }                                                                                                                                                            │ │
│ │  input_tensors = [<tf.Tensor 'serving_default_args_0:0' shape=(1, 3, 8, 8) dtype=float16>]                                                                                    │ │
│ │ output_tensors = [<tf.Tensor 'StatefulPartitionedCall:0' shape=(1, 3) dtype=float16>]                                                                                         │ │
│ │           self = <tensorflow.lite.python.lite.TFLiteSavedModelConverterV2 object at 0x70751c76eba0>                                                                           │ │
│ ╰───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯ │
│                                                                                                                                                                                   │
│ /data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/tensorflow/lite/python/lite.py:1448 in _convert_from_saved_model                     │
│                                                                                                                                                                                   │
│   1445 │   converter_kwargs.update(self._get_base_converter_args())                                                                                                               │
│   1446 │   converter_kwargs.update(quant_mode.converter_flags())                                                                                                                  │
│   1447 │                                                                                                                                                                          │
│ ❱ 1448 │   result = _convert_saved_model(**converter_kwargs)                                                                                                                      │
│   1449 │   return self._optimize_tflite_model(                                                                                                                                    │
│   1450 │   │   result,                                                                                                                                                            │
│   1451 │   │   quant_mode,                                                                                                                                                        │
│                                                                                                                                                                                   │
│ ╭─────────────────────────────────────────────────────────────────────────────────── locals ────────────────────────────────────────────────────────────────────────────────────╮ │
│ │ converter_kwargs = {                                                                                                                                                          │ │
│ │                    │   'enable_tflite_resource_variables': True,                                                                                                              │ │
│ │                    │   'input_format': 1,                                                                                                                                     │ │
│ │                    │   'allow_custom_ops': False,                                                                                                                             │ │
│ │                    │   'debug_info': ,                                                                                                                                        │ │
│ │                    │   'target_ops': {<OpsSet.TFLITE_BUILTINS: 'TFLITE_BUILTINS'>},                                                                                           │ │
│ │                    │   'select_user_tf_ops': set(),                                                                                                                           │ │
│ │                    │   'supported_backends': None,                                                                                                                            │ │
│ │                    │   'unfold_batchmatmul': False,                                                                                                                           │ │
│ │                    │   'legalize_custom_tensor_list_ops': False,                                                                                                              │ │
│ │                    │   'lower_tensor_list_ops': True,                                                                                                                         │ │
│ │                    │   ... +39                                                                                                                                                │ │
│ │                    }                                                                                                                                                          │ │
│ │        graph_def = node {                                                                                                                                                     │ │
│ │                      name: "Variable"                                                                                                                                         │ │
│ │                      op: "VarHandleOp"                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "shared_name"                                                                                                                                     │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     s: "Variable"                                                                                                                                        │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "shape"                                                                                                                                           │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     shape {                                                                                                                                              │ │
│ │                    │   │   dim {                                                                                                                                              │ │
│ │                    │   │     size: 3                                                                                                                                          │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "dtype"                                                                                                                                           │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     type: DT_HALF                                                                                                                                        │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "debug_name"                                                                                                                                      │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     s: "Variable/"                                                                                                                                       │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "container"                                                                                                                                       │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     s: ""                                                                                                                                                │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "allowed_devices"                                                                                                                                 │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     list {                                                                                                                                               │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "_output_shapes"                                                                                                                                  │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     list {                                                                                                                                               │ │
│ │                    │   │   shape {                                                                                                                                            │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    node {                                                                                                                                                     │ │
│ │                      name: "Variable/Read/ReadVariableOp"                                                                                                                     │ │
│ │                      op: "ReadVariableOp"                                                                                                                                     │ │
│ │                      input: "Variable"                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "dtype"                                                                                                                                           │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     type: DT_HALF                                                                                                                                        │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "_output_shapes"                                                                                                                                  │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     list {                                                                                                                                               │ │
│ │                    │   │   shape {                                                                                                                                            │ │
│ │                    │   │     dim {                                                                                                                                            │ │
│ │                    │   │   │   size: 3                                                                                                                                        │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    node {                                                                                                                                                     │ │
│ │                      name: "Variable_1"                                                                                                                                       │ │
│ │                      op: "VarHandleOp"                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "shared_name"                                                                                                                                     │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     s: "Variable_1"                                                                                                                                      │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "shape"                                                                                                                                           │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     shape {                                                                                                                                              │ │
│ │                    │   │   dim {                                                                                                                                              │ │
│ │                    │   │     size: 3                                                                                                                                          │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │   │   dim {                                                                                                                                              │ │
│ │                    │   │     size: 3                                                                                                                                          │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │   │   dim {                                                                                                                                              │ │
│ │                    │   │     size: 8                                                                                                                                          │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │   │   dim {                                                                                                                                              │ │
│ │                    │   │     size: 8                                                                                                                                          │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "dtype"                                                                                                                                           │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     type: DT_HALF                                                                                                                                        │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "debug_name"                                                                                                                                      │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     s: "Variable_1/"                                                                                                                                     │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "container"                                                                                                                                       │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     s: ""                                                                                                                                                │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "allowed_devices"                                                                                                                                 │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     list {                                                                                                                                               │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "_output_shapes"                                                                                                                                  │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     list {                                                                                                                                               │ │
│ │                    │   │   shape {                                                                                                                                            │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    node {                                                                                                                                                     │ │
│ │                      name: "Variable_1/Read/ReadVariableOp"                                                                                                                   │ │
│ │                      op: "ReadVariableOp"                                                                                                                                     │ │
│ │                      input: "Variable_1"                                                                                                                                      │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "dtype"                                                                                                                                           │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     type: DT_HALF                                                                                                                                        │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "_output_shapes"                                                                                                                                  │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     list {                                                                                                                                               │ │
│ │                    │   │   shape {                                                                                                                                            │ │
│ │                    │   │     dim {                                                                                                                                            │ │
│ │                    │   │   │   size: 3                                                                                                                                        │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │     dim {                                                                                                                                            │ │
│ │                    │   │   │   size: 3                                                                                                                                        │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │     dim {                                                                                                                                            │ │
│ │                    │   │   │   size: 8                                                                                                                                        │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │     dim {                                                                                                                                            │ │
│ │                    │   │   │   size: 8                                                                                                                                        │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    node {                                                                                                                                                     │ │
│ │                      name: "serving_default_args_0"                                                                                                                           │ │
│ │                      op: "Placeholder"                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "shape"                                                                                                                                           │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     shape {                                                                                                                                              │ │
│ │                    │   │   dim {                                                                                                                                              │ │
│ │                    │   │     size: 1                                                                                                                                          │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │   │   dim {                                                                                                                                              │ │
│ │                    │   │     size: 3                                                                                                                                          │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │   │   dim {                                                                                                                                              │ │
│ │                    │   │     size: 8                                                                                                                                          │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │   │   dim {                                                                                                                                              │ │
│ │                    │   │     size: 8                                                                                                                                          │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "dtype"                                                                                                                                           │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     type: DT_HALF                                                                                                                                        │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "_output_shapes"                                                                                                                                  │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     list {                                                                                                                                               │ │
│ │                    │   │   shape {                                                                                                                                            │ │
│ │                    │   │     dim {                                                                                                                                            │ │
│ │                    │   │   │   size: 1                                                                                                                                        │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │     dim {                                                                                                                                            │ │
│ │                    │   │   │   size: 3                                                                                                                                        │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │     dim {                                                                                                                                            │ │
│ │                    │   │   │   size: 8                                                                                                                                        │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │     dim {                                                                                                                                            │ │
│ │                    │   │   │   size: 8                                                                                                                                        │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    node {                                                                                                                                                     │ │
│ │                      name: "StatefulPartitionedCall"                                                                                                                          │ │
│ │                      op: "StatefulPartitionedCall"                                                                                                                            │ │
│ │                      input: "serving_default_args_0"                                                                                                                          │ │
│ │                      input: "Variable_1"                                                                                                                                      │ │
│ │                      input: "Variable"                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "f"                                                                                                                                               │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     func {                                                                                                                                               │ │
│ │                    │   │   name: "__inference_signature_wrapper_25"                                                                                                           │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "executor_type"                                                                                                                                   │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     s: ""                                                                                                                                                │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "config"                                                                                                                                          │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     s: ""                                                                                                                                                │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "config_proto"                                                                                                                                    │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     s: "\n\007\n\003CPU\020\001\n\007\n\003GPU\020\0002\002J\0008\001\202\001\000\222\001\002J\000"                                                      │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "_read_only_resource_inputs"                                                                                                                      │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     list {                                                                                                                                               │ │
│ │                    │   │   i: 1                                                                                                                                               │ │
│ │                    │   │   i: 2                                                                                                                                               │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "_output_shapes"                                                                                                                                  │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     list {                                                                                                                                               │ │
│ │                    │   │   shape {                                                                                                                                            │ │
│ │                    │   │     dim {                                                                                                                                            │ │
│ │                    │   │   │   size: 1                                                                                                                                        │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │     dim {                                                                                                                                            │ │
│ │                    │   │   │   size: 3                                                                                                                                        │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "_collective_manager_ids"                                                                                                                         │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     list {                                                                                                                                               │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "Tout"                                                                                                                                            │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     list {                                                                                                                                               │ │
│ │                    │   │   type: DT_HALF                                                                                                                                      │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "Tin"                                                                                                                                             │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     list {                                                                                                                                               │ │
│ │                    │   │   type: DT_HALF                                                                                                                                      │ │
│ │                    │   │   type: DT_RESOURCE                                                                                                                                  │ │
│ │                    │   │   type: DT_RESOURCE                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    node {                                                                                                                                                     │ │
│ │                      name: "NoOp"                                                                                                                                             │ │
│ │                      op: "NoOp"                                                                                                                                               │ │
│ │                    }                                                                                                                                                          │ │
│ │                    node {                                                                                                                                                     │ │
│ │                      name: "Const"                                                                                                                                            │ │
│ │                      op: "Const"                                                                                                                                              │ │
│ │                      device: "/device:CPU:0"                                                                                                                                  │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "value"                                                                                                                                           │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     tensor {                                                                                                                                             │ │
│ │                    │   │   dtype: DT_STRING                                                                                                                                   │ │
│ │                    │   │   tensor_shape {                                                                                                                                     │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │   │   string_val:                                                                                                                                        │ │
│ │                    "\n+\n\005\010\001\022\001f\n\016\010\002\022\n_variables\n\016\010\003\022\nsignatures*\002\010\001\n\t\n\005\010\004\022\0010*\000\n\022\n\005\010\005\… │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "dtype"                                                                                                                                           │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     type: DT_STRING                                                                                                                                      │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "_output_shapes"                                                                                                                                  │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     list {                                                                                                                                               │ │
│ │                    │   │   shape {                                                                                                                                            │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    node {                                                                                                                                                     │ │
│ │                      name: "saver_filename"                                                                                                                                   │ │
│ │                      op: "Placeholder"                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "shape"                                                                                                                                           │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     shape {                                                                                                                                              │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "dtype"                                                                                                                                           │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     type: DT_STRING                                                                                                                                      │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "_output_shapes"                                                                                                                                  │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     list {                                                                                                                                               │ │
│ │                    │   │   shape {                                                                                                                                            │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    node {                                                                                                                                                     │ │
│ │                      name: "StatefulPartitionedCall_1"                                                                                                                        │ │
│ │                      op: "StatefulPartitionedCall"                                                                                                                            │ │
│ │                      input: "saver_filename"                                                                                                                                  │ │
│ │                      input: "Variable_1"                                                                                                                                      │ │
│ │                      input: "Variable"                                                                                                                                        │ │
│ │                      input: "Const"                                                                                                                                           │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "f"                                                                                                                                               │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     func {                                                                                                                                               │ │
│ │                    │   │   name: "__inference__traced_save_68"                                                                                                                │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "executor_type"                                                                                                                                   │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     s: ""                                                                                                                                                │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "config"                                                                                                                                          │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     s: ""                                                                                                                                                │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "config_proto"                                                                                                                                    │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     s: "\n\007\n\003CPU\020\001\n\007\n\003GPU\020\0002\002J\0008\001\202\001\000\222\001\002J\000"                                                      │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "_read_only_resource_inputs"                                                                                                                      │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     list {                                                                                                                                               │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "_output_shapes"                                                                                                                                  │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     list {                                                                                                                                               │ │
│ │                    │   │   shape {                                                                                                                                            │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "_collective_manager_ids"                                                                                                                         │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     list {                                                                                                                                               │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "Tout"                                                                                                                                            │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     list {                                                                                                                                               │ │
│ │                    │   │   type: DT_STRING                                                                                                                                    │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "Tin"                                                                                                                                             │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     list {                                                                                                                                               │ │
│ │                    │   │   type: DT_STRING                                                                                                                                    │ │
│ │                    │   │   type: DT_RESOURCE                                                                                                                                  │ │
│ │                    │   │   type: DT_RESOURCE                                                                                                                                  │ │
│ │                    │   │   type: DT_STRING                                                                                                                                    │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    node {                                                                                                                                                     │ │
│ │                      name: "StatefulPartitionedCall_2"                                                                                                                        │ │
│ │                      op: "StatefulPartitionedCall"                                                                                                                            │ │
│ │                      input: "saver_filename"                                                                                                                                  │ │
│ │                      input: "Variable_1"                                                                                                                                      │ │
│ │                      input: "Variable"                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "f"                                                                                                                                               │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     func {                                                                                                                                               │ │
│ │                    │   │   name: "__inference__traced_restore_83"                                                                                                             │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "executor_type"                                                                                                                                   │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     s: ""                                                                                                                                                │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "config"                                                                                                                                          │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     s: ""                                                                                                                                                │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "config_proto"                                                                                                                                    │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     s: "\n\007\n\003CPU\020\001\n\007\n\003GPU\020\0002\002J\0008\001\202\001\000\222\001\002J\000"                                                      │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "_read_only_resource_inputs"                                                                                                                      │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     list {                                                                                                                                               │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "_output_shapes"                                                                                                                                  │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     list {                                                                                                                                               │ │
│ │                    │   │   shape {                                                                                                                                            │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "_collective_manager_ids"                                                                                                                         │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     list {                                                                                                                                               │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "Tout"                                                                                                                                            │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     list {                                                                                                                                               │ │
│ │                    │   │   type: DT_STRING                                                                                                                                    │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      attr {                                                                                                                                                   │ │
│ │                    │   key: "Tin"                                                                                                                                             │ │
│ │                    │   value {                                                                                                                                                │ │
│ │                    │     list {                                                                                                                                               │ │
│ │                    │   │   type: DT_STRING                                                                                                                                    │ │
│ │                    │   │   type: DT_RESOURCE                                                                                                                                  │ │
│ │                    │   │   type: DT_RESOURCE                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    library {                                                                                                                                                  │ │
│ │                      function {                                                                                                                                               │ │
│ │                    │   signature {                                                                                                                                            │ │
│ │                    │     name: "__inference_signature_wrapper_25"                                                                                                             │ │
│ │                    │     input_arg {                                                                                                                                          │ │
│ │                    │   │   name: "args_0"                                                                                                                                     │ │
│ │                    │   │   type: DT_HALF                                                                                                                                      │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     input_arg {                                                                                                                                          │ │
│ │                    │   │   name: "unknown"                                                                                                                                    │ │
│ │                    │   │   type: DT_RESOURCE                                                                                                                                  │ │
│ │                    │   │   handle_data {                                                                                                                                      │ │
│ │                    │   │     dtype: DT_HALF                                                                                                                                   │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 3                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 3                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 8                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 8                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     input_arg {                                                                                                                                          │ │
│ │                    │   │   name: "unknown_0"                                                                                                                                  │ │
│ │                    │   │   type: DT_RESOURCE                                                                                                                                  │ │
│ │                    │   │   handle_data {                                                                                                                                      │ │
│ │                    │   │     dtype: DT_HALF                                                                                                                                   │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 3                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     output_arg {                                                                                                                                         │ │
│ │                    │   │   name: "identity"                                                                                                                                   │ │
│ │                    │   │   type: DT_HALF                                                                                                                                      │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     is_stateful: true                                                                                                                                    │ │
│ │                    │     control_output: "StatefulPartitionedCall"                                                                                                            │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "StatefulPartitionedCall"                                                                                                                      │ │
│ │                    │     op: "StatefulPartitionedCall"                                                                                                                        │ │
│ │                    │     input: "args_0"                                                                                                                                      │ │
│ │                    │     input: "unknown"                                                                                                                                     │ │
│ │                    │     input: "unknown_0"                                                                                                                                   │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "f"                                                                                                                                           │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     func {                                                                                                                                           │ │
│ │                    │   │   │   name: "__inference_inner_15"                                                                                                                   │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "config_proto"                                                                                                                                │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     s: "\n\007\n\003CPU\020\001\n\007\n\003GPU\020\0002\002J\0008\001\202\001\000\222\001\002J\000"                                                  │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_read_only_resource_inputs"                                                                                                                  │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   i: 1                                                                                                                                           │ │
│ │                    │   │   │   i: 2                                                                                                                                           │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 1                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_collective_manager_ids"                                                                                                                     │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "Tout"                                                                                                                                        │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   type: DT_HALF                                                                                                                                  │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "Tin"                                                                                                                                         │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   type: DT_HALF                                                                                                                                  │ │
│ │                    │   │   │   type: DT_RESOURCE                                                                                                                              │ │
│ │                    │   │   │   type: DT_RESOURCE                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "Identity"                                                                                                                                     │ │
│ │                    │     op: "Identity"                                                                                                                                       │ │
│ │                    │     input: "StatefulPartitionedCall:output:0"                                                                                                            │ │
│ │                    │     input: "^NoOp"                                                                                                                                       │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 1                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "T"                                                                                                                                           │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_HALF                                                                                                                                    │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "NoOp"                                                                                                                                         │ │
│ │                    │     op: "NoOp"                                                                                                                                           │ │
│ │                    │     input: "^StatefulPartitionedCall"                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   ret {                                                                                                                                                  │ │
│ │                    │     key: "identity"                                                                                                                                      │ │
│ │                    │     value: "Identity:output:0"                                                                                                                           │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   attr {                                                                                                                                                 │ │
│ │                    │     key: "_input_shapes"                                                                                                                                 │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   list {                                                                                                                                             │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 1                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 3                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 8                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 8                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   attr {                                                                                                                                                 │ │
│ │                    │     key: "_construction_context"                                                                                                                         │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   s: "kEagerRuntime"                                                                                                                                 │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   control_ret {                                                                                                                                          │ │
│ │                    │     key: "StatefulPartitionedCall"                                                                                                                       │ │
│ │                    │     value: "StatefulPartitionedCall"                                                                                                                     │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   arg_attr {                                                                                                                                             │ │
│ │                    │     key: 0                                                                                                                                               │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   attr {                                                                                                                                             │ │
│ │                    │   │     key: "_user_specified_name"                                                                                                                      │ │
│ │                    │   │     value {                                                                                                                                          │ │
│ │                    │   │   │   s: "args_0"                                                                                                                                    │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │   │   attr {                                                                                                                                             │ │
│ │                    │   │     key: "_output_shapes"                                                                                                                            │ │
│ │                    │   │     value {                                                                                                                                          │ │
│ │                    │   │   │   list {                                                                                                                                         │ │
│ │                    │   │   │     shape {                                                                                                                                      │ │
│ │                    │   │   │   │   dim {                                                                                                                                      │ │
│ │                    │   │   │   │     size: 1                                                                                                                                  │ │
│ │                    │   │   │   │   }                                                                                                                                          │ │
│ │                    │   │   │   │   dim {                                                                                                                                      │ │
│ │                    │   │   │   │     size: 3                                                                                                                                  │ │
│ │                    │   │   │   │   }                                                                                                                                          │ │
│ │                    │   │   │   │   dim {                                                                                                                                      │ │
│ │                    │   │   │   │     size: 8                                                                                                                                  │ │
│ │                    │   │   │   │   }                                                                                                                                          │ │
│ │                    │   │   │   │   dim {                                                                                                                                      │ │
│ │                    │   │   │   │     size: 8                                                                                                                                  │ │
│ │                    │   │   │   │   }                                                                                                                                          │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   arg_attr {                                                                                                                                             │ │
│ │                    │     key: 1                                                                                                                                               │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   attr {                                                                                                                                             │ │
│ │                    │   │     key: "_user_specified_name"                                                                                                                      │ │
│ │                    │   │     value {                                                                                                                                          │ │
│ │                    │   │   │   s: "19"                                                                                                                                        │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   arg_attr {                                                                                                                                             │ │
│ │                    │     key: 2                                                                                                                                               │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   attr {                                                                                                                                             │ │
│ │                    │   │     key: "_user_specified_name"                                                                                                                      │ │
│ │                    │   │     value {                                                                                                                                          │ │
│ │                    │   │   │   s: "21"                                                                                                                                        │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      function {                                                                                                                                               │ │
│ │                    │   signature {                                                                                                                                            │ │
│ │                    │     name: "__inference_inner_15"                                                                                                                         │ │
│ │                    │     input_arg {                                                                                                                                          │ │
│ │                    │   │   name: "args_0"                                                                                                                                     │ │
│ │                    │   │   type: DT_HALF                                                                                                                                      │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     input_arg {                                                                                                                                          │ │
│ │                    │   │   name: "xlacallmodule_readvariableop_resource"                                                                                                      │ │
│ │                    │   │   type: DT_RESOURCE                                                                                                                                  │ │
│ │                    │   │   handle_data {                                                                                                                                      │ │
│ │                    │   │     dtype: DT_HALF                                                                                                                                   │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 3                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 3                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 8                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 8                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     input_arg {                                                                                                                                          │ │
│ │                    │   │   name: "xlacallmodule_readvariableop_1_resource"                                                                                                    │ │
│ │                    │   │   type: DT_RESOURCE                                                                                                                                  │ │
│ │                    │   │   handle_data {                                                                                                                                      │ │
│ │                    │   │     dtype: DT_HALF                                                                                                                                   │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 3                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     output_arg {                                                                                                                                         │ │
│ │                    │   │   name: "identity"                                                                                                                                   │ │
│ │                    │   │   type: DT_HALF                                                                                                                                      │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     is_stateful: true                                                                                                                                    │ │
│ │                    │     control_output: "XlaCallModule"                                                                                                                      │ │
│ │                    │     control_output: "XlaCallModule/ReadVariableOp"                                                                                                       │ │
│ │                    │     control_output: "XlaCallModule/ReadVariableOp_1"                                                                                                     │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "XlaCallModule/ReadVariableOp"                                                                                                                 │ │
│ │                    │     op: "ReadVariableOp"                                                                                                                                 │ │
│ │                    │     input: "xlacallmodule_readvariableop_resource"                                                                                                       │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "dtype"                                                                                                                                       │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_HALF                                                                                                                                    │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 8                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 8                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "XlaCallModule/ReadVariableOp_1"                                                                                                               │ │
│ │                    │     op: "ReadVariableOp"                                                                                                                                 │ │
│ │                    │     input: "xlacallmodule_readvariableop_1_resource"                                                                                                     │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "dtype"                                                                                                                                       │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_HALF                                                                                                                                    │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "XlaCallModule"                                                                                                                                │ │
│ │                    │     op: "XlaCallModule"                                                                                                                                  │ │
│ │                    │     input: "XlaCallModule/ReadVariableOp:value:0"                                                                                                        │ │
│ │                    │     input: "XlaCallModule/ReadVariableOp_1:value:0"                                                                                                      │ │
│ │                    │     input: "args_0"                                                                                                                                      │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "version"                                                                                                                                     │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     i: 5                                                                                                                                             │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "module"                                                                                                                                      │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     s:                                                                                                                                               │ │
│ │                    "ML\357R\rStableHLO_v1.13.0\000\001#\005\001\005\023\001\003\013\003\021\017\023\027\033\037#\'+\003\257o1\001#\007\017\013\013\017\013\023\013\017\017\0… │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 1                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "Tout"                                                                                                                                        │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   type: DT_HALF                                                                                                                                  │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "Tin"                                                                                                                                         │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   type: DT_HALF                                                                                                                                  │ │
│ │                    │   │   │   type: DT_HALF                                                                                                                                  │ │
│ │                    │   │   │   type: DT_HALF                                                                                                                                  │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "Sout"                                                                                                                                        │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 1                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "Identity"                                                                                                                                     │ │
│ │                    │     op: "Identity"                                                                                                                                       │ │
│ │                    │     input: "XlaCallModule:output:0"                                                                                                                      │ │
│ │                    │     input: "^NoOp"                                                                                                                                       │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 1                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "T"                                                                                                                                           │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_HALF                                                                                                                                    │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "NoOp"                                                                                                                                         │ │
│ │                    │     op: "NoOp"                                                                                                                                           │ │
│ │                    │     input: "^XlaCallModule"                                                                                                                              │ │
│ │                    │     input: "^XlaCallModule/ReadVariableOp"                                                                                                               │ │
│ │                    │     input: "^XlaCallModule/ReadVariableOp_1"                                                                                                             │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   ret {                                                                                                                                                  │ │
│ │                    │     key: "identity"                                                                                                                                      │ │
│ │                    │     value: "Identity:output:0"                                                                                                                           │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   attr {                                                                                                                                                 │ │
│ │                    │     key: "_input_shapes"                                                                                                                                 │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   list {                                                                                                                                             │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 1                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 3                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 8                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 8                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   attr {                                                                                                                                                 │ │
│ │                    │     key: "_construction_context"                                                                                                                         │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   s: "kEagerRuntime"                                                                                                                                 │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   control_ret {                                                                                                                                          │ │
│ │                    │     key: "XlaCallModule"                                                                                                                                 │ │
│ │                    │     value: "XlaCallModule"                                                                                                                               │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   control_ret {                                                                                                                                          │ │
│ │                    │     key: "XlaCallModule/ReadVariableOp"                                                                                                                  │ │
│ │                    │     value: "XlaCallModule/ReadVariableOp"                                                                                                                │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   control_ret {                                                                                                                                          │ │
│ │                    │     key: "XlaCallModule/ReadVariableOp_1"                                                                                                                │ │
│ │                    │     value: "XlaCallModule/ReadVariableOp_1"                                                                                                              │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   arg_attr {                                                                                                                                             │ │
│ │                    │     key: 0                                                                                                                                               │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   attr {                                                                                                                                             │ │
│ │                    │   │     key: "_user_specified_name"                                                                                                                      │ │
│ │                    │   │     value {                                                                                                                                          │ │
│ │                    │   │   │   s: "args_0"                                                                                                                                    │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │   │   attr {                                                                                                                                             │ │
│ │                    │   │     key: "_output_shapes"                                                                                                                            │ │
│ │                    │   │     value {                                                                                                                                          │ │
│ │                    │   │   │   list {                                                                                                                                         │ │
│ │                    │   │   │     shape {                                                                                                                                      │ │
│ │                    │   │   │   │   dim {                                                                                                                                      │ │
│ │                    │   │   │   │     size: 1                                                                                                                                  │ │
│ │                    │   │   │   │   }                                                                                                                                          │ │
│ │                    │   │   │   │   dim {                                                                                                                                      │ │
│ │                    │   │   │   │     size: 3                                                                                                                                  │ │
│ │                    │   │   │   │   }                                                                                                                                          │ │
│ │                    │   │   │   │   dim {                                                                                                                                      │ │
│ │                    │   │   │   │     size: 8                                                                                                                                  │ │
│ │                    │   │   │   │   }                                                                                                                                          │ │
│ │                    │   │   │   │   dim {                                                                                                                                      │ │
│ │                    │   │   │   │     size: 8                                                                                                                                  │ │
│ │                    │   │   │   │   }                                                                                                                                          │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   arg_attr {                                                                                                                                             │ │
│ │                    │     key: 1                                                                                                                                               │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   attr {                                                                                                                                             │ │
│ │                    │   │     key: "_user_specified_name"                                                                                                                      │ │
│ │                    │   │     value {                                                                                                                                          │ │
│ │                    │   │   │   s: "resource"                                                                                                                                  │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   arg_attr {                                                                                                                                             │ │
│ │                    │     key: 2                                                                                                                                               │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   attr {                                                                                                                                             │ │
│ │                    │   │     key: "_user_specified_name"                                                                                                                      │ │
│ │                    │   │     value {                                                                                                                                          │ │
│ │                    │   │   │   s: "resource"                                                                                                                                  │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      function {                                                                                                                                               │ │
│ │                    │   signature {                                                                                                                                            │ │
│ │                    │     name: "__inference__traced_restore_83"                                                                                                               │ │
│ │                    │     input_arg {                                                                                                                                          │ │
│ │                    │   │   name: "file_prefix"                                                                                                                                │ │
│ │                    │   │   type: DT_STRING                                                                                                                                    │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     input_arg {                                                                                                                                          │ │
│ │                    │   │   name: "assignvariableop_variable_1"                                                                                                                │ │
│ │                    │   │   type: DT_RESOURCE                                                                                                                                  │ │
│ │                    │   │   handle_data {                                                                                                                                      │ │
│ │                    │   │     dtype: DT_HALF                                                                                                                                   │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 3                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 3                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 8                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 8                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     input_arg {                                                                                                                                          │ │
│ │                    │   │   name: "assignvariableop_1_variable"                                                                                                                │ │
│ │                    │   │   type: DT_RESOURCE                                                                                                                                  │ │
│ │                    │   │   handle_data {                                                                                                                                      │ │
│ │                    │   │     dtype: DT_HALF                                                                                                                                   │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 3                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     output_arg {                                                                                                                                         │ │
│ │                    │   │   name: "identity_3"                                                                                                                                 │ │
│ │                    │   │   type: DT_STRING                                                                                                                                    │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     is_stateful: true                                                                                                                                    │ │
│ │                    │     control_output: "AssignVariableOp"                                                                                                                   │ │
│ │                    │     control_output: "AssignVariableOp_1"                                                                                                                 │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "RestoreV2/tensor_names"                                                                                                                       │ │
│ │                    │     op: "Const"                                                                                                                                          │ │
│ │                    │     device: "/device:CPU:0"                                                                                                                              │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "value"                                                                                                                                       │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     tensor {                                                                                                                                         │ │
│ │                    │   │   │   dtype: DT_STRING                                                                                                                               │ │
│ │                    │   │   │   tensor_shape {                                                                                                                                 │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   string_val: "_variables/0/.ATTRIBUTES/VARIABLE_VALUE"                                                                                          │ │
│ │                    │   │   │   string_val: "_variables/1/.ATTRIBUTES/VARIABLE_VALUE"                                                                                          │ │
│ │                    │   │   │   string_val: "_CHECKPOINTABLE_OBJECT_GRAPH"                                                                                                     │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "dtype"                                                                                                                                       │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_STRING                                                                                                                                  │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "RestoreV2/shape_and_slices"                                                                                                                   │ │
│ │                    │     op: "Const"                                                                                                                                          │ │
│ │                    │     device: "/device:CPU:0"                                                                                                                              │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "value"                                                                                                                                       │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     tensor {                                                                                                                                         │ │
│ │                    │   │   │   dtype: DT_STRING                                                                                                                               │ │
│ │                    │   │   │   tensor_shape {                                                                                                                                 │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   string_val: ""                                                                                                                                 │ │
│ │                    │   │   │   string_val: ""                                                                                                                                 │ │
│ │                    │   │   │   string_val: ""                                                                                                                                 │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "dtype"                                                                                                                                       │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_STRING                                                                                                                                  │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "RestoreV2"                                                                                                                                    │ │
│ │                    │     op: "RestoreV2"                                                                                                                                      │ │
│ │                    │     input: "file_prefix"                                                                                                                                 │ │
│ │                    │     input: "RestoreV2/tensor_names:output:0"                                                                                                             │ │
│ │                    │     input: "RestoreV2/shape_and_slices:output:0"                                                                                                         │ │
│ │                    │     device: "/device:CPU:0"                                                                                                                              │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "dtypes"                                                                                                                                      │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   type: DT_HALF                                                                                                                                  │ │
│ │                    │   │   │   type: DT_HALF                                                                                                                                  │ │
│ │                    │   │   │   type: DT_STRING                                                                                                                                │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     unknown_rank: true                                                                                                                           │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     unknown_rank: true                                                                                                                           │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     unknown_rank: true                                                                                                                           │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "Identity"                                                                                                                                     │ │
│ │                    │     op: "Identity"                                                                                                                                       │ │
│ │                    │     input: "RestoreV2:tensors:0"                                                                                                                         │ │
│ │                    │     device: "/device:CPU:0"                                                                                                                              │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     unknown_rank: true                                                                                                                           │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "T"                                                                                                                                           │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_HALF                                                                                                                                    │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "AssignVariableOp"                                                                                                                             │ │
│ │                    │     op: "AssignVariableOp"                                                                                                                               │ │
│ │                    │     input: "assignvariableop_variable_1"                                                                                                                 │ │
│ │                    │     input: "Identity:output:0"                                                                                                                           │ │
│ │                    │     device: "/device:CPU:0"                                                                                                                              │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "dtype"                                                                                                                                       │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_HALF                                                                                                                                    │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_has_manual_control_dependencies"                                                                                                            │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     b: true                                                                                                                                          │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "Identity_1"                                                                                                                                   │ │
│ │                    │     op: "Identity"                                                                                                                                       │ │
│ │                    │     input: "RestoreV2:tensors:1"                                                                                                                         │ │
│ │                    │     device: "/device:CPU:0"                                                                                                                              │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     unknown_rank: true                                                                                                                           │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "T"                                                                                                                                           │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_HALF                                                                                                                                    │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "AssignVariableOp_1"                                                                                                                           │ │
│ │                    │     op: "AssignVariableOp"                                                                                                                               │ │
│ │                    │     input: "assignvariableop_1_variable"                                                                                                                 │ │
│ │                    │     input: "Identity_1:output:0"                                                                                                                         │ │
│ │                    │     device: "/device:CPU:0"                                                                                                                              │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "dtype"                                                                                                                                       │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_HALF                                                                                                                                    │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_has_manual_control_dependencies"                                                                                                            │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     b: true                                                                                                                                          │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "NoOp"                                                                                                                                         │ │
│ │                    │     op: "NoOp"                                                                                                                                           │ │
│ │                    │     device: "/device:CPU:0"                                                                                                                              │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_has_manual_control_dependencies"                                                                                                            │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     b: true                                                                                                                                          │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "Identity_2"                                                                                                                                   │ │
│ │                    │     op: "Identity"                                                                                                                                       │ │
│ │                    │     input: "file_prefix"                                                                                                                                 │ │
│ │                    │     input: "^AssignVariableOp"                                                                                                                           │ │
│ │                    │     input: "^AssignVariableOp_1"                                                                                                                         │ │
│ │                    │     input: "^NoOp"                                                                                                                                       │ │
│ │                    │     device: "/device:CPU:0"                                                                                                                              │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "T"                                                                                                                                           │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_STRING                                                                                                                                  │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "Identity_3"                                                                                                                                   │ │
│ │                    │     op: "Identity"                                                                                                                                       │ │
│ │                    │     input: "Identity_2:output:0"                                                                                                                         │ │
│ │                    │     input: "^NoOp_1"                                                                                                                                     │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "T"                                                                                                                                           │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_STRING                                                                                                                                  │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "NoOp_1"                                                                                                                                       │ │
│ │                    │     op: "NoOp"                                                                                                                                           │ │
│ │                    │     input: "^AssignVariableOp"                                                                                                                           │ │
│ │                    │     input: "^AssignVariableOp_1"                                                                                                                         │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   ret {                                                                                                                                                  │ │
│ │                    │     key: "identity_3"                                                                                                                                    │ │
│ │                    │     value: "Identity_3:output:0"                                                                                                                         │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   attr {                                                                                                                                                 │ │
│ │                    │     key: "_input_shapes"                                                                                                                                 │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   list {                                                                                                                                             │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   attr {                                                                                                                                                 │ │
│ │                    │     key: "_construction_context"                                                                                                                         │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   s: "kEagerRuntime"                                                                                                                                 │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   control_ret {                                                                                                                                          │ │
│ │                    │     key: "AssignVariableOp"                                                                                                                              │ │
│ │                    │     value: "AssignVariableOp"                                                                                                                            │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   control_ret {                                                                                                                                          │ │
│ │                    │     key: "AssignVariableOp_1"                                                                                                                            │ │
│ │                    │     value: "AssignVariableOp_1"                                                                                                                          │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   arg_attr {                                                                                                                                             │ │
│ │                    │     key: 0                                                                                                                                               │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   attr {                                                                                                                                             │ │
│ │                    │   │     key: "_user_specified_name"                                                                                                                      │ │
│ │                    │   │     value {                                                                                                                                          │ │
│ │                    │   │   │   s: "file_prefix"                                                                                                                               │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │   │   attr {                                                                                                                                             │ │
│ │                    │   │     key: "_output_shapes"                                                                                                                            │ │
│ │                    │   │     value {                                                                                                                                          │ │
│ │                    │   │   │   list {                                                                                                                                         │ │
│ │                    │   │   │     shape {                                                                                                                                      │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   arg_attr {                                                                                                                                             │ │
│ │                    │     key: 1                                                                                                                                               │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   attr {                                                                                                                                             │ │
│ │                    │   │     key: "_user_specified_name"                                                                                                                      │ │
│ │                    │   │     value {                                                                                                                                          │ │
│ │                    │   │   │   s: "Variable_1"                                                                                                                                │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   arg_attr {                                                                                                                                             │ │
│ │                    │     key: 2                                                                                                                                               │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   attr {                                                                                                                                             │ │
│ │                    │   │     key: "_user_specified_name"                                                                                                                      │ │
│ │                    │   │     value {                                                                                                                                          │ │
│ │                    │   │   │   s: "Variable"                                                                                                                                  │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      function {                                                                                                                                               │ │
│ │                    │   signature {                                                                                                                                            │ │
│ │                    │     name: "__inference__traced_save_68"                                                                                                                  │ │
│ │                    │     input_arg {                                                                                                                                          │ │
│ │                    │   │   name: "file_prefix"                                                                                                                                │ │
│ │                    │   │   type: DT_STRING                                                                                                                                    │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     input_arg {                                                                                                                                          │ │
│ │                    │   │   name: "read_disablecopyonread_variable_1"                                                                                                          │ │
│ │                    │   │   type: DT_RESOURCE                                                                                                                                  │ │
│ │                    │   │   handle_data {                                                                                                                                      │ │
│ │                    │   │     dtype: DT_HALF                                                                                                                                   │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 3                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 3                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 8                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 8                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     input_arg {                                                                                                                                          │ │
│ │                    │   │   name: "read_1_disablecopyonread_variable"                                                                                                          │ │
│ │                    │   │   type: DT_RESOURCE                                                                                                                                  │ │
│ │                    │   │   handle_data {                                                                                                                                      │ │
│ │                    │   │     dtype: DT_HALF                                                                                                                                   │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 3                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     input_arg {                                                                                                                                          │ │
│ │                    │   │   name: "savev2_const"                                                                                                                               │ │
│ │                    │   │   type: DT_STRING                                                                                                                                    │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     output_arg {                                                                                                                                         │ │
│ │                    │   │   name: "identity_5"                                                                                                                                 │ │
│ │                    │   │   type: DT_STRING                                                                                                                                    │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     is_stateful: true                                                                                                                                    │ │
│ │                    │     control_output: "MergeV2Checkpoints"                                                                                                                 │ │
│ │                    │     control_output: "Read/DisableCopyOnRead"                                                                                                             │ │
│ │                    │     control_output: "Read/ReadVariableOp"                                                                                                                │ │
│ │                    │     control_output: "Read_1/DisableCopyOnRead"                                                                                                           │ │
│ │                    │     control_output: "Read_1/ReadVariableOp"                                                                                                              │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "StaticRegexFullMatch"                                                                                                                         │ │
│ │                    │     op: "StaticRegexFullMatch"                                                                                                                           │ │
│ │                    │     input: "file_prefix"                                                                                                                                 │ │
│ │                    │     device: "/device:CPU:*"                                                                                                                              │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "pattern"                                                                                                                                     │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     s: "^s3://.*"                                                                                                                                    │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "Const"                                                                                                                                        │ │
│ │                    │     op: "Const"                                                                                                                                          │ │
│ │                    │     device: "/device:CPU:*"                                                                                                                              │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "value"                                                                                                                                       │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     tensor {                                                                                                                                         │ │
│ │                    │   │   │   dtype: DT_STRING                                                                                                                               │ │
│ │                    │   │   │   tensor_shape {                                                                                                                                 │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   string_val: ".part"                                                                                                                            │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "dtype"                                                                                                                                       │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_STRING                                                                                                                                  │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "Const_1"                                                                                                                                      │ │
│ │                    │     op: "Const"                                                                                                                                          │ │
│ │                    │     device: "/device:CPU:*"                                                                                                                              │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "value"                                                                                                                                       │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     tensor {                                                                                                                                         │ │
│ │                    │   │   │   dtype: DT_STRING                                                                                                                               │ │
│ │                    │   │   │   tensor_shape {                                                                                                                                 │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   string_val: "_temp/part"                                                                                                                       │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "dtype"                                                                                                                                       │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_STRING                                                                                                                                  │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "Select"                                                                                                                                       │ │
│ │                    │     op: "Select"                                                                                                                                         │ │
│ │                    │     input: "StaticRegexFullMatch:output:0"                                                                                                               │ │
│ │                    │     input: "Const:output:0"                                                                                                                              │ │
│ │                    │     input: "Const_1:output:0"                                                                                                                            │ │
│ │                    │     device: "/device:CPU:*"                                                                                                                              │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "T"                                                                                                                                           │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_STRING                                                                                                                                  │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "StringJoin"                                                                                                                                   │ │
│ │                    │     op: "StringJoin"                                                                                                                                     │ │
│ │                    │     input: "file_prefix"                                                                                                                                 │ │
│ │                    │     input: "Select:output:0"                                                                                                                             │ │
│ │                    │     device: "/device:CPU:*"                                                                                                                              │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "N"                                                                                                                                           │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     i: 2                                                                                                                                             │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "Read/DisableCopyOnRead"                                                                                                                       │ │
│ │                    │     op: "DisableCopyOnRead"                                                                                                                              │ │
│ │                    │     input: "read_disablecopyonread_variable_1"                                                                                                           │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "Read/ReadVariableOp"                                                                                                                          │ │
│ │                    │     op: "ReadVariableOp"                                                                                                                                 │ │
│ │                    │     input: "read_disablecopyonread_variable_1"                                                                                                           │ │
│ │                    │     input: "^Read/DisableCopyOnRead"                                                                                                                     │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "dtype"                                                                                                                                       │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_HALF                                                                                                                                    │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 8                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 8                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "Identity"                                                                                                                                     │ │
│ │                    │     op: "Identity"                                                                                                                                       │ │
│ │                    │     input: "Read/ReadVariableOp:value:0"                                                                                                                 │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 8                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 8                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "T"                                                                                                                                           │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_HALF                                                                                                                                    │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "Identity_1"                                                                                                                                   │ │
│ │                    │     op: "Identity"                                                                                                                                       │ │
│ │                    │     input: "Identity:output:0"                                                                                                                           │ │
│ │                    │     device: "/device:CPU:0"                                                                                                                              │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 8                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 8                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "T"                                                                                                                                           │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_HALF                                                                                                                                    │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "Read_1/DisableCopyOnRead"                                                                                                                     │ │
│ │                    │     op: "DisableCopyOnRead"                                                                                                                              │ │
│ │                    │     input: "read_1_disablecopyonread_variable"                                                                                                           │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "Read_1/ReadVariableOp"                                                                                                                        │ │
│ │                    │     op: "ReadVariableOp"                                                                                                                                 │ │
│ │                    │     input: "read_1_disablecopyonread_variable"                                                                                                           │ │
│ │                    │     input: "^Read_1/DisableCopyOnRead"                                                                                                                   │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "dtype"                                                                                                                                       │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_HALF                                                                                                                                    │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "Identity_2"                                                                                                                                   │ │
│ │                    │     op: "Identity"                                                                                                                                       │ │
│ │                    │     input: "Read_1/ReadVariableOp:value:0"                                                                                                               │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "T"                                                                                                                                           │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_HALF                                                                                                                                    │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "Identity_3"                                                                                                                                   │ │
│ │                    │     op: "Identity"                                                                                                                                       │ │
│ │                    │     input: "Identity_2:output:0"                                                                                                                         │ │
│ │                    │     device: "/device:CPU:0"                                                                                                                              │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "T"                                                                                                                                           │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_HALF                                                                                                                                    │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "num_shards"                                                                                                                                   │ │
│ │                    │     op: "Const"                                                                                                                                          │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "value"                                                                                                                                       │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     tensor {                                                                                                                                         │ │
│ │                    │   │   │   dtype: DT_INT32                                                                                                                                │ │
│ │                    │   │   │   tensor_shape {                                                                                                                                 │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   int_val: 1                                                                                                                                     │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "dtype"                                                                                                                                       │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_INT32                                                                                                                                   │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "ShardedFilename/shard"                                                                                                                        │ │
│ │                    │     op: "Const"                                                                                                                                          │ │
│ │                    │     device: "/device:CPU:0"                                                                                                                              │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "value"                                                                                                                                       │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     tensor {                                                                                                                                         │ │
│ │                    │   │   │   dtype: DT_INT32                                                                                                                                │ │
│ │                    │   │   │   tensor_shape {                                                                                                                                 │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   int_val: 0                                                                                                                                     │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "dtype"                                                                                                                                       │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_INT32                                                                                                                                   │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "ShardedFilename"                                                                                                                              │ │
│ │                    │     op: "ShardedFilename"                                                                                                                                │ │
│ │                    │     input: "StringJoin:output:0"                                                                                                                         │ │
│ │                    │     input: "ShardedFilename/shard:output:0"                                                                                                              │ │
│ │                    │     input: "num_shards:output:0"                                                                                                                         │ │
│ │                    │     device: "/device:CPU:0"                                                                                                                              │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "SaveV2/tensor_names"                                                                                                                          │ │
│ │                    │     op: "Const"                                                                                                                                          │ │
│ │                    │     device: "/device:CPU:0"                                                                                                                              │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "value"                                                                                                                                       │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     tensor {                                                                                                                                         │ │
│ │                    │   │   │   dtype: DT_STRING                                                                                                                               │ │
│ │                    │   │   │   tensor_shape {                                                                                                                                 │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   string_val: "_variables/0/.ATTRIBUTES/VARIABLE_VALUE"                                                                                          │ │
│ │                    │   │   │   string_val: "_variables/1/.ATTRIBUTES/VARIABLE_VALUE"                                                                                          │ │
│ │                    │   │   │   string_val: "_CHECKPOINTABLE_OBJECT_GRAPH"                                                                                                     │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "dtype"                                                                                                                                       │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_STRING                                                                                                                                  │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "SaveV2/shape_and_slices"                                                                                                                      │ │
│ │                    │     op: "Const"                                                                                                                                          │ │
│ │                    │     device: "/device:CPU:0"                                                                                                                              │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "value"                                                                                                                                       │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     tensor {                                                                                                                                         │ │
│ │                    │   │   │   dtype: DT_STRING                                                                                                                               │ │
│ │                    │   │   │   tensor_shape {                                                                                                                                 │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   string_val: ""                                                                                                                                 │ │
│ │                    │   │   │   string_val: ""                                                                                                                                 │ │
│ │                    │   │   │   string_val: ""                                                                                                                                 │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "dtype"                                                                                                                                       │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_STRING                                                                                                                                  │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "SaveV2"                                                                                                                                       │ │
│ │                    │     op: "SaveV2"                                                                                                                                         │ │
│ │                    │     input: "ShardedFilename:filename:0"                                                                                                                  │ │
│ │                    │     input: "SaveV2/tensor_names:output:0"                                                                                                                │ │
│ │                    │     input: "SaveV2/shape_and_slices:output:0"                                                                                                            │ │
│ │                    │     input: "Identity_1:output:0"                                                                                                                         │ │
│ │                    │     input: "Identity_3:output:0"                                                                                                                         │ │
│ │                    │     input: "savev2_const"                                                                                                                                │ │
│ │                    │     device: "/device:CPU:0"                                                                                                                              │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "dtypes"                                                                                                                                      │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   type: DT_HALF                                                                                                                                  │ │
│ │                    │   │   │   type: DT_HALF                                                                                                                                  │ │
│ │                    │   │   │   type: DT_STRING                                                                                                                                │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_has_manual_control_dependencies"                                                                                                            │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     b: true                                                                                                                                          │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "MergeV2Checkpoints/checkpoint_prefixes"                                                                                                       │ │
│ │                    │     op: "Pack"                                                                                                                                           │ │
│ │                    │     input: "ShardedFilename:filename:0"                                                                                                                  │ │
│ │                    │     input: "^SaveV2"                                                                                                                                     │ │
│ │                    │     device: "/device:CPU:0"                                                                                                                              │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 1                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "T"                                                                                                                                           │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_STRING                                                                                                                                  │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "N"                                                                                                                                           │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     i: 1                                                                                                                                             │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "MergeV2Checkpoints"                                                                                                                           │ │
│ │                    │     op: "MergeV2Checkpoints"                                                                                                                             │ │
│ │                    │     input: "MergeV2Checkpoints/checkpoint_prefixes:output:0"                                                                                             │ │
│ │                    │     input: "file_prefix"                                                                                                                                 │ │
│ │                    │     device: "/device:CPU:0"                                                                                                                              │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_has_manual_control_dependencies"                                                                                                            │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     b: true                                                                                                                                          │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "Identity_4"                                                                                                                                   │ │
│ │                    │     op: "Identity"                                                                                                                                       │ │
│ │                    │     input: "file_prefix"                                                                                                                                 │ │
│ │                    │     input: "^MergeV2Checkpoints"                                                                                                                         │ │
│ │                    │     device: "/device:CPU:0"                                                                                                                              │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "T"                                                                                                                                           │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_STRING                                                                                                                                  │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "Identity_5"                                                                                                                                   │ │
│ │                    │     op: "Identity"                                                                                                                                       │ │
│ │                    │     input: "Identity_4:output:0"                                                                                                                         │ │
│ │                    │     input: "^NoOp"                                                                                                                                       │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "T"                                                                                                                                           │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_STRING                                                                                                                                  │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "NoOp"                                                                                                                                         │ │
│ │                    │     op: "NoOp"                                                                                                                                           │ │
│ │                    │     input: "^MergeV2Checkpoints"                                                                                                                         │ │
│ │                    │     input: "^Read/DisableCopyOnRead"                                                                                                                     │ │
│ │                    │     input: "^Read/ReadVariableOp"                                                                                                                        │ │
│ │                    │     input: "^Read_1/DisableCopyOnRead"                                                                                                                   │ │
│ │                    │     input: "^Read_1/ReadVariableOp"                                                                                                                      │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   ret {                                                                                                                                                  │ │
│ │                    │     key: "identity_5"                                                                                                                                    │ │
│ │                    │     value: "Identity_5:output:0"                                                                                                                         │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   attr {                                                                                                                                                 │ │
│ │                    │     key: "_input_shapes"                                                                                                                                 │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   list {                                                                                                                                             │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   attr {                                                                                                                                                 │ │
│ │                    │     key: "_construction_context"                                                                                                                         │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   s: "kEagerRuntime"                                                                                                                                 │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   control_ret {                                                                                                                                          │ │
│ │                    │     key: "Read_1/ReadVariableOp"                                                                                                                         │ │
│ │                    │     value: "Read_1/ReadVariableOp"                                                                                                                       │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   control_ret {                                                                                                                                          │ │
│ │                    │     key: "Read_1/DisableCopyOnRead"                                                                                                                      │ │
│ │                    │     value: "Read_1/DisableCopyOnRead"                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   control_ret {                                                                                                                                          │ │
│ │                    │     key: "Read/ReadVariableOp"                                                                                                                           │ │
│ │                    │     value: "Read/ReadVariableOp"                                                                                                                         │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   control_ret {                                                                                                                                          │ │
│ │                    │     key: "Read/DisableCopyOnRead"                                                                                                                        │ │
│ │                    │     value: "Read/DisableCopyOnRead"                                                                                                                      │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   control_ret {                                                                                                                                          │ │
│ │                    │     key: "MergeV2Checkpoints"                                                                                                                            │ │
│ │                    │     value: "MergeV2Checkpoints"                                                                                                                          │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   arg_attr {                                                                                                                                             │ │
│ │                    │     key: 0                                                                                                                                               │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   attr {                                                                                                                                             │ │
│ │                    │   │     key: "_user_specified_name"                                                                                                                      │ │
│ │                    │   │     value {                                                                                                                                          │ │
│ │                    │   │   │   s: "file_prefix"                                                                                                                               │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │   │   attr {                                                                                                                                             │ │
│ │                    │   │     key: "_output_shapes"                                                                                                                            │ │
│ │                    │   │     value {                                                                                                                                          │ │
│ │                    │   │   │   list {                                                                                                                                         │ │
│ │                    │   │   │     shape {                                                                                                                                      │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   arg_attr {                                                                                                                                             │ │
│ │                    │     key: 1                                                                                                                                               │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   attr {                                                                                                                                             │ │
│ │                    │   │     key: "_user_specified_name"                                                                                                                      │ │
│ │                    │   │     value {                                                                                                                                          │ │
│ │                    │   │   │   s: "Variable_1"                                                                                                                                │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   arg_attr {                                                                                                                                             │ │
│ │                    │     key: 2                                                                                                                                               │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   attr {                                                                                                                                             │ │
│ │                    │   │     key: "_user_specified_name"                                                                                                                      │ │
│ │                    │   │     value {                                                                                                                                          │ │
│ │                    │   │   │   s: "Variable"                                                                                                                                  │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   arg_attr {                                                                                                                                             │ │
│ │                    │     key: 3                                                                                                                                               │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   attr {                                                                                                                                             │ │
│ │                    │   │     key: "_user_specified_name"                                                                                                                      │ │
│ │                    │   │     value {                                                                                                                                          │ │
│ │                    │   │   │   s: "Const"                                                                                                                                     │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │   │   attr {                                                                                                                                             │ │
│ │                    │   │     key: "_output_shapes"                                                                                                                            │ │
│ │                    │   │     value {                                                                                                                                          │ │
│ │                    │   │   │   list {                                                                                                                                         │ │
│ │                    │   │   │     shape {                                                                                                                                      │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                      function {                                                                                                                                               │ │
│ │                    │   signature {                                                                                                                                            │ │
│ │                    │     name: "__inference_inner_34"                                                                                                                         │ │
│ │                    │     input_arg {                                                                                                                                          │ │
│ │                    │   │   name: "args_0"                                                                                                                                     │ │
│ │                    │   │   type: DT_HALF                                                                                                                                      │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     input_arg {                                                                                                                                          │ │
│ │                    │   │   name: "xlacallmodule_readvariableop_resource"                                                                                                      │ │
│ │                    │   │   type: DT_RESOURCE                                                                                                                                  │ │
│ │                    │   │   handle_data {                                                                                                                                      │ │
│ │                    │   │     dtype: DT_HALF                                                                                                                                   │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 3                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 3                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 8                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 8                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     input_arg {                                                                                                                                          │ │
│ │                    │   │   name: "xlacallmodule_readvariableop_1_resource"                                                                                                    │ │
│ │                    │   │   type: DT_RESOURCE                                                                                                                                  │ │
│ │                    │   │   handle_data {                                                                                                                                      │ │
│ │                    │   │     dtype: DT_HALF                                                                                                                                   │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 3                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     output_arg {                                                                                                                                         │ │
│ │                    │   │   name: "identity"                                                                                                                                   │ │
│ │                    │   │   type: DT_HALF                                                                                                                                      │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     is_stateful: true                                                                                                                                    │ │
│ │                    │     control_output: "XlaCallModule"                                                                                                                      │ │
│ │                    │     control_output: "XlaCallModule/ReadVariableOp"                                                                                                       │ │
│ │                    │     control_output: "XlaCallModule/ReadVariableOp_1"                                                                                                     │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "XlaCallModule/ReadVariableOp"                                                                                                                 │ │
│ │                    │     op: "ReadVariableOp"                                                                                                                                 │ │
│ │                    │     input: "xlacallmodule_readvariableop_resource"                                                                                                       │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "dtype"                                                                                                                                       │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_HALF                                                                                                                                    │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 8                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 8                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "XlaCallModule/ReadVariableOp_1"                                                                                                               │ │
│ │                    │     op: "ReadVariableOp"                                                                                                                                 │ │
│ │                    │     input: "xlacallmodule_readvariableop_1_resource"                                                                                                     │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "dtype"                                                                                                                                       │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_HALF                                                                                                                                    │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "XlaCallModule"                                                                                                                                │ │
│ │                    │     op: "XlaCallModule"                                                                                                                                  │ │
│ │                    │     input: "XlaCallModule/ReadVariableOp:value:0"                                                                                                        │ │
│ │                    │     input: "XlaCallModule/ReadVariableOp_1:value:0"                                                                                                      │ │
│ │                    │     input: "args_0"                                                                                                                                      │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "version"                                                                                                                                     │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     i: 5                                                                                                                                             │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "module"                                                                                                                                      │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     s:                                                                                                                                               │ │
│ │                    "ML\357R\rStableHLO_v1.13.0\000\001#\005\001\005\023\001\003\013\003\021\017\023\027\033\037#\'+\003\257o1\001#\007\017\013\013\017\013\023\013\017\017\0… │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 1                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "Tout"                                                                                                                                        │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   type: DT_HALF                                                                                                                                  │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "Tin"                                                                                                                                         │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   type: DT_HALF                                                                                                                                  │ │
│ │                    │   │   │   type: DT_HALF                                                                                                                                  │ │
│ │                    │   │   │   type: DT_HALF                                                                                                                                  │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "Sout"                                                                                                                                        │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 1                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "Identity"                                                                                                                                     │ │
│ │                    │     op: "Identity"                                                                                                                                       │ │
│ │                    │     input: "XlaCallModule:output:0"                                                                                                                      │ │
│ │                    │     input: "^NoOp"                                                                                                                                       │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │   │   shape {                                                                                                                                        │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 1                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │     dim {                                                                                                                                        │ │
│ │                    │   │   │   │   size: 3                                                                                                                                    │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "T"                                                                                                                                           │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     type: DT_HALF                                                                                                                                    │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   node_def {                                                                                                                                             │ │
│ │                    │     name: "NoOp"                                                                                                                                         │ │
│ │                    │     op: "NoOp"                                                                                                                                           │ │
│ │                    │     input: "^XlaCallModule"                                                                                                                              │ │
│ │                    │     input: "^XlaCallModule/ReadVariableOp"                                                                                                               │ │
│ │                    │     input: "^XlaCallModule/ReadVariableOp_1"                                                                                                             │ │
│ │                    │     attr {                                                                                                                                               │ │
│ │                    │   │   key: "_output_shapes"                                                                                                                              │ │
│ │                    │   │   value {                                                                                                                                            │ │
│ │                    │   │     list {                                                                                                                                           │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   ret {                                                                                                                                                  │ │
│ │                    │     key: "identity"                                                                                                                                      │ │
│ │                    │     value: "Identity:output:0"                                                                                                                           │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   attr {                                                                                                                                                 │ │
│ │                    │     key: "_input_shapes"                                                                                                                                 │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   list {                                                                                                                                             │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 1                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 3                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 8                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │   │   dim {                                                                                                                                          │ │
│ │                    │   │   │     size: 8                                                                                                                                      │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │     shape {                                                                                                                                          │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   attr {                                                                                                                                                 │ │
│ │                    │     key: "_construction_context"                                                                                                                         │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   s: "kEagerRuntime"                                                                                                                                 │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   control_ret {                                                                                                                                          │ │
│ │                    │     key: "XlaCallModule"                                                                                                                                 │ │
│ │                    │     value: "XlaCallModule"                                                                                                                               │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   control_ret {                                                                                                                                          │ │
│ │                    │     key: "XlaCallModule/ReadVariableOp"                                                                                                                  │ │
│ │                    │     value: "XlaCallModule/ReadVariableOp"                                                                                                                │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   control_ret {                                                                                                                                          │ │
│ │                    │     key: "XlaCallModule/ReadVariableOp_1"                                                                                                                │ │
│ │                    │     value: "XlaCallModule/ReadVariableOp_1"                                                                                                              │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   arg_attr {                                                                                                                                             │ │
│ │                    │     key: 0                                                                                                                                               │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   attr {                                                                                                                                             │ │
│ │                    │   │     key: "_user_specified_name"                                                                                                                      │ │
│ │                    │   │     value {                                                                                                                                          │ │
│ │                    │   │   │   s: "args_0"                                                                                                                                    │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │   │   attr {                                                                                                                                             │ │
│ │                    │   │     key: "_output_shapes"                                                                                                                            │ │
│ │                    │   │     value {                                                                                                                                          │ │
│ │                    │   │   │   list {                                                                                                                                         │ │
│ │                    │   │   │     shape {                                                                                                                                      │ │
│ │                    │   │   │   │   dim {                                                                                                                                      │ │
│ │                    │   │   │   │     size: 1                                                                                                                                  │ │
│ │                    │   │   │   │   }                                                                                                                                          │ │
│ │                    │   │   │   │   dim {                                                                                                                                      │ │
│ │                    │   │   │   │     size: 3                                                                                                                                  │ │
│ │                    │   │   │   │   }                                                                                                                                          │ │
│ │                    │   │   │   │   dim {                                                                                                                                      │ │
│ │                    │   │   │   │     size: 8                                                                                                                                  │ │
│ │                    │   │   │   │   }                                                                                                                                          │ │
│ │                    │   │   │   │   dim {                                                                                                                                      │ │
│ │                    │   │   │   │     size: 8                                                                                                                                  │ │
│ │                    │   │   │   │   }                                                                                                                                          │ │
│ │                    │   │   │     }                                                                                                                                            │ │
│ │                    │   │   │   }                                                                                                                                              │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   arg_attr {                                                                                                                                             │ │
│ │                    │     key: 1                                                                                                                                               │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   attr {                                                                                                                                             │ │
│ │                    │   │     key: "_user_specified_name"                                                                                                                      │ │
│ │                    │   │     value {                                                                                                                                          │ │
│ │                    │   │   │   s: "resource"                                                                                                                                  │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                    │   arg_attr {                                                                                                                                             │ │
│ │                    │     key: 2                                                                                                                                               │ │
│ │                    │     value {                                                                                                                                              │ │
│ │                    │   │   attr {                                                                                                                                             │ │
│ │                    │   │     key: "_user_specified_name"                                                                                                                      │ │
│ │                    │   │     value {                                                                                                                                          │ │
│ │                    │   │   │   s: "resource"                                                                                                                                  │ │
│ │                    │   │     }                                                                                                                                                │ │
│ │                    │   │   }                                                                                                                                                  │ │
│ │                    │     }                                                                                                                                                    │ │
│ │                    │   }                                                                                                                                                      │ │
│ │                      }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │                    versions {                                                                                                                                                 │ │
│ │                      producer: 2407                                                                                                                                           │ │
│ │                      min_consumer: 12                                                                                                                                         │ │
│ │                    }                                                                                                                                                          │ │
│ │       quant_mode = <tensorflow.lite.python.lite.QuantizationMode object at 0x70758f016f60>                                                                                    │ │
│ │             self = <tensorflow.lite.python.lite.TFLiteSavedModelConverterV2 object at 0x70751c76eba0>                                                                         │ │
│ ╰───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯ │
│                                                                                                                                                                                   │
│ /data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/tensorflow/lite/python/convert_phase.py:212 in wrapper                               │
│                                                                                                                                                                                   │
│   209 │   │   │   report_error(error_data)                                                                                                                                        │
│   210 │   │   else:                                                                                                                                                               │
│   211 │   │     report_error_message(str(converter_error))                                                                                                                        │
│ ❱ 212 │   │   raise converter_error from None  # Re-throws the exception.                                                                                                         │
│   213 │     except Exception as error:                                                                                                                                            │
│   214 │   │   report_error_message(str(error))                                                                                                                                    │
│   215 │   │   raise error from None  # Re-throws the exception.                                                                                                                   │
│                                                                                                                                                                                   │
│ ╭─────────────────────────────────────────────────────── locals ───────────────────────────────────────────────────────╮                                                          │
│ │       args = ()                                                                                                      │                                                          │
│ │ error_data = component: "CONVERT_TF_TO_TFLITE_MODEL"                                                                 │                                                          │
│ │              subcomponent: "LegalizeTFPass"                                                                          │                                                          │
│ │              error_code: UNKNOWN                                                                                     │                                                          │
│ │              error_message: "failed to legalize operation \'tfl.transpose\' that was explicitly marked illegal"      │                                                          │
│ │              operator {                                                                                              │                                                          │
│ │                name: "tfl.transpose"                                                                                 │                                                          │
│ │              }                                                                                                       │                                                          │
│ │              location {                                                                                              │                                                          │
│ │                type: NAMELOC                                                                                         │                                                          │
│ │                call {                                                                                                │                                                          │
│ │              │   name: "main.ChannelSum/torch.nn.modules.container.Sequential_model/torch.nn.modules.conv.Conv2d_0;" │                                                          │
│ │                }                                                                                                     │                                                          │
│ │              }                                                                                                       │                                                          │
│ │     kwargs = {                                                                                                       │                                                          │
│ │              │   'enable_tflite_resource_variables': True,                                                           │                                                          │
│ │              │   'input_format': 1,                                                                                  │                                                          │
│ │              │   'allow_custom_ops': False,                                                                          │                                                          │
│ │              │   'debug_info': ,                                                                                     │                                                          │
│ │              │   'target_ops': {<OpsSet.TFLITE_BUILTINS: 'TFLITE_BUILTINS'>},                                        │                                                          │
│ │              │   'select_user_tf_ops': set(),                                                                        │                                                          │
│ │              │   'supported_backends': None,                                                                         │                                                          │
│ │              │   'unfold_batchmatmul': False,                                                                        │                                                          │
│ │              │   'legalize_custom_tensor_list_ops': False,                                                           │                                                          │
│ │              │   'lower_tensor_list_ops': True,                                                                      │                                                          │
│ │              │   ... +39                                                                                             │                                                          │
│ │              }                                                                                                       │                                                          │
│ ╰──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯                                                          │
│                                                                                                                                                                                   │
│ /data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/tensorflow/lite/python/convert_phase.py:205 in wrapper                               │
│                                                                                                                                                                                   │
│   202 │   @functools.wraps(func)                                                                                                                                                  │
│   203 │   def wrapper(*args, **kwargs):                                                                                                                                           │
│   204 │     try:                                                                                                                                                                  │
│ ❱ 205 │   │   return func(*args, **kwargs)                                                                                                                                        │
│   206 │     except ConverterError as converter_error:                                                                                                                             │
│   207 │   │   if converter_error.errors:                                                                                                                                          │
│   208 │   │     for error_data in converter_error.errors:                                                                                                                         │
│                                                                                                                                                                                   │
│ ╭─────────────────────────────────────────────────────── locals ───────────────────────────────────────────────────────╮                                                          │
│ │       args = ()                                                                                                      │                                                          │
│ │ error_data = component: "CONVERT_TF_TO_TFLITE_MODEL"                                                                 │                                                          │
│ │              subcomponent: "LegalizeTFPass"                                                                          │                                                          │
│ │              error_code: UNKNOWN                                                                                     │                                                          │
│ │              error_message: "failed to legalize operation \'tfl.transpose\' that was explicitly marked illegal"      │                                                          │
│ │              operator {                                                                                              │                                                          │
│ │                name: "tfl.transpose"                                                                                 │                                                          │
│ │              }                                                                                                       │                                                          │
│ │              location {                                                                                              │                                                          │
│ │                type: NAMELOC                                                                                         │                                                          │
│ │                call {                                                                                                │                                                          │
│ │              │   name: "main.ChannelSum/torch.nn.modules.container.Sequential_model/torch.nn.modules.conv.Conv2d_0;" │                                                          │
│ │                }                                                                                                     │                                                          │
│ │              }                                                                                                       │                                                          │
│ │     kwargs = {                                                                                                       │                                                          │
│ │              │   'enable_tflite_resource_variables': True,                                                           │                                                          │
│ │              │   'input_format': 1,                                                                                  │                                                          │
│ │              │   'allow_custom_ops': False,                                                                          │                                                          │
│ │              │   'debug_info': ,                                                                                     │                                                          │
│ │              │   'target_ops': {<OpsSet.TFLITE_BUILTINS: 'TFLITE_BUILTINS'>},                                        │                                                          │
│ │              │   'select_user_tf_ops': set(),                                                                        │                                                          │
│ │              │   'supported_backends': None,                                                                         │                                                          │
│ │              │   'unfold_batchmatmul': False,                                                                        │                                                          │
│ │              │   'legalize_custom_tensor_list_ops': False,                                                           │                                                          │
│ │              │   'lower_tensor_list_ops': True,                                                                      │                                                          │
│ │              │   ... +39                                                                                             │                                                          │
│ │              }                                                                                                       │                                                          │
│ ╰──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯                                                          │
│                                                                                                                                                                                   │
│ /data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/tensorflow/lite/python/convert.py:893 in convert_saved_model                         │
│                                                                                                                                                                                   │
│    890   """Converts a SavedModel using TF Lite converter."""                                                                                                                     │
│    891   model_flags = build_model_flags(**kwargs)                                                                                                                                │
│    892   conversion_flags = build_conversion_flags(**kwargs)                                                                                                                      │
│ ❱  893   data = convert(                                                                                                                                                          │
│    894 │     model_flags,                                                                                                                                                         │
│    895 │     conversion_flags,                                                                                                                                                    │
│    896 │     input_data_str=None,                                                                                                                                                 │
│                                                                                                                                                                                   │
│ ╭────────────────────────────────────── locals ───────────────────────────────────────╮                                                                                           │
│ │ conversion_flags = input_format: TENSORFLOW_GRAPHDEF                                │                                                                                           │
│ │                    output_format: TFLITE                                            │                                                                                           │
│ │                    inference_type: FLOAT                                            │                                                                                           │
│ │                    reorder_across_fake_quant: false                                 │                                                                                           │
│ │                    allow_custom_ops: false                                          │                                                                                           │
│ │                    inference_input_type: FLOAT                                      │                                                                                           │
│ │                    drop_control_dependency: true                                    │                                                                                           │
│ │                    dump_graphviz_include_video: false                               │                                                                                           │
│ │                    post_training_quantize: false                                    │                                                                                           │
│ │                    quantize_to_float16: false                                       │                                                                                           │
│ │                    enable_tflite_resource_variables: true                           │                                                                                           │
│ │                    unfold_batchmatmul: false                                        │                                                                                           │
│ │                    lower_tensor_list_ops: true                                      │                                                                                           │
│ │                    allow_bfloat16: false                                            │                                                                                           │
│ │                    allow_all_select_tf_ops: false                                   │                                                                                           │
│ │                    unfold_large_splat_constant: false                               │                                                                                           │
│ │                    default_to_single_batch_in_tensor_list_ops: false                │                                                                                           │
│ │                    disable_per_channel_quantization: false                          │                                                                                           │
│ │                    enable_mlir_dynamic_range_quantizer: false                       │                                                                                           │
│ │                    disable_infer_tensor_range: false                                │                                                                                           │
│ │                    use_fake_quant_num_bits: false                                   │                                                                                           │
│ │                    enable_dynamic_update_slice: true                                │                                                                                           │
│ │                    preserve_assert_op: false                                        │                                                                                           │
│ │                    guarantee_all_funcs_one_use: false                               │                                                                                           │
│ │                    enable_mlir_variable_quantization: false                         │                                                                                           │
│ │                    disable_fuse_mul_and_fc: false                                   │                                                                                           │
│ │                    use_buffer_offset: false                                         │                                                                                           │
│ │                    legalize_custom_tensor_list_ops: false                           │                                                                                           │
│ │                    reduce_type_precision: false                                     │                                                                                           │
│ │                    disable_per_channel_quantization_for_dense_layers: false         │                                                                                           │
│ │                    enable_composite_direct_lowering: true                           │                                                                                           │
│ │                    model_origin_framework: PYTORCH                                  │                                                                                           │
│ │                    canonicalizing_inf_as_min_max_float: true                        │                                                                                           │
│ │                    serialize_debug_metadata: false                                  │                                                                                           │
│ │                    strict_qdq_mode: false                                           │                                                                                           │
│ │                    unsafe_fuse_dynamic_shaped_broadcast: false                      │                                                                                           │
│ │           kwargs = {                                                                │                                                                                           │
│ │                    │   'enable_tflite_resource_variables': True,                    │                                                                                           │
│ │                    │   'input_format': 1,                                           │                                                                                           │
│ │                    │   'allow_custom_ops': False,                                   │                                                                                           │
│ │                    │   'debug_info': ,                                              │                                                                                           │
│ │                    │   'target_ops': {                                              │                                                                                           │
│ │                    │   │   <OpsSet.TFLITE_BUILTINS: 'TFLITE_BUILTINS'>              │                                                                                           │
│ │                    │   },                                                           │                                                                                           │
│ │                    │   'select_user_tf_ops': set(),                                 │                                                                                           │
│ │                    │   'supported_backends': None,                                  │                                                                                           │
│ │                    │   'unfold_batchmatmul': False,                                 │                                                                                           │
│ │                    │   'legalize_custom_tensor_list_ops': False,                    │                                                                                           │
│ │                    │   'lower_tensor_list_ops': True,                               │                                                                                           │
│ │                    │   ... +39                                                      │                                                                                           │
│ │                    }                                                                │                                                                                           │
│ │      model_flags = allow_nonexistent_arrays: false                                  │                                                                                           │
│ │                    change_concat_input_ranges: false                                │                                                                                           │
│ │                    saved_model_dir: "/tmp/tmp3ykp77vo"                              │                                                                                           │
│ │                    saved_model_version: 1                                           │                                                                                           │
│ │                    saved_model_tags: "serve"                                        │                                                                                           │
│ │                    saved_model_exported_names: "serving_default"                    │                                                                                           │
│ ╰─────────────────────────────────────────────────────────────────────────────────────╯                                                                                           │
│                                                                                                                                                                                   │
│ /data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/tensorflow/lite/python/convert.py:348 in convert                                     │
│                                                                                                                                                                                   │
│    345 │   │   │   input_data_str,                                                                                                                                                │
│    346 │   │   │   debug_info_str,                                                                                                                                                │
│    347 │   │   )                                                                                                                                                                  │
│ ❱  348 │   raise converter_error                                                                                                                                                  │
│    349                                                                                                                                                                            │
│    350                                                                                                                                                                            │
│    351 def build_model_flags(                                                                                                                                                     │
│                                                                                                                                                                                   │
│ ╭─────────────────────────────────────────────────────────────────────────────────── locals ────────────────────────────────────────────────────────────────────────────────────╮ │
│ │ conversion_flags = input_format: TENSORFLOW_GRAPHDEF                                                                                                                          │ │
│ │                    output_format: TFLITE                                                                                                                                      │ │
│ │                    inference_type: FLOAT                                                                                                                                      │ │
│ │                    reorder_across_fake_quant: false                                                                                                                           │ │
│ │                    allow_custom_ops: false                                                                                                                                    │ │
│ │                    inference_input_type: FLOAT                                                                                                                                │ │
│ │                    drop_control_dependency: true                                                                                                                              │ │
│ │                    dump_graphviz_include_video: false                                                                                                                         │ │
│ │                    post_training_quantize: false                                                                                                                              │ │
│ │                    quantize_to_float16: false                                                                                                                                 │ │
│ │                    enable_tflite_resource_variables: true                                                                                                                     │ │
│ │                    unfold_batchmatmul: false                                                                                                                                  │ │
│ │                    lower_tensor_list_ops: true                                                                                                                                │ │
│ │                    allow_bfloat16: false                                                                                                                                      │ │
│ │                    allow_all_select_tf_ops: false                                                                                                                             │ │
│ │                    unfold_large_splat_constant: false                                                                                                                         │ │
│ │                    default_to_single_batch_in_tensor_list_ops: false                                                                                                          │ │
│ │                    disable_per_channel_quantization: false                                                                                                                    │ │
│ │                    enable_mlir_dynamic_range_quantizer: false                                                                                                                 │ │
│ │                    disable_infer_tensor_range: false                                                                                                                          │ │
│ │                    use_fake_quant_num_bits: false                                                                                                                             │ │
│ │                    enable_dynamic_update_slice: true                                                                                                                          │ │
│ │                    preserve_assert_op: false                                                                                                                                  │ │
│ │                    guarantee_all_funcs_one_use: false                                                                                                                         │ │
│ │                    enable_mlir_variable_quantization: false                                                                                                                   │ │
│ │                    disable_fuse_mul_and_fc: false                                                                                                                             │ │
│ │                    use_buffer_offset: false                                                                                                                                   │ │
│ │                    legalize_custom_tensor_list_ops: false                                                                                                                     │ │
│ │                    reduce_type_precision: false                                                                                                                               │ │
│ │                    disable_per_channel_quantization_for_dense_layers: false                                                                                                   │ │
│ │                    enable_composite_direct_lowering: true                                                                                                                     │ │
│ │                    model_origin_framework: PYTORCH                                                                                                                            │ │
│ │                    canonicalizing_inf_as_min_max_float: true                                                                                                                  │ │
│ │                    serialize_debug_metadata: false                                                                                                                            │ │
│ │                    strict_qdq_mode: false                                                                                                                                     │ │
│ │                    unsafe_fuse_dynamic_shaped_broadcast: false                                                                                                                │ │
│ │  converter_error = ConverterError("/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/main.py:18:0: error: failed to legalize operation 'tfl.transpose' that was       │ │
│ │                    explicitly marked illegal\n")                                                                                                                              │ │
│ │   debug_info_str = None                                                                                                                                                       │ │
│ │       error_data = component: "CONVERT_TF_TO_TFLITE_MODEL"                                                                                                                    │ │
│ │                    subcomponent: "LegalizeTFPass"                                                                                                                             │ │
│ │                    error_code: UNKNOWN                                                                                                                                        │ │
│ │                    error_message: "failed to legalize operation \'tfl.transpose\' that was explicitly marked illegal"                                                         │ │
│ │                    operator {                                                                                                                                                 │ │
│ │                      name: "tfl.transpose"                                                                                                                                    │ │
│ │                    }                                                                                                                                                          │ │
│ │                    location {                                                                                                                                                 │ │
│ │                      type: NAMELOC                                                                                                                                            │ │
│ │                      call {                                                                                                                                                   │ │
│ │                    │   name: "main.ChannelSum/torch.nn.modules.container.Sequential_model/torch.nn.modules.conv.Conv2d_0;"                                                    │ │
│ │                      }                                                                                                                                                        │ │
│ │                    }                                                                                                                                                          │ │
│ │   input_data_str = None                                                                                                                                                       │ │
│ │      model_flags = allow_nonexistent_arrays: false                                                                                                                            │ │
│ │                    change_concat_input_ranges: false                                                                                                                          │ │
│ │                    saved_model_dir: "/tmp/tmp3ykp77vo"                                                                                                                        │ │
│ │                    saved_model_version: 1                                                                                                                                     │ │
│ │                    saved_model_tags: "serve"                                                                                                                                  │ │
│ │                    saved_model_exported_names: "serving_default"                                                                                                              │ │
│ ╰───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯ │
╰───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
ConverterError: /data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/main.py:18:0: error: failed to legalize operation 'tfl.transpose' that was explicitly marked illegal

Done

Converting to tflite with quantization int8 --calibration-dataset Random
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/keras/src/export/tf2onnx_lib.py:8: FutureWarning: In the future `np.object` will be defined as the corresponding NumPy scalar.
  if not hasattr(np, "object"):
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/distributed/distributed_c10d.py:351: UserWarning: Device capability of jax unspecified, assuming `cpu` and `cuda`. Please specify it via the `devices` argument of `register_backend`.
  warnings.warn(
WARNING:2025-12-20 21:39:47,628:jax._src.xla_bridge:852: An NVIDIA GPU may be present on this machine, but a CUDA-enabled jaxlib is not installed. Falling back to cpu.
[WARNING  21:39:47] An NVIDIA GPU may be present on this machine, but a CUDA-enabled jaxlib is not installed. Falling back to cpu.
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:283: UserWarning: 
    Found GPU0 NVIDIA GeForce GTX 1050 Ti which is of cuda capability 6.1.
    Minimum and Maximum cuda capability supported by this version of PyTorch is
    (7.0) - (12.0)
    
  warnings.warn(
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:304: UserWarning: 
    Please install PyTorch with a following CUDA
    configurations:  12.6 following instructions at
    https://pytorch.org/get-started/locally/
    
  warnings.warn(matched_cuda_warn.format(matched_arches))
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:326: UserWarning: 
NVIDIA GeForce GTX 1050 Ti with CUDA capability sm_61 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_70 sm_75 sm_80 sm_86 sm_90 sm_100 sm_120.
If you want to use the NVIDIA GeForce GTX 1050 Ti GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

  warnings.warn(
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
W0000 00:00:1766266790.433382  109194 tf_tfl_flatbuffer_helpers.cc:364] Ignored output_format.
W0000 00:00:1766266790.433431  109194 tf_tfl_flatbuffer_helpers.cc:367] Ignored drop_control_dependency.
I0000 00:00:1766266790.434114  109194 reader.cc:83] Reading SavedModel from: /tmp/tmpoz35ykj7
I0000 00:00:1766266790.434407  109194 reader.cc:52] Reading meta graph with tags { serve }
I0000 00:00:1766266790.434420  109194 reader.cc:147] Reading SavedModel debug info (if present) from: /tmp/tmpoz35ykj7
I0000 00:00:1766266790.436157  109194 mlir_graph_optimization_pass.cc:437] MLIR V1 optimization pass is not enabled
I0000 00:00:1766266790.436465  109194 loader.cc:236] Restoring SavedModel bundle.
I0000 00:00:1766266790.447081  109194 loader.cc:220] Running initialization op on SavedModel bundle at path: /tmp/tmpoz35ykj7
I0000 00:00:1766266790.451422  109194 loader.cc:471] SavedModel load for tags { serve }; Status: success: OK. Took 17319 microseconds.
I0000 00:00:1766266790.460660  109194 dump_mlir_util.cc:269] disabling MLIR crash reproducer, set env var `MLIR_CRASH_REPRODUCER_DIRECTORY` to enable.
I0000 00:00:1766266790.517685  109194 flatbuffer_export.cc:4160] Estimated count of arithmetic ops: 1155  ops, equivalently 577  MACs
fully_quantize: 0, inference_type: 6, input_inference_type: FLOAT32, output_inference_type: FLOAT32
I0000 00:00:1766266790.554315  109194 flatbuffer_export.cc:4160] Estimated count of arithmetic ops: 1155  ops, equivalently 577  MACs
W0000 00:00:1766266790.554357  109194 flatbuffer_export.cc:3715] Skipping runtime version metadata in the model. This will be generated by the exporter.
Done

Converting to ncnn with quantization float32
pnnxparam = build/model.pnnx.param
pnnxbin = build/model.pnnx.bin
pnnxpy = build/model_pnnx.py
pnnxonnx = build/model.pnnx.onnx
ncnnparam = build/model.ncnn.param
ncnnbin = build/model.ncnn.bin
ncnnpy = build/model_ncnn.py
fp16 = 0
optlevel = 2
device = cpu
inputshape = [1,3,8,8]f32
inputshape2 = 
customop = 
moduleop = 
get inputshape from traced inputs
inputshape = [1,3,8,8]f32
############# pass_level0
inline module = torch.nn.modules.flatten.Flatten
inline module = torch.nn.modules.flatten.Flatten

----------------

[W1220 21:39:54.715783305 NNPACK.cpp:56] Could not initialize NNPACK! Reason: Unsupported hardware.
############# pass_level1
############# pass_level2
############# pass_level3
############# pass_level4
############# pass_level5
############# pass_ncnn
Done

Converting to ncnn with quantization float16
pnnxparam = build/model.pnnx.param
pnnxbin = build/model.pnnx.bin
pnnxpy = build/model_pnnx.py
pnnxonnx = build/model.pnnx.onnx
ncnnparam = build/model.ncnn.param
ncnnbin = build/model.ncnn.bin
ncnnpy = build/model_ncnn.py
fp16 = 1
optlevel = 2
device = cpu
inputshape = [1,3,8,8]f32
inputshape2 = 
customop = 
moduleop = 
get inputshape from traced inputs
inputshape = [1,3,8,8]f32
############# pass_level0
inline module = torch.nn.modules.flatten.Flatten
inline module = torch.nn.modules.flatten.Flatten

----------------

[W1220 21:39:59.138144991 NNPACK.cpp:56] Could not initialize NNPACK! Reason: Unsupported hardware.
############# pass_level1
############# pass_level2
############# pass_level3
############# pass_level4
############# pass_level5
############# pass_ncnn
Done

Converting to ncnn with quantization int8 --calibration-dataset Random
build/ncnn.int8.all.ncnn
Done

Converting to executorch --hw-acceleration XNNPACK with quantization float32
[WARNING  21:40:06] Skipping import of cpp extensions due to incompatible torch version 2.9.1+cu128 for torchao version 0.14.0         Please see GitHub issue #2919 for more info
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:283: UserWarning: 
    Found GPU0 NVIDIA GeForce GTX 1050 Ti which is of cuda capability 6.1.
    Minimum and Maximum cuda capability supported by this version of PyTorch is
    (7.0) - (12.0)
    
  warnings.warn(
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:304: UserWarning: 
    Please install PyTorch with a following CUDA
    configurations:  12.6 following instructions at
    https://pytorch.org/get-started/locally/
    
  warnings.warn(matched_cuda_warn.format(matched_arches))
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:326: UserWarning: 
NVIDIA GeForce GTX 1050 Ti with CUDA capability sm_61 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_70 sm_75 sm_80 sm_86 sm_90 sm_100 sm_120.
If you want to use the NVIDIA GeForce GTX 1050 Ti GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

  warnings.warn(
Done

Converting to executorch --hw-acceleration XNNPACK with quantization float16
[WARNING  21:40:19] Skipping import of cpp extensions due to incompatible torch version 2.9.1+cu128 for torchao version 0.14.0         Please see GitHub issue #2919 for more info
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:283: UserWarning: 
    Found GPU0 NVIDIA GeForce GTX 1050 Ti which is of cuda capability 6.1.
    Minimum and Maximum cuda capability supported by this version of PyTorch is
    (7.0) - (12.0)
    
  warnings.warn(
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:304: UserWarning: 
    Please install PyTorch with a following CUDA
    configurations:  12.6 following instructions at
    https://pytorch.org/get-started/locally/
    
  warnings.warn(matched_cuda_warn.format(matched_arches))
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:326: UserWarning: 
NVIDIA GeForce GTX 1050 Ti with CUDA capability sm_61 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_70 sm_75 sm_80 sm_86 sm_90 sm_100 sm_120.
If you want to use the NVIDIA GeForce GTX 1050 Ti GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

  warnings.warn(
[W1220 21:40:29.819756465 NNPACK.cpp:56] Could not initialize NNPACK! Reason: Unsupported hardware.
Done

Converting to executorch --hw-acceleration XNNPACK with quantization int8 --calibration-dataset Random
[WARNING  21:40:34] Skipping import of cpp extensions due to incompatible torch version 2.9.1+cu128 for torchao version 0.14.0         Please see GitHub issue #2919 for more info
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:283: UserWarning: 
    Found GPU0 NVIDIA GeForce GTX 1050 Ti which is of cuda capability 6.1.
    Minimum and Maximum cuda capability supported by this version of PyTorch is
    (7.0) - (12.0)
    
  warnings.warn(
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:304: UserWarning: 
    Please install PyTorch with a following CUDA
    configurations:  12.6 following instructions at
    https://pytorch.org/get-started/locally/
    
  warnings.warn(matched_cuda_warn.format(matched_arches))
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:326: UserWarning: 
NVIDIA GeForce GTX 1050 Ti with CUDA capability sm_61 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_70 sm_75 sm_80 sm_86 sm_90 sm_100 sm_120.
If you want to use the NVIDIA GeForce GTX 1050 Ti GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

  warnings.warn(
Done

Converting to executorch --hw-acceleration Vulkan with quantization float32
[WARNING  21:40:52] Skipping import of cpp extensions due to incompatible torch version 2.9.1+cu128 for torchao version 0.14.0         Please see GitHub issue #2919 for more info
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:283: UserWarning: 
    Found GPU0 NVIDIA GeForce GTX 1050 Ti which is of cuda capability 6.1.
    Minimum and Maximum cuda capability supported by this version of PyTorch is
    (7.0) - (12.0)
    
  warnings.warn(
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:304: UserWarning: 
    Please install PyTorch with a following CUDA
    configurations:  12.6 following instructions at
    https://pytorch.org/get-started/locally/
    
  warnings.warn(matched_cuda_warn.format(matched_arches))
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:326: UserWarning: 
NVIDIA GeForce GTX 1050 Ti with CUDA capability sm_61 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_70 sm_75 sm_80 sm_86 sm_90 sm_100 sm_120.
If you want to use the NVIDIA GeForce GTX 1050 Ti GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

  warnings.warn(
[INFO     21:40:57] Found 1 Vulkan subgraphs to be partitioned.
[INFO     21:40:58] Operators included in this Vulkan partition: 
[INFO     21:40:58]     aten.convolution.default
[INFO     21:40:58]     aten.view_copy.default
Done

Converting to executorch --hw-acceleration Vulkan with quantization float16
[WARNING  21:41:02] Skipping import of cpp extensions due to incompatible torch version 2.9.1+cu128 for torchao version 0.14.0         Please see GitHub issue #2919 for more info
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:283: UserWarning: 
    Found GPU0 NVIDIA GeForce GTX 1050 Ti which is of cuda capability 6.1.
    Minimum and Maximum cuda capability supported by this version of PyTorch is
    (7.0) - (12.0)
    
  warnings.warn(
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:304: UserWarning: 
    Please install PyTorch with a following CUDA
    configurations:  12.6 following instructions at
    https://pytorch.org/get-started/locally/
    
  warnings.warn(matched_cuda_warn.format(matched_arches))
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:326: UserWarning: 
NVIDIA GeForce GTX 1050 Ti with CUDA capability sm_61 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_70 sm_75 sm_80 sm_86 sm_90 sm_100 sm_120.
If you want to use the NVIDIA GeForce GTX 1050 Ti GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

  warnings.warn(
[INFO     21:41:07] Found 1 Vulkan subgraphs to be partitioned.
[INFO     21:41:08] Operators included in this Vulkan partition: 
[INFO     21:41:08]     aten.convolution.default
[INFO     21:41:08]     aten.view_copy.default
[W1220 21:41:08.686654992 NNPACK.cpp:56] Could not initialize NNPACK! Reason: Unsupported hardware.
Done

Converting to executorch --hw-acceleration Vulkan with quantization int8 --calibration-dataset Random
[WARNING  21:41:12] Skipping import of cpp extensions due to incompatible torch version 2.9.1+cu128 for torchao version 0.14.0         Please see GitHub issue #2919 for more info
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:283: UserWarning: 
    Found GPU0 NVIDIA GeForce GTX 1050 Ti which is of cuda capability 6.1.
    Minimum and Maximum cuda capability supported by this version of PyTorch is
    (7.0) - (12.0)
    
  warnings.warn(
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:304: UserWarning: 
    Please install PyTorch with a following CUDA
    configurations:  12.6 following instructions at
    https://pytorch.org/get-started/locally/
    
  warnings.warn(matched_cuda_warn.format(matched_arches))
/data/ai-parade/toolkit/tests/end2end/fixtures/NHWCvsNCHW/.venv/lib/python3.12/site-packages/torch/cuda/__init__.py:326: UserWarning: 
NVIDIA GeForce GTX 1050 Ti with CUDA capability sm_61 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_70 sm_75 sm_80 sm_86 sm_90 sm_100 sm_120.
If you want to use the NVIDIA GeForce GTX 1050 Ti GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

  warnings.warn(
[INFO     21:41:17] Found 1 Vulkan subgraphs to be partitioned.
[INFO     21:41:18] Operators included in this Vulkan partition: 
[INFO     21:41:18]     aten.view_copy.default
[INFO     21:41:18]     aten.convolution.default
Done

