Lines Matching refs:side_effects
105 def _subscribe_new(tensor, side_effects, control_cache):
106 """Helper method that subscribes a single tensor to a list of side_effects.
110 side_effects: List of side_effect functions see subscribe for details.
126 for s in side_effects:
156 def _subscribe_extend(tensor, side_effects):
157 """Helper method to extend the list of side_effects for a subscribed tensor.
161 side_effects: List of side_effect functions, see subscribe for details.
174 for s in side_effects:
216 def _subscribe(tensor, side_effects, control_cache):
217 """Helper method that subscribes a single tensor to a list of side_effects.
226 side_effects: List of side_effect functions, see subscribe for details.
240 return _subscribe_extend(tensor, side_effects)
252 return _subscribe_extend(candidate_tensor, side_effects)
254 return _subscribe_new(tensor, side_effects, control_cache)
288 def _scoped_subscribe(tensor, side_effects, control_cache):
289 """Helper method that subscribes a single tensor to a list of side_effects.
297 side_effects: List of side_effect functions, see subscribe for details.
306 return _subscribe(tensor, side_effects, control_cache)
309 def subscribe(tensors, side_effects):
334 side_effects: Function(s) that takes a `Tensor`, construct a subgraph, and
344 if not hasattr(side_effects, '__iter__'):
345 side_effects = [side_effects]
349 tensors, lambda t: _scoped_subscribe(t, side_effects, control_outputs))
