• 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_common.py 11.5 KB