Searched refs:mutable_nested_splits (Results 1 – 4 of 4) sorted by relevance
60 *to->mutable_nested_splits() = from.nested_splits(); in RaggedTensorVariantDeviceCopy()
45 ragged_component.mutable_nested_splits()->reserve(num_splits); in UnbatchRaggedZerothDim()145 batched_ragged_input.mutable_nested_splits()->reserve( in Compute()
57 std::vector<Tensor>* mutable_nested_splits() { return &nested_splits_; } in mutable_nested_splits() function
84 output_ragged->mutable_nested_splits()->reserve(output_ragged_rank); in NestedStackRaggedTensors()