Searched refs:_serving_ops (Results 1 – 1 of 1) sorted by relevance
145 def _serving_ops(self, features): member in TimeSeriesRegressionHead277 return self._serving_ops(features)335 def _serving_ops(self, features): member in OneShotPredictionHead