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