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 |
|---|---|---|
| .github | ||
| benchmarks | ||
| build | ||
| cloud_tpu_colabs | ||
| design_notes | ||
| docs | ||
| examples | ||
| images | ||
| jax | ||
| jaxlib | ||
| tests | ||
| third_party | ||
| .bazelrc | -rw-r--r-- | 16.0 KB |
| .bazelversion | -rw-r--r-- | 6 bytes |
| .gitignore | -rw-r--r-- | 291 bytes |
| .pre-commit-config.yaml | -rw-r--r-- | 563 bytes |
| .readthedocs.yml | -rw-r--r-- | 569 bytes |
| CHANGELOG.md | -rw-r--r-- | 47.7 KB |
| CONTRIBUTING.md | -rw-r--r-- | 150 bytes |
| LICENSE | -rw-r--r-- | 11.1 KB |
| README.md | -rw-r--r-- | 23.0 KB |
| WORKSPACE | -rw-r--r-- | 1.3 KB |
| conftest.py | -rw-r--r-- | 856 bytes |
| mypy.ini | -rw-r--r-- | 618 bytes |
| pylintrc | -rw-r--r-- | 1.6 KB |
| pytest.ini | -rw-r--r-- | 1.3 KB |
| setup.cfg | -rw-r--r-- | 329 bytes |
| setup.py | -rw-r--r-- | 3.2 KB |
Computing file changes ...
