Home
last modified time | relevance | path

Searched refs:partial_dim_sizes (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dunbatch_dataset_op.cc47 gtl::InlinedVector<int64, 4> partial_dim_sizes; in Dataset() local
49 partial_dim_sizes.push_back(shape.dim_size(i)); in Dataset()
51 shapes_.emplace_back(std::move(partial_dim_sizes)); in Dataset()
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_test_base.cc1107 gtl::InlinedVector<int64, 4> partial_dim_sizes; in TensorSliceShapes() local
1109 partial_dim_sizes.push_back(component.dim_size(i)); in TensorSliceShapes()
1111 shapes.emplace_back(std::move(partial_dim_sizes)); in TensorSliceShapes()