• Hongkun Yu's avatar
    Remove compute_output_shape. · 0e0a94a6
    Hongkun Yu authored
    Keras: "manual" shape inference is only required if the layer is dynamic (otherwise we use TF's static shape inference capabilities)
    
    PiperOrigin-RevId: 290821518
    0e0a94a6
attention.py 10.6 KB