Searched refs:partial_dim_sizes (Results 1 – 3 of 3) sorted by relevance
52 gtl::InlinedVector<int64_t, 4> partial_dim_sizes; in Dataset() local54 partial_dim_sizes.push_back(shape.dim_size(i)); in Dataset()56 shapes_.emplace_back(std::move(partial_dim_sizes)); in Dataset()
99 gtl::InlinedVector<int64_t, 4> partial_dim_sizes; in ListOutputShapes() local100 partial_dim_sizes.reserve(tensor.dims()); in ListOutputShapes()102 partial_dim_sizes.push_back(tensor.dim_size(i)); in ListOutputShapes()104 output_shapes.emplace_back(std::move(partial_dim_sizes)); in ListOutputShapes()
1138 gtl::InlinedVector<int64_t, 4> partial_dim_sizes; in TensorSliceShapes() local1140 partial_dim_sizes.push_back(component.dim_size(i)); in TensorSliceShapes()1142 shapes.emplace_back(std::move(partial_dim_sizes)); in TensorSliceShapes()