Searched refs:inner_split (Results 1 – 2 of 2) sorted by relevance
171 inner_split = 0174 inner_split = dim + 1177 return RaggedTensorDynamicShape(dim_sizes[:inner_split],178 dim_sizes[inner_split:])
2814 size_t inner_split = 0; // total number of elements we've seen so far in ParseSequenceRaggedFeatures() local2837 inner_split += num_added; in ParseSequenceRaggedFeatures()2855 *int32_inner_splits++ = inner_split; in ParseSequenceRaggedFeatures()2857 *int64_inner_splits++ = inner_split; in ParseSequenceRaggedFeatures()2872 if (inner_split != expected_num_elements) { in ParseSequenceRaggedFeatures()