• Toby Boyd's avatar
    Unit tests pass TF 2.0 GPU and CPU locally. (#7101) · 49097655
    Toby Boyd authored
    * Fix unit tests failures.
    
    * 96% of TF 2.0 tests on GPU are passing.
    
    * Currently all passing GPU and CPU TF 2.0
    
    * Address code comments.
    
    * use tf 2.0 cast.
    
    * Comment about working on TF 2.0 CPU
    
    * Uses contrib turn off for TF 2.0.
    
    * Fix wide_deep and add keras_common_tests.
    
    * use context to get num_gpus.
    
    * Switch to tf.keras.metrics
    49097655
reference_data.py 12.9 KB