Lines Matching refs:subscribe
15 """Tests for tf.subscribe."""
27 from tensorflow.python.framework import subscribe
45 all(subscribe._is_subscribed_identity(x) for x in container))
63 c = subscribe.subscribe(c,
90 subscribed = subscribe.subscribe(
96 subscribed = subscribe.subscribe(
102 subscribed = subscribe.subscribe({
114 subscribed = subscribe.subscribe(
121 subscribe.subscribe(c.name,
137 a = subscribe.subscribe(a,
145 b = subscribe.subscribe(b,
162 c_sub = subscribe.subscribe(c, [])
164 self.assertFalse(subscribe._is_subscribed_identity(a))
165 self.assertFalse(subscribe._is_subscribed_identity(c))
166 self.assertFalse(subscribe._is_subscribed_identity(idop))
167 self.assertTrue(subscribe._is_subscribed_identity(c_sub))
181 # Subscribe with a first side effect graph, passing an unsubscribed tensor.
183 c_sub = subscribe.subscribe(
187 # previous call to subscribe().
189 c_sub2 = subscribe.subscribe(
194 c_sub3 = subscribe.subscribe(
203 # Expect the objects returned by subscribe() to reference the same tensor.
227 v1_sub = subscribe.subscribe(
229 self.assertTrue(subscribe._is_subscribed_identity(v1_sub))
253 """Confirm that subscribe correctly handles tensors with 'resource' type."""
271 tensor_array_sub = subscribe.subscribe(
274 self.assertFalse(subscribe._is_subscribed_identity(tensor_array.handle))
303 # Subscribe the three outputs at once.
304 subscribe.subscribe(sparse_add.op.outputs,
327 add_sub = subscribe.subscribe(
330 mul_sub = subscribe.subscribe(
333 # Expect the identity tensors injected by subscribe to have been created
366 subscriptions = subscribe.subscribe(
