stable-diffusion-tensorflow
stable-diffusion-tensorflow copied to clipboard
Published
20 hours ago •
divamgupta
divamgupta
TF Lite convert error
Hi, Leaving this here in case someone is also trying to convert to a TF lite model.
From the keras_cv documentation:
!pip install --upgrade keras-cv
!pip install --upgrade tensorflow
Load the model:
import time
import keras_cv
from tensorflow import keras
import matplotlib.pyplot as plt
model = keras_cv.models.StableDiffusion(img_width=512, img_height=512)
# Convert the model.
converter = tf.lite.TFLiteConverter.from_keras_model(model.diffusion_model)
tflite_model = converter.convert()
It seems a similar error occurs when trying to save the model.
model.diffusion_model.save(save_dir)
The error in conversion:
[/usr/local/lib/python3.7/dist-packages/keras_cv/models/generative/stable_diffusion/__internal__/layers/group_normalization.py](https://localhost:8080/#) in _create_broadcast_shape(self, input_shape)
85
86 def _create_broadcast_shape(self, input_shape):
---> 87 broadcast_shape = [1] * len(input_shape)
88 broadcast_shape[self.axis] = input_shape[self.axis] // self.groups
TypeError: Exception encountered when calling layer 'group_normalization_60' (type GroupNormalization).
len is not well defined for a symbolic Tensor (Shape:0). Please call `x.shape` rather than `len(x)` for shape information.
Call arguments received by layer 'group_normalization_60' (type GroupNormalization):
• args=('tf.Tensor(shape=(None, 64, 64, 320), dtype=float32)',)
• kwargs=<class 'inspect._empty'>
It appears that the problem is related to KerasCV's implementation because they are using their own implementation of the GroupNormalization layer, which is causing this error, whereas this repo's implementation uses the GroupNormalization layer from tensorflow-addons. I was able to save the diffusion model without errors by using this repo.
!pip install git+https://github.com/divamgupta/stable-diffusion-tensorflow --upgrade --quiet
!pip install tensorflow tensorflow_addons ftfy --upgrade --quiet
from stable_diffusion_tf.stable_diffusion import StableDiffusion
model = StableDiffusion(
img_height=512,
img_width=512,
)
model.diffusion_model.save('/content/saved_model')
**EDIT: While saving is possible, converting to TFLite appears to be impossible. It appears that the protobuf has a size limit of 2GB. Because the saved model exceeds 3GB, the conversion returns the following error:
[/usr/local/lib/python3.7/dist-packages/tensorflow/lite/python/lite.py] in wrapper(self, *args, **kwargs)
931 def wrapper(self, *args, **kwargs):
932 # pylint: disable=protected-access
--> 933 return self._convert_and_export_metrics(convert_func, *args, **kwargs)
934 # pylint: enable=protected-access
935
[/usr/local/lib/python3.7/dist-packages/tensorflow/lite/python/lite.py] in _convert_and_export_metrics(self, convert_func, *args, **kwargs)
909 self._save_conversion_params_metric()
910 start_time = time.process_time()
--> 911 result = convert_func(self, *args, **kwargs)
912 elapsed_time_ms = (time.process_time() - start_time) * 1000
913 if result:
[/usr/local/lib/python3.7/dist-packages/tensorflow/lite/python/lite.py] in convert(self)
1345
1346 graph_def, input_tensors, output_tensors, frozen_func = (
-> 1347 self._freeze_keras_model())
1348
1349 graph_def = self._optimize_tf_model(graph_def, input_tensors,
[/usr/local/lib/python3.7/dist-packages/tensorflow/lite/python/convert_phase.py] in wrapper(*args, **kwargs)
213 except Exception as error:
214 report_error_message(str(error))
--> 215 raise error from None # Re-throws the exception.
216
217 return wrapper
[/usr/local/lib/python3.7/dist-packages/tensorflow/lite/python/convert_phase.py] in wrapper(*args, **kwargs)
203 def wrapper(*args, **kwargs):
204 try:
--> 205 return func(*args, **kwargs)
206 except ConverterError as converter_error:
207 if converter_error.errors:
[/usr/local/lib/python3.7/dist-packages/tensorflow/lite/python/lite.py] in _freeze_keras_model(self)
1302 frozen_func, graph_def = (
1303 _convert_to_constants.convert_variables_to_constants_v2_as_graph(
-> 1304 self._funcs[0], lower_control_flow=False))
1305
1306 input_tensors = [
[/usr/local/lib/python3.7/dist-packages/tensorflow/python/framework/convert_to_constants.py] in convert_variables_to_constants_v2_as_graph(func, lower_control_flow, aggressive_inlining)
1249
1250 frozen_func = _construct_concrete_function(func, output_graph_def,
-> 1251 converted_input_indices)
1252 return frozen_func, output_graph_def
1253
[/usr/local/lib/python3.7/dist-packages/tensorflow/python/framework/convert_to_constants.py] in _construct_concrete_function(func, output_graph_def, converted_input_indices)
1094 new_func = wrap_function.function_from_graph_def(output_graph_def,
1095 new_input_names,
-> 1096 new_output_names)
1097
1098 # Manually propagate shape for input tensors where the shape is not correctly
[/usr/local/lib/python3.7/dist-packages/tensorflow/python/eager/wrap_function.py] in function_from_graph_def(graph_def, inputs, outputs, captures)
653 graph.add_capture(captures[c], graph.get_tensor_by_name(str(c) + ":0"))
654
--> 655 wrapped_import = wrap_function(_imports_graph_def, [])
656 import_graph = wrapped_import.graph
657 return wrapped_import.prune(
[/usr/local/lib/python3.7/dist-packages/tensorflow/python/eager/wrap_function.py] in wrap_function(fn, signature, name)
624 signature=signature,
625 add_control_dependencies=False,
--> 626 collections={}),
627 variable_holder=holder,
628 signature=signature)
[/usr/local/lib/python3.7/dist-packages/tensorflow/python/framework/func_graph.py] in func_graph_from_py_func(name, python_func, args, kwargs, signature, func_graph, autograph, autograph_options, add_control_dependencies, arg_names, op_return_value, collections, capture_by_value, acd_record_initial_resource_uses)
1281 _, original_func = tf_decorator.unwrap(python_func)
1282
-> 1283 func_outputs = python_func(*func_args, **func_kwargs)
1284
1285 # invariant: `func_outputs` contains only Tensors, CompositeTensors,
[/usr/local/lib/python3.7/dist-packages/tensorflow/python/eager/wrap_function.py] in __call__(self, *args, **kwargs)
81
82 def __call__(self, *args, **kwargs):
---> 83 return self.call_with_variable_creator_scope(self._fn)(*args, **kwargs)
84
85 def call_with_variable_creator_scope(self, fn):
[/usr/local/lib/python3.7/dist-packages/tensorflow/python/eager/wrap_function.py] in wrapped(*args, **kwargs)
87 def wrapped(*args, **kwargs):
88 with variable_scope.variable_creator_scope(self.variable_creator_scope):
---> 89 return fn(*args, **kwargs)
90
91 return wrapped
[/usr/local/lib/python3.7/dist-packages/tensorflow/python/eager/wrap_function.py] in _imports_graph_def()
647
648 def _imports_graph_def():
--> 649 importer.import_graph_def(graph_def, name="")
650 graph = ops.get_default_graph()
651 if captures is not None:
[/usr/local/lib/python3.7/dist-packages/tensorflow/python/util/deprecation.py] in new_func(*args, **kwargs)
559 'in a future version' if date is None else ('after %s' % date),
560 instructions)
--> 561 return func(*args, **kwargs)
562
563 doc = _add_deprecated_arg_notice_to_docstring(
[/usr/local/lib/python3.7/dist-packages/tensorflow/python/framework/importer.py] in import_graph_def(***failed resolving arguments***)
406 return_elements=return_elements,
407 name=name,
--> 408 producer_op_list=producer_op_list)
409
410
[/usr/local/lib/python3.7/dist-packages/tensorflow/python/framework/importer.py] in _import_graph_def_internal(graph_def, input_map, return_elements, validate_colocation_constraints, name, producer_op_list)
495 # _ProcessNewOps.
496 with graph._mutation_lock(): # pylint: disable=protected-access
--> 497 with c_api_util.tf_buffer(graph_def.SerializeToString()) as serialized:
498 try:
499 with graph._c_graph.get() as c_graph: # pylint: disable=protected-access
ValueError: Message tensorflow.GraphDef exceeds maximum protobuf size of 2GB: 3440790027
Thank you @costiash Is this a stopper? Seems some were going away with the hard limit with checkpoints?
Another one for using pb model?
https://github.com/ultralytics/yolov5/issues/6262
I tried exporting the saved model. Had a different error than yours
---------------------------------------------------------------------------
ConverterError Traceback (most recent call last)
<ipython-input-41-96c734e2e017> in <module>
2
3 converter = tf.lite.TFLiteConverter.from_saved_model('./saved_model') # path to the SavedModel directory
----> 4 tflite_model = converter.convert()
/opt/conda/envs/rapids/lib/python3.8/site-packages/tensorflow/lite/python/lite.py in wrapper(self, *args, **kwargs)
931 def wrapper(self, *args, **kwargs):
932 # pylint: disable=protected-access
--> 933 return self._convert_and_export_metrics(convert_func, *args, **kwargs)
934 # pylint: enable=protected-access
935
/opt/conda/envs/rapids/lib/python3.8/site-packages/tensorflow/lite/python/lite.py in _convert_and_export_metrics(self, convert_func, *args, **kwargs)
909 self._save_conversion_params_metric()
910 start_time = time.process_time()
--> 911 result = convert_func(self, *args, **kwargs)
912 elapsed_time_ms = (time.process_time() - start_time) * 1000
913 if result:
/opt/conda/envs/rapids/lib/python3.8/site-packages/tensorflow/lite/python/lite.py in convert(self)
1214 graph_def)
1215
-> 1216 return self._convert_from_saved_model(graph_def)
1217
1218
/opt/conda/envs/rapids/lib/python3.8/site-packages/tensorflow/lite/python/lite.py in _convert_from_saved_model(self, graph_def)
1097 converter_kwargs.update(quant_mode.converter_flags())
1098
-> 1099 result = _convert_saved_model(**converter_kwargs)
1100 return self._optimize_tflite_model(
1101 result, quant_mode, quant_io=self.experimental_new_quantizer)
/opt/conda/envs/rapids/lib/python3.8/site-packages/tensorflow/lite/python/convert_phase.py in wrapper(*args, **kwargs)
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))
/opt/conda/envs/rapids/lib/python3.8/site-packages/tensorflow/lite/python/convert_phase.py in wrapper(*args, **kwargs)
203 def wrapper(*args, **kwargs):
204 try:
--> 205 return func(*args, **kwargs)
206 except ConverterError as converter_error:
207 if converter_error.errors:
/opt/conda/envs/rapids/lib/python3.8/site-packages/tensorflow/lite/python/convert.py in convert_saved_model(**kwargs)
806 model_flags = build_model_flags(**kwargs)
807 conversion_flags = build_conversion_flags(**kwargs)
--> 808 data = convert(
809 model_flags.SerializeToString(),
810 conversion_flags.SerializeToString(),
/opt/conda/envs/rapids/lib/python3.8/site-packages/tensorflow/lite/python/convert.py in convert(model_flags_str, conversion_flags_str, input_data_str, debug_info_str, enable_mlir_converter)
308 for error_data in _metrics_wrapper.retrieve_collected_errors():
309 converter_error.append_error(error_data)
--> 310 raise converter_error
311
312 return _run_deprecated_conversion_binary(model_flags_str,
ConverterError: <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_110/padded_conv2d_691/conv2d_691/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_110/padded_conv2d_691/conv2d_691/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_110/padded_conv2d_692/conv2d_692/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_110/padded_conv2d_692/conv2d_692/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_80/padded_conv2d_693/conv2d_693/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_80/padded_conv2d_693/conv2d_693/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_111/padded_conv2d_695/conv2d_695/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_111/padded_conv2d_695/conv2d_695/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_111/padded_conv2d_696/conv2d_696/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_111/padded_conv2d_696/conv2d_696/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_81/padded_conv2d_697/conv2d_697/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_81/padded_conv2d_697/conv2d_697/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_112/padded_conv2d_700/conv2d_700/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_112/padded_conv2d_700/conv2d_700/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_112/padded_conv2d_701/conv2d_701/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_112/padded_conv2d_701/conv2d_701/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_82/padded_conv2d_703/conv2d_703/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_82/padded_conv2d_703/conv2d_703/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_113/padded_conv2d_705/conv2d_705/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_113/padded_conv2d_705/conv2d_705/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_113/padded_conv2d_706/conv2d_706/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_113/padded_conv2d_706/conv2d_706/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_83/padded_conv2d_707/conv2d_707/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_83/padded_conv2d_707/conv2d_707/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_114/padded_conv2d_710/conv2d_710/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_114/padded_conv2d_710/conv2d_710/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_114/padded_conv2d_711/conv2d_711/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_114/padded_conv2d_711/conv2d_711/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_84/padded_conv2d_713/conv2d_713/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_84/padded_conv2d_713/conv2d_713/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_115/padded_conv2d_715/conv2d_715/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_115/padded_conv2d_715/conv2d_715/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_115/padded_conv2d_716/conv2d_716/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_115/padded_conv2d_716/conv2d_716/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_85/padded_conv2d_717/conv2d_717/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_85/padded_conv2d_717/conv2d_717/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_116/padded_conv2d_720/conv2d_720/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_116/padded_conv2d_720/conv2d_720/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_116/padded_conv2d_721/conv2d_721/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_116/padded_conv2d_721/conv2d_721/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_117/padded_conv2d_722/conv2d_722/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_117/padded_conv2d_722/conv2d_722/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_117/padded_conv2d_723/conv2d_723/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_117/padded_conv2d_723/conv2d_723/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_118/padded_conv2d_724/conv2d_724/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_118/padded_conv2d_724/conv2d_724/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_118/padded_conv2d_725/conv2d_725/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_118/padded_conv2d_725/conv2d_725/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_86/padded_conv2d_726/conv2d_726/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_86/padded_conv2d_726/conv2d_726/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_119/padded_conv2d_728/conv2d_728/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_119/padded_conv2d_728/conv2d_728/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_119/padded_conv2d_729/conv2d_729/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_119/padded_conv2d_729/conv2d_729/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_120/padded_conv2d_730/conv2d_730/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_120/padded_conv2d_730/conv2d_730/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_120/padded_conv2d_731/conv2d_731/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_120/padded_conv2d_731/conv2d_731/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_121/padded_conv2d_733/conv2d_733/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_121/padded_conv2d_733/conv2d_733/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_121/padded_conv2d_734/conv2d_734/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_121/padded_conv2d_734/conv2d_734/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_122/padded_conv2d_736/conv2d_736/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_122/padded_conv2d_736/conv2d_736/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_122/padded_conv2d_737/conv2d_737/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_122/padded_conv2d_737/conv2d_737/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_123/padded_conv2d_740/conv2d_740/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_123/padded_conv2d_740/conv2d_740/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_123/padded_conv2d_741/conv2d_741/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_123/padded_conv2d_741/conv2d_741/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_87/padded_conv2d_743/conv2d_743/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_87/padded_conv2d_743/conv2d_743/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_124/padded_conv2d_745/conv2d_745/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_124/padded_conv2d_745/conv2d_745/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_124/padded_conv2d_746/conv2d_746/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_124/padded_conv2d_746/conv2d_746/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_88/padded_conv2d_748/conv2d_748/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_88/padded_conv2d_748/conv2d_748/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_125/padded_conv2d_750/conv2d_750/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_125/padded_conv2d_750/conv2d_750/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_125/padded_conv2d_751/conv2d_751/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_125/padded_conv2d_751/conv2d_751/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_89/padded_conv2d_753/conv2d_753/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_89/padded_conv2d_753/conv2d_753/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_126/padded_conv2d_756/conv2d_756/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_126/padded_conv2d_756/conv2d_756/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_126/padded_conv2d_757/conv2d_757/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_126/padded_conv2d_757/conv2d_757/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_90/padded_conv2d_759/conv2d_759/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_90/padded_conv2d_759/conv2d_759/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_127/padded_conv2d_761/conv2d_761/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_127/padded_conv2d_761/conv2d_761/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_127/padded_conv2d_762/conv2d_762/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_127/padded_conv2d_762/conv2d_762/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_91/padded_conv2d_764/conv2d_764/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_91/padded_conv2d_764/conv2d_764/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_128/padded_conv2d_766/conv2d_766/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_128/padded_conv2d_766/conv2d_766/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_128/padded_conv2d_767/conv2d_767/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_128/padded_conv2d_767/conv2d_767/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_92/padded_conv2d_769/conv2d_769/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_92/padded_conv2d_769/conv2d_769/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_129/padded_conv2d_772/conv2d_772/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_129/padded_conv2d_772/conv2d_772/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_129/padded_conv2d_773/conv2d_773/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_129/padded_conv2d_773/conv2d_773/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_93/padded_conv2d_775/conv2d_775/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_93/padded_conv2d_775/conv2d_775/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_130/padded_conv2d_777/conv2d_777/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_130/padded_conv2d_777/conv2d_777/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_130/padded_conv2d_778/conv2d_778/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_130/padded_conv2d_778/conv2d_778/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_94/padded_conv2d_780/conv2d_780/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_94/padded_conv2d_780/conv2d_780/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_131/padded_conv2d_782/conv2d_782/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_131/padded_conv2d_782/conv2d_782/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_131/padded_conv2d_783/conv2d_783/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_131/padded_conv2d_783/conv2d_783/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_95/padded_conv2d_785/conv2d_785/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_95/padded_conv2d_785/conv2d_785/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/padded_conv2d_787/conv2d_787/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op
<unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from
<unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/padded_conv2d_787/conv2d_787/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS
<unknown>:0: error: failed while converting: 'main':
Some ops are not supported by the native TFLite runtime, you can enable TF kernels fallback using TF Select. See instructions: https://www.tensorflow.org/lite/guide/ops_select
TF Select ops: Conv2D
Details:
tf.Conv2D(tensor<?x?x?x?xf32>, tensor<1x1x1280x1280xf32>) -> (tensor<?x?x?x1280xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true}
tf.Conv2D(tensor<?x?x?x?xf32>, tensor<1x1x320x320xf32>) -> (tensor<?x?x?x320xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true}
tf.Conv2D(tensor<?x?x?x?xf32>, tensor<1x1x640x640xf32>) -> (tensor<?x?x?x640xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true}
tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x1280x1280xf32>) -> (tensor<?x?x?x1280xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true}
tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x1280x640xf32>) -> (tensor<?x?x?x640xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true}
tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x1920x1280xf32>) -> (tensor<?x?x?x1280xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true}
tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x1920x640xf32>) -> (tensor<?x?x?x640xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true}
tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x2560x1280xf32>) -> (tensor<?x?x?x1280xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true}
tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x320x320xf32>) -> (tensor<?x?x?x320xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true}
tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x320x4xf32>) -> (tensor<?x?x?x4xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true}
tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x320x640xf32>) -> (tensor<?x?x?x640xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true}
tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x640x1280xf32>) -> (tensor<?x?x?x1280xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true}
tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x640x320xf32>) -> (tensor<?x?x?x320xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true}
tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x640x640xf32>) -> (tensor<?x?x?x640xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true}
tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x960x320xf32>) -> (tensor<?x?x?x320xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true}
tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x960x640xf32>) -> (tensor<?x?x?x640xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true}
I also tried the following:
- generate an .h5 model from diffusion
!pip install git+https://github.com/divamgupta/stable-diffusion-tensorflow --upgrade
!pip install tensorflow tensorflow_addons ftfy --upgrade
model = st.Text2Image(
img_height=512,
img_width=512,
)
model.diffusion_model.save('saved_model.h5', save_format='h5')
- Then it turned out the new tf version does not load h5 models. so tried an earlier version:
!pip uninstall tensorflow -y
!pip install tensorflow==2.1.0
However when loading with the earlier version
import tensorflow as tf
model2 = tf.keras.models.load_model('stable_diffusion.h5')
converter = tf.lite.TFLiteConverter.from_keras_model(model2)
tflite_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_model)
looks like it was not able to load it either:
AttributeError Traceback (most recent call last)
[<ipython-input-5-72d9a214a713>](https://localhost:8080/#) in <module>
1 import tensorflow as tf
2
----> 3 model2 = tf.keras.models.load_model('stable_diffusion.h5')
4 converter = tf.lite.TFLiteConverter.from_keras_model(model2)
5 tflite_model = converter.convert()
1 frames
[/usr/local/lib/python3.7/dist-packages/tensorflow_core/python/keras/saving/hdf5_format.py](https://localhost:8080/#) in load_model_from_hdf5(filepath, custom_objects, compile)
164 if model_config is None:
165 raise ValueError('No model found in config file.')
--> 166 model_config = json.loads(model_config.decode('utf-8'))
167 model = model_config_lib.model_from_config(model_config,
168 custom_objects=custom_objects)
AttributeError: 'str' object has no attribute 'decode'
I tried exporting the saved model. Had a different error than yours
--------------------------------------------------------------------------- ConverterError Traceback (most recent call last) <ipython-input-41-96c734e2e017> in <module> 2 3 converter = tf.lite.TFLiteConverter.from_saved_model('./saved_model') # path to the SavedModel directory ----> 4 tflite_model = converter.convert() /opt/conda/envs/rapids/lib/python3.8/site-packages/tensorflow/lite/python/lite.py in wrapper(self, *args, **kwargs) 931 def wrapper(self, *args, **kwargs): 932 # pylint: disable=protected-access --> 933 return self._convert_and_export_metrics(convert_func, *args, **kwargs) 934 # pylint: enable=protected-access 935 /opt/conda/envs/rapids/lib/python3.8/site-packages/tensorflow/lite/python/lite.py in _convert_and_export_metrics(self, convert_func, *args, **kwargs) 909 self._save_conversion_params_metric() 910 start_time = time.process_time() --> 911 result = convert_func(self, *args, **kwargs) 912 elapsed_time_ms = (time.process_time() - start_time) * 1000 913 if result: /opt/conda/envs/rapids/lib/python3.8/site-packages/tensorflow/lite/python/lite.py in convert(self) 1214 graph_def) 1215 -> 1216 return self._convert_from_saved_model(graph_def) 1217 1218 /opt/conda/envs/rapids/lib/python3.8/site-packages/tensorflow/lite/python/lite.py in _convert_from_saved_model(self, graph_def) 1097 converter_kwargs.update(quant_mode.converter_flags()) 1098 -> 1099 result = _convert_saved_model(**converter_kwargs) 1100 return self._optimize_tflite_model( 1101 result, quant_mode, quant_io=self.experimental_new_quantizer) /opt/conda/envs/rapids/lib/python3.8/site-packages/tensorflow/lite/python/convert_phase.py in wrapper(*args, **kwargs) 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)) /opt/conda/envs/rapids/lib/python3.8/site-packages/tensorflow/lite/python/convert_phase.py in wrapper(*args, **kwargs) 203 def wrapper(*args, **kwargs): 204 try: --> 205 return func(*args, **kwargs) 206 except ConverterError as converter_error: 207 if converter_error.errors: /opt/conda/envs/rapids/lib/python3.8/site-packages/tensorflow/lite/python/convert.py in convert_saved_model(**kwargs) 806 model_flags = build_model_flags(**kwargs) 807 conversion_flags = build_conversion_flags(**kwargs) --> 808 data = convert( 809 model_flags.SerializeToString(), 810 conversion_flags.SerializeToString(), /opt/conda/envs/rapids/lib/python3.8/site-packages/tensorflow/lite/python/convert.py in convert(model_flags_str, conversion_flags_str, input_data_str, debug_info_str, enable_mlir_converter) 308 for error_data in _metrics_wrapper.retrieve_collected_errors(): 309 converter_error.append_error(error_data) --> 310 raise converter_error 311 312 return _run_deprecated_conversion_binary(model_flags_str, ConverterError: <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_110/padded_conv2d_691/conv2d_691/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_110/padded_conv2d_691/conv2d_691/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_110/padded_conv2d_692/conv2d_692/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_110/padded_conv2d_692/conv2d_692/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_80/padded_conv2d_693/conv2d_693/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_80/padded_conv2d_693/conv2d_693/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_111/padded_conv2d_695/conv2d_695/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_111/padded_conv2d_695/conv2d_695/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_111/padded_conv2d_696/conv2d_696/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_111/padded_conv2d_696/conv2d_696/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_81/padded_conv2d_697/conv2d_697/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_81/padded_conv2d_697/conv2d_697/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_112/padded_conv2d_700/conv2d_700/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_112/padded_conv2d_700/conv2d_700/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_112/padded_conv2d_701/conv2d_701/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_112/padded_conv2d_701/conv2d_701/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_82/padded_conv2d_703/conv2d_703/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_82/padded_conv2d_703/conv2d_703/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_113/padded_conv2d_705/conv2d_705/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_113/padded_conv2d_705/conv2d_705/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_113/padded_conv2d_706/conv2d_706/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_113/padded_conv2d_706/conv2d_706/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_83/padded_conv2d_707/conv2d_707/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_83/padded_conv2d_707/conv2d_707/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_114/padded_conv2d_710/conv2d_710/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_114/padded_conv2d_710/conv2d_710/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_114/padded_conv2d_711/conv2d_711/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_114/padded_conv2d_711/conv2d_711/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_84/padded_conv2d_713/conv2d_713/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_84/padded_conv2d_713/conv2d_713/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_115/padded_conv2d_715/conv2d_715/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_115/padded_conv2d_715/conv2d_715/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_115/padded_conv2d_716/conv2d_716/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_115/padded_conv2d_716/conv2d_716/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_85/padded_conv2d_717/conv2d_717/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_85/padded_conv2d_717/conv2d_717/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_116/padded_conv2d_720/conv2d_720/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_116/padded_conv2d_720/conv2d_720/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_116/padded_conv2d_721/conv2d_721/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_116/padded_conv2d_721/conv2d_721/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_117/padded_conv2d_722/conv2d_722/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_117/padded_conv2d_722/conv2d_722/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_117/padded_conv2d_723/conv2d_723/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_117/padded_conv2d_723/conv2d_723/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_118/padded_conv2d_724/conv2d_724/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_118/padded_conv2d_724/conv2d_724/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_118/padded_conv2d_725/conv2d_725/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_118/padded_conv2d_725/conv2d_725/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_86/padded_conv2d_726/conv2d_726/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_86/padded_conv2d_726/conv2d_726/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_119/padded_conv2d_728/conv2d_728/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_119/padded_conv2d_728/conv2d_728/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_119/padded_conv2d_729/conv2d_729/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_119/padded_conv2d_729/conv2d_729/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_120/padded_conv2d_730/conv2d_730/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_120/padded_conv2d_730/conv2d_730/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_120/padded_conv2d_731/conv2d_731/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_120/padded_conv2d_731/conv2d_731/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_121/padded_conv2d_733/conv2d_733/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_121/padded_conv2d_733/conv2d_733/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_121/padded_conv2d_734/conv2d_734/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_121/padded_conv2d_734/conv2d_734/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_122/padded_conv2d_736/conv2d_736/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_122/padded_conv2d_736/conv2d_736/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_122/padded_conv2d_737/conv2d_737/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_122/padded_conv2d_737/conv2d_737/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_123/padded_conv2d_740/conv2d_740/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_123/padded_conv2d_740/conv2d_740/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_123/padded_conv2d_741/conv2d_741/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_123/padded_conv2d_741/conv2d_741/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_87/padded_conv2d_743/conv2d_743/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_87/padded_conv2d_743/conv2d_743/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_124/padded_conv2d_745/conv2d_745/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_124/padded_conv2d_745/conv2d_745/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_124/padded_conv2d_746/conv2d_746/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_124/padded_conv2d_746/conv2d_746/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_88/padded_conv2d_748/conv2d_748/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_88/padded_conv2d_748/conv2d_748/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_125/padded_conv2d_750/conv2d_750/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_125/padded_conv2d_750/conv2d_750/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_125/padded_conv2d_751/conv2d_751/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_125/padded_conv2d_751/conv2d_751/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_89/padded_conv2d_753/conv2d_753/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_89/padded_conv2d_753/conv2d_753/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_126/padded_conv2d_756/conv2d_756/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_126/padded_conv2d_756/conv2d_756/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_126/padded_conv2d_757/conv2d_757/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_126/padded_conv2d_757/conv2d_757/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_90/padded_conv2d_759/conv2d_759/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_90/padded_conv2d_759/conv2d_759/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_127/padded_conv2d_761/conv2d_761/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_127/padded_conv2d_761/conv2d_761/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_127/padded_conv2d_762/conv2d_762/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_127/padded_conv2d_762/conv2d_762/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_91/padded_conv2d_764/conv2d_764/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_91/padded_conv2d_764/conv2d_764/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_128/padded_conv2d_766/conv2d_766/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_128/padded_conv2d_766/conv2d_766/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_128/padded_conv2d_767/conv2d_767/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_128/padded_conv2d_767/conv2d_767/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_92/padded_conv2d_769/conv2d_769/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_92/padded_conv2d_769/conv2d_769/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_129/padded_conv2d_772/conv2d_772/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_129/padded_conv2d_772/conv2d_772/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_129/padded_conv2d_773/conv2d_773/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_129/padded_conv2d_773/conv2d_773/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_93/padded_conv2d_775/conv2d_775/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_93/padded_conv2d_775/conv2d_775/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_130/padded_conv2d_777/conv2d_777/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_130/padded_conv2d_777/conv2d_777/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_130/padded_conv2d_778/conv2d_778/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_130/padded_conv2d_778/conv2d_778/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_94/padded_conv2d_780/conv2d_780/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_94/padded_conv2d_780/conv2d_780/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_131/padded_conv2d_782/conv2d_782/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_131/padded_conv2d_782/conv2d_782/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_131/padded_conv2d_783/conv2d_783/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/res_block_131/padded_conv2d_783/conv2d_783/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_95/padded_conv2d_785/conv2d_785/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/spatial_transformer_95/padded_conv2d_785/conv2d_785/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/padded_conv2d_787/conv2d_787/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): 'tf.Conv2D' op is neither a custom op nor a flex op <unknown>:0: note: loc(fused["StatefulPartitionedCall:", "StatefulPartitionedCall"]): called from <unknown>:0: note: loc(callsite(callsite(fused["Conv2D:", "model_16/u_net_model_5/padded_conv2d_787/conv2d_787/Conv2D@__inference__wrapped_model_706123"] at fused["StatefulPartitionedCall:", "StatefulPartitionedCall@__inference_signature_wrapper_738202"]) at fused["StatefulPartitionedCall:", "StatefulPartitionedCall"])): Error code: ERROR_NEEDS_FLEX_OPS <unknown>:0: error: failed while converting: 'main': Some ops are not supported by the native TFLite runtime, you can enable TF kernels fallback using TF Select. See instructions: https://www.tensorflow.org/lite/guide/ops_select TF Select ops: Conv2D Details: tf.Conv2D(tensor<?x?x?x?xf32>, tensor<1x1x1280x1280xf32>) -> (tensor<?x?x?x1280xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} tf.Conv2D(tensor<?x?x?x?xf32>, tensor<1x1x320x320xf32>) -> (tensor<?x?x?x320xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} tf.Conv2D(tensor<?x?x?x?xf32>, tensor<1x1x640x640xf32>) -> (tensor<?x?x?x640xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x1280x1280xf32>) -> (tensor<?x?x?x1280xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x1280x640xf32>) -> (tensor<?x?x?x640xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x1920x1280xf32>) -> (tensor<?x?x?x1280xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x1920x640xf32>) -> (tensor<?x?x?x640xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x2560x1280xf32>) -> (tensor<?x?x?x1280xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x320x320xf32>) -> (tensor<?x?x?x320xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x320x4xf32>) -> (tensor<?x?x?x4xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x320x640xf32>) -> (tensor<?x?x?x640xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x640x1280xf32>) -> (tensor<?x?x?x1280xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x640x320xf32>) -> (tensor<?x?x?x320xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x640x640xf32>) -> (tensor<?x?x?x640xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x960x320xf32>) -> (tensor<?x?x?x320xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} tf.Conv2D(tensor<?x?x?x?xf32>, tensor<3x3x960x640xf32>) -> (tensor<?x?x?x640xf32>) : {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true}
regarding the difference in the errors that you got. you need to add TensorFlow ops and in case it needed also add enable TensorFlow ops:
converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)
converter.target_spec.supported_ops = [
tf.lite.OpsSet.TFLITE_BUILTINS, # enable TensorFlow Lite ops.
tf.lite.OpsSet.SELECT_TF_OPS # enable TensorFlow ops.
]
tflite_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_model)
from the docs: https://www.tensorflow.org/lite/guide/ops_select
Thank you @costiash,
I tried this finally and it worked:
import tensorflow as tf
model2 = tf.keras.models.load_model('/content/saved_model')
converter = tf.lite.TFLiteConverter.from_keras_model(model2)
converter.target_spec.supported_ops = [
tf.lite.OpsSet.TFLITE_BUILTINS, # enable TensorFlow Lite ops.
tf.lite.OpsSet.SELECT_TF_OPS # enable TensorFlow ops.
]
converter.optimizations = [tf.lite.Optimize.DEFAULT]
converter.target_spec.supported_types = [tf.float16]
tflite_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_model)
I have an output model with 1.6 GB. I will try to run it with edge tpu and get back to update here.
@charbull , any chance you've had time to test this? I just ordered an Edge TPU to try out stable diffusion on (Seemed cheaper than buying a new graphics card). If it works any chance you have a full working code example upstream of generating the tflite file? Also curious what you got for timing and quality of output :)
Hi @joe-albert,
Ended up opening an issue on keras_cv repo here: https://github.com/keras-team/keras-cv/issues/1033
It seems there is a need to convert the three parts separately to tf lite int8 and then merge them to use them on the edge tpu. I am not exactly sure how to merge yet, will leave comments on the issue above as I progress.
Let us know if you make progress on your end too :)
Cheers Charbull
Just so I can keep following along, your goal is to make a TF Lite model so you can use a TPU? I looked up a TPU and is it basically a "raspberry pi" that can do ML? Any idea of the expected speed? (IE, can return an image in x amount of seconds)
@soten355 yes, the idea is to use the TPU on the board instead of the CPU. the raspberry pi does not have a TPU. you can look online it seems the TPU's performance tops GPUs/CPUs
@charbull , to add to that--
@soten355 you might have seen the Coral dev board, which is similar to a raspberry pi, with one difference being that it has a TPU integrated into it. TPU stands for Tensor Processing Unit, essentially it is a specialized chip to perform vector and matrix operations used in tensorflow.
I have no idea what the expected speed would be unfortunately. Googles TPU cards with 8 M2 chips (I think that's what was used but might be wrong!) have been clocked at 8 images in ~7 seconds . I believe the Coral USB accelerator has 1 M2 module
I came here wondering if it's worth buying a Coral USB TPU? I saw a similar question at https://github.com/google-coral/edgetpu/issues/621
It seems you're in the middle of testing it? Does it work?
@rvalitov I tried it so far on the provided examples it is pretty cool ! My next step is to make stable diffusion run on it.
@SkyyySi sorry for the late reply, is it still on my radar, didnt try things out yet.
@SkyyySi , I've been busy with other things too and haven't had time to look at this :/
@joe-albert Don't worry don't worry, just wanted to check since maybe you found something and just forgot to post.
Good look with whatever you're doing.
Hello @charbull @joe-albert Sorry to disturb you. I just wanted to know if you have finally succeeded to run stable diffusion on Coral USB accelerator? :-)
