Searched refs:parallel_copy_ (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/data/ |
D | batch_dataset_op.cc | 61 parallel_copy_(parallel_copy), in Dataset() 140 b->BuildAttrValue(parallel_copy_, ¶llel_copy); in AsGraphDefInternal() 202 TF_RETURN_IF_ERROR(CopyBatch(/*parallel_copy=*/dataset()->parallel_copy_, in GetNextInternal() 249 const bool parallel_copy_; member in tensorflow::data::BatchDatasetOp::Dataset 260 OP_REQUIRES_OK(ctx, ctx->GetAttr(kParallelCopy, ¶llel_copy_)); in BatchDatasetOp() 277 *output = new Dataset(ctx, batch_size, drop_remainder, parallel_copy_, input, in MakeDataset()
|
D | padded_batch_dataset_op.cc | 58 parallel_copy_(parallel_copy), in Dataset() 164 b->BuildAttrValue(parallel_copy_, ¶llel_copy); in AsGraphDefInternal() 323 if (TF_PREDICT_FALSE(dataset()->parallel_copy_)) { in GetNextInternal() 385 const bool parallel_copy_; member in tensorflow::data::PaddedBatchDatasetOp::Dataset 398 OP_REQUIRES_OK(ctx, ctx->GetAttr(kParallelCopy, ¶llel_copy_)); in PaddedBatchDatasetOp() 456 *output = new Dataset(ctx, batch_size, drop_remainder, parallel_copy_, in MakeDataset()
|
D | batch_dataset_op.h | 42 bool parallel_copy_ = false; variable
|
D | padded_batch_dataset_op.h | 45 bool parallel_copy_ = false; variable
|
D | padded_batch_dataset_op_test.cc | 41 parallel_copy_(parallel_copy), in PaddedBatchDatasetParams() 84 {PaddedBatchDatasetOp::kParallelCopy, parallel_copy_}, in GetAttributes() 100 bool parallel_copy_; member in tensorflow::data::__anon1aa214390111::PaddedBatchDatasetParams
|
D | dataset_test_base.h | 217 parallel_copy_(parallel_copy) { in BatchDatasetParams() 236 bool parallel_copy_; variable
|
D | dataset_test_base.cc | 1025 *attr_vector = {{BatchDatasetOp::kParallelCopy, parallel_copy_}, in GetAttributes()
|