Searched refs:num_slices_in_dimension (Results 1 – 1 of 1) sorted by relevance
347 def num_slices_in_dimension(self, axis): member in RaggedTensorDynamicShape354 return self.dimension_size(axis) * self.num_slices_in_dimension(axis - 1)365 splits = array_ops.stack([0, self.num_slices_in_dimension(axis)])