Add some pydoc to other columns. Add weight_collections tests for all columns. Rename embedding 'weights' -> 'embedding_weights', linear 'weight' -> 'weights', and linear 'bias_weight' -> 'bias_weights'. PiperOrigin-RevId: 155648428
136 B
136 B