Searched refs:nested_splits_ (Results 1 – 2 of 2) sorted by relevance
41 : values_(std::move(values)), nested_splits_(nested_splits) {} in RaggedTensorVariant()55 int ragged_rank() const { return nested_splits_.size(); } in ragged_rank()56 const std::vector<Tensor>& nested_splits() const { return nested_splits_; } in nested_splits()57 std::vector<Tensor>* mutable_nested_splits() { return &nested_splits_; } in mutable_nested_splits()58 const Tensor& splits(int i) const { return nested_splits_[i]; } in splits()59 Tensor* mutable_splits(int i) { return &nested_splits_[i]; } in mutable_splits()61 nested_splits_ = nested_splits; in set_nested_splits()63 void append_splits(const Tensor& splits) { nested_splits_.push_back(splits); } in append_splits()67 std::vector<Tensor> nested_splits_; variable
30 ", ragged_rank=", nested_splits_.size(), ", splits_dtype=", in DebugString()31 DataTypeString(nested_splits_.empty() ? DT_INVALID in DebugString()32 : nested_splits_.back().dtype())); in DebugString()37 for (const auto& splits : nested_splits_) { in Encode()47 nested_splits_.assign(data.tensors().begin(), in Decode()