Searched refs:_serving_input_receiver_fn (Results 1 – 2 of 2) sorted by relevance
42 self._serving_input_receiver_fn = testing_common.get_arithmetic_input_fn(52 serving_input_receiver_fn=self._serving_input_receiver_fn)66 serving_input_receiver_fn=self._serving_input_receiver_fn,
165 def _serving_input_receiver_fn(): function211 return _serving_input_receiver_fn234 def _serving_input_receiver_fn(): function285 return _serving_input_receiver_fn