Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
Dtensor_slice_dataset_op.cc70 gtl::InlinedVector<int64, 4> partial_dim_sizes; in Dataset() local
74 partial_dim_sizes.push_back(t.dim_size(i)); in Dataset()
76 shapes_.emplace_back(std::move(partial_dim_sizes)); in Dataset()
Ddataset_test_base.cc66 gtl::InlinedVector<int64, 4> partial_dim_sizes; in CreateTensorSliceDataset() local
68 partial_dim_sizes.push_back(t.dim_size(i)); in CreateTensorSliceDataset()
70 shapes.emplace_back(std::move(partial_dim_sizes)); in CreateTensorSliceDataset()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dunbatch_dataset_op.cc45 gtl::InlinedVector<int64, 4> partial_dim_sizes; in Dataset() local
47 partial_dim_sizes.push_back(shape.dim_size(i)); in Dataset()
49 shapes_.emplace_back(std::move(partial_dim_sizes)); in Dataset()