Searched refs:mutable_splits (Results 1 – 4 of 4) sorted by relevance
92 auto splits_vec = output_ragged->mutable_splits(i)->vec<SPLIT_TYPE>(); in NestedStackRaggedTensors()104 output_ragged->mutable_splits(dims - 1)->vec<SPLIT_TYPE>(); in NestedStackRaggedTensors()126 output_ragged->mutable_splits(split_index)->vec<SPLIT_TYPE>(); in NestedStackRaggedTensors()
59 Tensor* mutable_splits(int i) { return &nested_splits_[i]; } in mutable_splits() function
97 (*ragged_components)[i].mutable_splits(j)->vec<SPLIT_TYPE>()); in UnbatchRaggedZerothDim()
539 test::FillValues<int64>(encoded.mutable_splits(0), {1, 2}); in TEST_F()