• Haoyu Zhang's avatar
    Add Keras XLA Tests (#6286) · fa9ed456
    Haoyu Zhang authored
    * Added XLA test with a monkey-patched op to avoid OOM
    
    * Added doc strings in Keras benchmarks to avoid Lint error
    fa9ed456
keras_imagenet_benchmark.py 9.12 KB