Searched refs:_assume_mirrored (Results 1 – 1 of 1) sorted by relevance
709 def _assume_mirrored(grad): function714 grad = nest.map_structure(_assume_mirrored, grad)