def __call__(self, state, *args, **kwargs): """Performs the stateful function call. Args: state: A `tff.Value` placed on the `tff.SERVER`. *args: Arguments to the function. **kwargs: Arguments to the function. Returns: A tuple of `tff.Value`s (state@SERVER, ...) where * state: The updated state, to be passed to the next invocation of call. * ...: The result of the aggregation. """ return self._next_fn(values.to_value(state), *args, **kwargs)
def one_arg_initialize(one_arg): del one_arg # unused return values.to_value(0)
def test_to_value_with_int_constant(self): val = values.to_value(10) self.assertIsInstance(val, value_base.Value) self.assertEqual(str(val.type_signature), 'int32')