-
Sample IR shown below produces two dispatches. For `linalg_ext.scatter` we need to support fusion with transposition and collapse / expand. We do this as `arg0` could be created via a concatenation of…
-
There are a number of operators, such as `transpose()` and `reshape()` which are no-ops when applied to a scalar tensor. In addition there are operators like `gather()` which expect indices but a scal…
-
### Context
OpenVINO component responsible for support of JAX/Flax models is called as JAX Frontend (JAX FE). JAX FE converts a JAX/Flax model represented by `ClosedJAXpr` graph object with operation…
-
Is there a way to convert an existing tensor to a new tensor with the same content, but with the modes ordered differently? This would be a generalization of matrix transposition to tensors.
-
### Context
OpenVINO component responsible for support of JAX/Flax models is called as JAX Frontend (JAX FE). JAX FE converts a JAX/Flax model represented by `ClosedJAXpr` graph object with operati…
-
### Context
OpenVINO component responsible for support of JAX/Flax models is called as JAX Frontend (JAX FE). JAX FE converts a JAX/Flax model represented by `ClosedJAXpr` graph object with operation…
-
I know this is an experimental feature, but as I need it, I have performed some tests and identified the two following issues, for which I can propose a correction.
First is an error with the batch…
bwery updated
3 years ago
-
### Context
OpenVINO component responsible for support of JAX/Flax models is called as JAX Frontend (JAX FE). JAX FE converts a JAX/Flax model represented by `ClosedJAXpr` graph object with operation…
-
### Context
OpenVINO component responsible for support of JAX/Flax models is called as JAX Frontend (JAX FE). JAX FE converts a JAX/Flax model represented by `ClosedJAXpr` graph object with operation…
-
# GenericTensor 2.0: plan
Alright, time for improvement
## 1 action - 1 interface
Currently we have a large interface which forces the user to implement all its methods. It's not as good as i…