Searched defs:splits_vec (Results 1 – 2 of 2) sorted by relevance
149 auto splits_vec = output_ragged->mutable_splits(i)->vec<SPLIT_TYPE>(); in NestedStackRaggedTensors() local182 auto splits_vec = in NestedStackRaggedTensors() local
77 const auto& splits_vec = splits->flat<SPLITS_TYPE>(); in Compute() local