Searched refs:get_positive_axis (Results 1 – 5 of 5) sorted by relevance
43 def get_positive_axis(axis, ndims): function152 axis = get_positive_axis(axis, data.shape.ndims)
155 axis = ragged_util.get_positive_axis(axis, out_ndims)
473 axis = ragged_util.get_positive_axis(axis, ndims)
477 axis = ragged_util.get_positive_axis(axis, rt_input.shape.ndims)
911 axis = ragged_util.get_positive_axis(axis, self.shape.ndims)