Searched refs:nested_dim_sizes (Results 1 – 1 of 1) sorted by relevance
129 const std::vector<int>& nested_dim_sizes, const int input_ragged_rank, in NestedStackRaggedTensors() argument132 const int dims = nested_dim_sizes.size(); in NestedStackRaggedTensors()146 int dims_splits_size = nested_dim_sizes[i] + 1; in NestedStackRaggedTensors()150 int split_diff = nested_dim_sizes[i + 1]; in NestedStackRaggedTensors()