Searched refs:drop_remainder_ (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | batch_dataset_op.cc | 58 drop_remainder_(drop_remainder), in Dataset()
68 if (drop_remainder_) { in Dataset()
104 (n % batch_size_ == 0 || drop_remainder_ ? 0 : 1); in Cardinality()
116 TF_RETURN_IF_ERROR(b->AddScalar(drop_remainder_, &drop_remainder)); in AsGraphDefInternal()
164 if (dataset()->drop_remainder_ && in GetNextInternal()
249 const bool drop_remainder_; member in tensorflow::data::__anonb889a47b0111::BatchDatasetOp::Dataset
|
| D | padded_batch_dataset_op.cc | 108 drop_remainder_(drop_remainder), in Dataset()
124 if (drop_remainder_) { in Dataset()
161 (n % batch_size_ == 0 || drop_remainder_ ? 0 : 1); in Cardinality()
194 TF_RETURN_IF_ERROR(b->AddScalar(drop_remainder_, &drop_remainder)); in AsGraphDefInternal()
256 if (dataset()->drop_remainder_ && in GetNextInternal()
383 const bool drop_remainder_; member in tensorflow::data::__anone2805d3c0111::PaddedBatchDatasetOp::Dataset
|
| D | window_dataset_op.cc | 73 drop_remainder_(drop_remainder) { in Dataset()
98 window_stride_, drop_remainder_, ")::Dataset"); in DebugString()
107 (n % window_shift_ == 0 || drop_remainder_ ? 0 : 1); in Cardinality()
123 TF_RETURN_IF_ERROR(b->AddScalar(drop_remainder_, &drop_remainder_node)); in AsGraphDefInternal()
178 (dataset()->drop_remainder_ && buffer_.size() < target_size)) { in GetNextInternal()
370 const bool drop_remainder_; member in tensorflow::data::__anon256b48110111::WindowDatasetOp::Dataset
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | numa_map_and_batch_dataset_op.cc | 111 drop_remainder_(drop_remainder), in Dataset()
147 (n % batch_size_ == 0 || drop_remainder_ ? 0 : 1); in Cardinality()
163 TF_RETURN_IF_ERROR(b->AddScalar(drop_remainder_, &drop_remainder_node)); in AsGraphDefInternal()
251 Status s = worker->manager.GetBatch(ctx, dataset()->drop_remainder_, in GetNextInternal()
1003 if (dataset()->drop_remainder_) { in RunnerThread()
1141 const bool drop_remainder_; member in tensorflow::data::__anon9271f8770111::NumaMapAndBatchDatasetOp::Dataset
|
| D | map_and_batch_dataset_op.cc | 159 drop_remainder_(drop_remainder), in Dataset()
196 (n % batch_size_ == 0 || drop_remainder_ ? 0 : 1); in Cardinality()
212 TF_RETURN_IF_ERROR(b->AddScalar(drop_remainder_, &drop_remainder_node)); in AsGraphDefInternal()
570 if (dataset()->drop_remainder_) { in ProcessResult()
814 const bool drop_remainder_; member in tensorflow::data::__anon2c387ec70111::MapAndBatchDatasetOp::Dataset
|