We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7d16952 commit 877c95bCopy full SHA for 877c95b
1 file changed
lib/ortex/backend.ex
@@ -88,6 +88,7 @@ defmodule Ortex.Backend do
88
@impl true
89
def reshape(out, %T{data: %B{ref: ref}}) do
90
shape = Nx.shape(out) |> Tuple.to_list()
91
+
92
case Ortex.Native.reshape(ref, shape) do
93
{:error, msg} -> raise msg
94
res -> put_in(out.data, %B{ref: res})
@@ -106,14 +107,13 @@ defmodule Ortex.Backend do
106
107
out
108
| shape: new_shape,
109
names: new_names,
- data:
110
- %B{
111
- ref:
112
- case Ortex.Native.reshape(ref, new_shape |> Tuple.to_list()) do
113
- {:error, msg} -> raise msg
114
- res -> res
115
- end
116
- }
+ data: %B{
+ ref:
+ case Ortex.Native.reshape(ref, new_shape |> Tuple.to_list()) do
+ {:error, msg} -> raise msg
+ res -> res
+ end
+ }
117
}
118
end
119
0 commit comments