Adding a _create_state method to FeatureColumn so that we can decouple...
Adding a _create_state method to FeatureColumn so that we can decouple variable (more generally, state) creation from getting dense tensors etc. This lays the groundwork to create a Layer type interface (which is more eager friendly) on top of feature columns where variable creation is separated from getting the tensors out. PiperOrigin-RevId: 191647517
Loading
Please sign in to comment