def test_plus_for(self):
self.assertEqual(
str(
intrinsic_utils.plus_for(
tf.int32,
context_stack_impl.context_stack).type_signature),
'(<int32,int32> -> int32)')
def foo(x, y):
return intrinsic_utils.plus_for(type_spec,
context_stack_impl.context_stack)(x, y)
def foo(x, y):
return intrinsic_utils.plus_for(tf.int32,
context_stack_impl.context_stack)(x, y)
def federated_sum(x): zero = intrinsic_utils.zero_for(x.type_signature.member, context_stack) plus = intrinsic_utils.plus_for(x.type_signature.member, context_stack) return intrinsics.federated_reduce(x, zero, plus)
