kerastensor move location

tensorleap · Jan 2, 2025 · 5791703 · 5791703
1 parent 38840da
commit 5791703
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/onnx2kerastl/reshape_layers.py b/onnx2kerastl/reshape_layers.py
@@ -4,7 +4,7 @@
 import numpy as np
 import tensorflow as tf
 from keras import backend as K
-from keras.engine.keras_tensor import KerasTensor
+from tensorflow.python.keras.engine.keras_tensor import KerasTensor
 from keras.layers import SlicingOpLambda, Lambda
 from typing import Union
 from .utils import is_numpy, ensure_tf_type, unsqueeze_tensors_of_rank_one

diff --git a/onnx2kerastl/utils.py b/onnx2kerastl/utils.py
@@ -2,7 +2,7 @@
 
 import numpy as np
 import keras
-from keras.engine.keras_tensor import KerasTensor
+from tensorflow.python.keras.engine.keras_tensor import KerasTensor
 from keras_data_format_converter import convert_channels_first_to_last
 import tensorflow as tf
 from.tfops_funcs import tf_reshape