Searched refs:_reduce_tensor (Results 1 – 1 of 1) sorted by relevance
57 def _reduce_tensor(tensor): function70 return nest.map_structure(_reduce_tensor, inputs)