Home
last modified time | relevance | path

Searched refs:parallel_copy_ (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dbatch_dataset_op.cc61 parallel_copy_(parallel_copy), in Dataset()
140 b->BuildAttrValue(parallel_copy_, &parallel_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, &parallel_copy_)); in BatchDatasetOp()
277 *output = new Dataset(ctx, batch_size, drop_remainder, parallel_copy_, input, in MakeDataset()
Dpadded_batch_dataset_op.cc58 parallel_copy_(parallel_copy), in Dataset()
164 b->BuildAttrValue(parallel_copy_, &parallel_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, &parallel_copy_)); in PaddedBatchDatasetOp()
456 *output = new Dataset(ctx, batch_size, drop_remainder, parallel_copy_, in MakeDataset()
Dbatch_dataset_op.h42 bool parallel_copy_ = false; variable
Dpadded_batch_dataset_op.h45 bool parallel_copy_ = false; variable
Dpadded_batch_dataset_op_test.cc41 parallel_copy_(parallel_copy), in PaddedBatchDatasetParams()
84 {PaddedBatchDatasetOp::kParallelCopy, parallel_copy_}, in GetAttributes()
100 bool parallel_copy_; member in tensorflow::data::__anon1aa214390111::PaddedBatchDatasetParams
Ddataset_test_base.h217 parallel_copy_(parallel_copy) { in BatchDatasetParams()
236 bool parallel_copy_; variable
Ddataset_test_base.cc1025 *attr_vector = {{BatchDatasetOp::kParallelCopy, parallel_copy_}, in GetAttributes()