Searched refs:_moveaxis (Results 1 – 2 of 2) sorted by relevance
| /dports/science/py-chainer/chainer-7.8.0/chainer/functions/array/ |
| H A D | moveaxis.py | 15 def _moveaxis(a, source, destination, xp): function
91 return _moveaxis(inputs[0], self.source, self.destination, xp),
|
| /dports/math/py-jax/jax-0.2.9/jax/_src/lax/ |
| H A D | parallel.py | 684 def _moveaxis(src, dst, x): function
695 return _moveaxis(0, concat_axis, local_slice)
772 return _moveaxis(d, concat_axis_adj, x), split_axis_adj
929 return _moveaxis(d, all_gather_dimension, x), batching.not_mapped
|