Searched refs:num_partitioned_dimensions (Results 1 – 2 of 2) sorted by relevance
225 def num_partitioned_dimensions(self): member in RaggedTensorDynamicShape328 if axis < self.num_partitioned_dimensions:388 axis_in_inner_dims = axis - self.num_partitioned_dimensions399 axis_in_inner_dims = axis - self.num_partitioned_dimensions462 if shape.num_partitioned_dimensions == 0:488 rt_input.ragged_rank >= dst_shape.num_partitioned_dimensions):516 rt_input, ragged_rank=dst_shape.num_partitioned_dimensions - 1)521 for axis in range(dst_shape.num_partitioned_dimensions):541 for axis in range(dst_shape.num_partitioned_dimensions):
174 params_shape.num_partitioned_dimensions -183 params_shape.num_partitioned_dimensions - num_batch_dimensions)