Revision cae2cf7d8b75ffe7231f93cd3ec48862f1d328cf authored by Sandeep Dasgupta on 24 January 2022, 22:41:24 UTC, committed by jax authors on 01 February 2022, 17:57:45 UTC
The CL does the following: 1. Added mhlo verifiers to ascertain that the imported structure of the layout is as expected. 2. Modified the tf lowering (new/old bridge) or jax lowering for infeed/outfeed op in this CL to generate the non-tuple variant of infeed op. PiperOrigin-RevId: 423914979
1 parent e3fe4a2
| File | Mode | Size |
|---|---|---|
| jax_cpp | ||
| __init__.py | -rw-r--r-- | 575 bytes |
| advi.py | -rw-r--r-- | 4.8 KB |
| control.py | -rw-r--r-- | 5.7 KB |
| control_test.py | -rw-r--r-- | 7.4 KB |
| datasets.py | -rw-r--r-- | 3.1 KB |
| differentially_private_sgd.py | -rw-r--r-- | 8.8 KB |
| examples_test.py | -rw-r--r-- | 3.5 KB |
| gaussian_process_regression.py | -rw-r--r-- | 4.5 KB |
| kernel_lsq.py | -rw-r--r-- | 2.5 KB |
| mnist_classifier.py | -rw-r--r-- | 3.1 KB |
| mnist_classifier_fromscratch.py | -rw-r--r-- | 3.0 KB |
| mnist_vae.py | -rw-r--r-- | 4.9 KB |
| onnx2xla.py | -rw-r--r-- | 4.7 KB |
| resnet50.py | -rw-r--r-- | 4.3 KB |
| spmd_mnist_classifier_fromscratch.py | -rw-r--r-- | 4.9 KB |
Computing file changes ...
