Searched refs:_gather_state (Results 1 – 1 of 1) sorted by relevance
189 def _gather_state(self, features): member in TimeSeriesRegressionHead255 features, passed_flat_state = self._gather_state(features)