Searched refs:target_shape_without_bdim_to_reduce_to (Results 1 – 1 of 1) sorted by relevance
624 target_shape_without_bdim_to_reduce_to=None, argument633 if target_shape_without_bdim_to_reduce_to is None:634 target_shape_without_bdim_to_reduce_to = len(grad_input) * (None,)641 target_shape_without_bdim_to_reduce_to,652 target_shape_without_bdim_to_reduce_to=None, argument694 if target_shape_without_bdim_to_reduce_to is not None:699 )(grad_input, target_shape_without_bdim_to_reduce_to)