| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | batch_dataset_op.cc | 60 drop_remainder_(drop_remainder), in Dataset()
77 if (drop_remainder_ || input_->Cardinality() == kInfiniteCardinality) {
117 return n / batch_size_ + (n % batch_size_ == 0 || drop_remainder_ ? 0 : 1); in Cardinality()
138 TF_RETURN_IF_ERROR(b->AddScalar(drop_remainder_, &drop_remainder)); in AsGraphDefInternal()
188 if (dataset()->drop_remainder_ && in GetNextInternal()
248 const bool drop_remainder_; member in tensorflow::data::BatchDatasetOp::Dataset
|
| D | window_dataset_op.cc | 53 drop_remainder_(drop_remainder), in Dataset()
85 drop_remainder_); in DebugString()
95 if (drop_remainder_) { in Cardinality()
130 TF_RETURN_IF_ERROR(b->AddScalar(drop_remainder_, &drop_remainder_node)); in AsGraphDefInternal()
163 (dataset()->drop_remainder_ && buffer_.size() < target_size))) { in GetNextInternal()
188 (dataset()->drop_remainder_ && buffer_.size() < target_size)) { in GetNextInternal()
385 const bool drop_remainder_; member in tensorflow::data::WindowDatasetOp::Dataset
|
| D | padded_batch_dataset_op.cc | 57 drop_remainder_(drop_remainder), in Dataset()
79 if (drop_remainder_ || input_->Cardinality() == kInfiniteCardinality) {
119 return n / batch_size_ + (n % batch_size_ == 0 || drop_remainder_ ? 0 : 1); in Cardinality()
161 TF_RETURN_IF_ERROR(b->AddScalar(drop_remainder_, &drop_remainder)); in AsGraphDefInternal()
226 if (dataset()->drop_remainder_ && in GetNextInternal()
384 const bool drop_remainder_; member in tensorflow::data::PaddedBatchDatasetOp::Dataset
|
| D | parallel_batch_dataset_op.cc | 77 drop_remainder_(drop_remainder), in Dataset()
90 if (drop_remainder_ || input_->Cardinality() == kInfiniteCardinality) {
127 return n / batch_size_ + (n % batch_size_ == 0 || drop_remainder_ ? 0 : 1); in Cardinality()
157 TF_RETURN_IF_ERROR(b->AddScalar(drop_remainder_, &drop_remainder)); in AsGraphDefInternal()
226 dataset()->drop_remainder_, result->status, ctx, in GetNextInternal()
518 const bool drop_remainder_; member in tensorflow::data::ParallelBatchDatasetOp::Dataset
|
| D | parallel_batch_dataset_op_test.cc | 38 drop_remainder_(drop_remainder) { in ParallelBatchDatasetParams()
51 CreateTensor<bool>(TensorShape({}), {drop_remainder_}); in GetInputTensors()
76 bool drop_remainder_; member in tensorflow::data::__anon213a87dd0111::ParallelBatchDatasetParams
|
| D | window_dataset_op_test.cc | 36 drop_remainder_(drop_remainder) { in WindowDatasetParams()
47 CreateTensor<bool>(TensorShape({}), {drop_remainder_})}; in GetInputTensors()
73 bool drop_remainder_; member in tensorflow::data::__anon5d0c34070111::WindowDatasetParams
|
| D | padded_batch_dataset_op_test.cc | 40 drop_remainder_(drop_remainder), in PaddedBatchDatasetParams()
61 CreateTensor<bool>(TensorShape({}), {drop_remainder_})); in GetInputTensors()
99 bool drop_remainder_; member in tensorflow::data::__anonaa3739520111::PaddedBatchDatasetParams
|
| D | dataset_test_base.h | 216 drop_remainder_(drop_remainder), in BatchDatasetParams()
235 bool drop_remainder_; variable
|
| D | dataset_test_base.cc | 1013 CreateTensor<bool>(TensorShape({}), {drop_remainder_}); in GetInputTensors()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | map_and_batch_dataset_op_test.cc | 38 drop_remainder_(drop_remainder), in MapAndBatchDatasetParams()
54 inputs.emplace_back(CreateTensor<bool>(TensorShape({}), {drop_remainder_})); in GetInputTensors()
93 bool drop_remainder_; member in tensorflow::data::experimental::__anon3550fdfd0111::MapAndBatchDatasetParams
|
| D | rebatch_dataset_op.cc | 320 drop_remainder_(drop_remainder), in Dataset()
362 TF_RETURN_IF_ERROR(b->AddScalar(drop_remainder_, &drop_remainder)); in AsGraphDefInternal()
439 (dataset()->drop_remainder_ && batch_size < desired_batch_size)) { in GetNextInternal()
604 const bool drop_remainder_; member in tensorflow::data::experimental::__anon6916dfc60111::RebatchDatasetV2Op::Dataset
|
| D | map_and_batch_dataset_op.cc | 94 drop_remainder_(drop_remainder), in Dataset()
134 return n / batch_size_ + (n % batch_size_ == 0 || drop_remainder_ ? 0 : 1); in Cardinality()
159 TF_RETURN_IF_ERROR(b->AddScalar(drop_remainder_, &drop_remainder_node)); in AsGraphDefInternal()
264 dataset()->drop_remainder_, result->status, ctx, in GetNextInternal()
683 const bool drop_remainder_; member in tensorflow::data::experimental::MapAndBatchDatasetOp::Dataset
|