diff --git a/ivy/functional/backends/tensorflow/layers.py b/ivy/functional/backends/tensorflow/layers.py index 4a80b23746a8..8c24a4232310 100644 --- a/ivy/functional/backends/tensorflow/layers.py +++ b/ivy/functional/backends/tensorflow/layers.py @@ -790,7 +790,7 @@ def lstm_update( recurrent_bias = ( recurrent_bias.data if recurrent_bias is not None else recurrent_bias ) - if "cpu" in dev: + if "cpu" in dev.lower(): outputs, new_states = _cpu_lstm( x, init_h, diff --git a/ivy/functional/frontends/torch/_VF/__init__.py b/ivy/functional/frontends/torch/_VF/__init__.py new file mode 100644 index 000000000000..80c1dc86f4f9 --- /dev/null +++ b/ivy/functional/frontends/torch/_VF/__init__.py @@ -0,0 +1,4 @@ +import ivy.functional.frontends.torch as torch_frontend + +def lstm(*args, **kwargs): + return torch_frontend.lstm(*args, **kwargs) diff --git a/ivy/functional/frontends/torch/__init__.py b/ivy/functional/frontends/torch/__init__.py index c1fd0d888f66..f3538d09b3cf 100644 --- a/ivy/functional/frontends/torch/__init__.py +++ b/ivy/functional/frontends/torch/__init__.py @@ -268,6 +268,7 @@ def promote_types_of_torch_inputs( from .nn.functional import softmax, relu, lstm from . import special from . import tensor +from . import _VF from .tensor import * from . import blas_and_lapack_ops from .blas_and_lapack_ops import *