| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | sliding_window_dataset_op.cc | 36 OP_REQUIRES_OK(ctx, ctx->GetAttr(kDropRemainder, &drop_remainder_)); in SlidingWindowDatasetOp()
66 drop_remainder_, input); in MakeDataset()
79 drop_remainder_(drop_remainder), in Dataset()
110 drop_remainder_, ")::Dataset"); in DebugString()
118 return (drop_remainder_ ? n : n + window_shift_ - 1) / window_shift_; in CardinalityInternal()
143 b->BuildAttrValue(drop_remainder_, &drop_remainder_attr); in AsGraphDefInternal()
171 const bool drop_remainder = dataset()->drop_remainder_; in GetNextInternal()
324 const bool drop_remainder_; member in tensorflow::data::experimental::__anon126406330111::SlidingWindowDatasetOp::Dataset
328 bool drop_remainder_ = true; member in tensorflow::data::experimental::__anon126406330111::SlidingWindowDatasetOp
|
| 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::__anond49d67420111::MapAndBatchDatasetParams
|
| D | rebatch_dataset_op.cc | 326 drop_remainder_(drop_remainder), in Dataset()
374 TF_RETURN_IF_ERROR(b->AddScalar(drop_remainder_, &drop_remainder)); in AsGraphDefInternal()
452 (dataset()->drop_remainder_ && batch_size < desired_batch_size)) { in GetNextInternal()
618 const bool drop_remainder_; member in tensorflow::data::experimental::__anon1d26b22b0111::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 CardinalityInternal()
159 TF_RETURN_IF_ERROR(b->AddScalar(drop_remainder_, &drop_remainder_node)); in AsGraphDefInternal()
269 dataset()->drop_remainder_, result->status, ctx, in GetNextInternal()
709 const bool drop_remainder_; member in tensorflow::data::experimental::MapAndBatchDatasetOp::Dataset
|
| /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 CardinalityInternal()
125 return n / batch_size_ + (n % batch_size_ == 0 || drop_remainder_ ? 0 : 1); in CardinalityInternal()
168 TF_RETURN_IF_ERROR(b->AddScalar(drop_remainder_, &drop_remainder)); in AsGraphDefInternal()
218 if (dataset()->drop_remainder_ && in GetNextInternal()
279 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 CardinalityInternal()
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 | 58 drop_remainder_(drop_remainder), in Dataset()
81 if (drop_remainder_ || input_->Cardinality() == kInfiniteCardinality) {
121 return n / batch_size_ + (n % batch_size_ == 0 || drop_remainder_ ? 0 : 1); in CardinalityInternal()
163 TF_RETURN_IF_ERROR(b->AddScalar(drop_remainder_, &drop_remainder)); in AsGraphDefInternal()
228 if (dataset()->drop_remainder_ && in GetNextInternal()
406 const bool drop_remainder_; member in tensorflow::data::PaddedBatchDatasetOp::Dataset
|
| D | parallel_batch_dataset_op.cc | 83 drop_remainder_(drop_remainder), in Dataset()
99 if (drop_remainder_ || input_->Cardinality() == kInfiniteCardinality) {
136 return n / batch_size_ + (n % batch_size_ == 0 || drop_remainder_ ? 0 : 1); in CardinalityInternal()
166 TF_RETURN_IF_ERROR(b->AddScalar(drop_remainder_, &drop_remainder)); in AsGraphDefInternal()
257 dataset()->drop_remainder_, result->status, ctx, in GetNextInternal()
626 const bool drop_remainder_; member in tensorflow::data::ParallelBatchDatasetOp::Dataset
|
| D | parallel_batch_dataset_op_test.cc | 39 drop_remainder_(drop_remainder), in ParallelBatchDatasetParams()
54 CreateTensor<bool>(TensorShape({}), {drop_remainder_}); in GetInputTensors()
84 bool drop_remainder_; member in tensorflow::data::__anon45b721a20111::ParallelBatchDatasetParams
|
| D | window_dataset_op_test.cc | 40 drop_remainder_(drop_remainder) { in WindowDatasetParams()
51 CreateTensor<bool>(TensorShape({}), {drop_remainder_})}; in GetInputTensors()
78 bool drop_remainder_; member in tensorflow::data::__anona061d8cc0111::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::__anone4c89dd70111::PaddedBatchDatasetParams
|
| /external/tensorflow/tensorflow/core/data/ |
| D | dataset_test_base.h | 233 drop_remainder_(drop_remainder), in BatchDatasetParams()
252 bool drop_remainder_; variable
|
| D | dataset_test_base.cc | 1046 CreateTensor<bool>(TensorShape({}), {drop_remainder_}); in GetInputTensors()
|