Searched refs:element_dim_sizes (Results 1 – 2 of 2) sorted by relevance
44 gtl::InlinedVector<int64, 4> element_dim_sizes; in Dataset() local48 element_dim_sizes.push_back(t.dim_size(i)); in Dataset()50 partial_shapes_.emplace_back(element_dim_sizes); in Dataset()51 shapes_.emplace_back(std::move(element_dim_sizes)); in Dataset()
488 gtl::InlinedVector<int64, 1> element_dim_sizes; in NestedDataset() local489 element_dim_sizes.push_back(1); in NestedDataset()490 partial_shapes_.emplace_back(element_dim_sizes); in NestedDataset()