Searched refs:split_diff (Results 1 – 1 of 1) sorted by relevance
93 int split_diff = nested_dim_sizes[i + 1]; in NestedStackRaggedTensors() local95 splits_vec(j) = j * split_diff; in NestedStackRaggedTensors()