Searched refs:outer_splits_shape (Results 1 – 2 of 2) sorted by relevance
2359 outer_splits_shape = [None if outer_dim is None else outer_dim + 1]2364 tensor_spec.TensorSpec(outer_splits_shape, self._row_splits_dtype)
2769 TensorShape values_shape, inner_splits_shape, outer_splits_shape; in ParseSequenceRaggedFeatures() local2777 outer_splits_shape.AddDim(num_examples + 1); in ParseSequenceRaggedFeatures()2783 Tensor(allocator, splits_dtype, outer_splits_shape); in ParseSequenceRaggedFeatures()