jax

9da9156b - Change image resize implementation to use a matmul per dimension. (#3720)

Change image resize implementation to use a matmul per dimension. (#3720) * Change image resize implementation to use a matmul per dimension. This should have better space scaling behaviors than the previous gather approach. In particular, it does not require temporary memory that scales with the batch size or number of features of an image. * Plumb precision option through to image resize API.