Searched refs:splits_shape (Results 1 – 2 of 2) sorted by relevance
278 ShapeHandle splits_shape = c->input(1); in __anonb4ab91930802() local279 TF_RETURN_IF_ERROR(c->WithRank(splits_shape, 1, &unused)); in __anonb4ab91930802()283 TF_RETURN_IF_ERROR(c->Subtract(c->Dim(splits_shape, 0), 1, &dims[0])); in __anonb4ab91930802()
979 splits_shape = array_ops.shape(self.row_splits, out_type=dtypes.int64)982 ragged_dimensions = array_ops.stack([splits_shape[0] - 1] + [