Searched refs:partial_dim_sizes (Results 1 – 2 of 2) sorted by relevance
47 gtl::InlinedVector<int64, 4> partial_dim_sizes; in Dataset() local49 partial_dim_sizes.push_back(shape.dim_size(i)); in Dataset()51 shapes_.emplace_back(std::move(partial_dim_sizes)); in Dataset()
1107 gtl::InlinedVector<int64, 4> partial_dim_sizes; in TensorSliceShapes() local1109 partial_dim_sizes.push_back(component.dim_size(i)); in TensorSliceShapes()1111 shapes.emplace_back(std::move(partial_dim_sizes)); in TensorSliceShapes()