• Matt's avatar
    Stop storing references to bound methods via tf.function (#24146) · 3bd1fe43
    Matt authored
    * Stop storing references to bound methods in tf.functions
    
    * Remove the gc.collect calls now that we resolved the underlying problem
    
    * Remove the default signature from model.serving entirely, big cleanup
    
    * Remove _prune_signature as self.input_signature can prune itself
    
    * Restore serving docstring
    
    * Update int support test to check the input signature
    
    * Make sure other tests also use model.input_signature and not serving.input_signature
    
    * Restore _prune_signature
    
    * Remove the doctest GC now it's no longer needed
    
    * Correct core tests to use the pruned sig
    
    * order lines correctly in core tests
    
    * Add eager_serving back with a deprecation warning
    3bd1fe43
test_modeling_tf_rag.py 39.7 KB