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