Searched refs:_broadcast_fn (Results 1 – 1 of 1) sorted by relevance
94 def _broadcast_fn(): function99 return nest.map_structure(lambda _: _broadcast_fn(), outputs)