Searched refs:kParallelCopy (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | batch_dataset_op.cc | 41 /* static */ constexpr const char* const BatchDatasetOp::kParallelCopy; member in tensorflow::data::BatchDatasetOp
143 {{kParallelCopy, parallel_copy}}, output)); in AsGraphDefInternal()
259 if (ctx->HasAttr(kParallelCopy)) { in BatchDatasetOp()
260 OP_REQUIRES_OK(ctx, ctx->GetAttr(kParallelCopy, ¶llel_copy_)); in BatchDatasetOp()
|
| D | batch_dataset_op.h | 29 static constexpr const char* const kParallelCopy = "parallel_copy"; variable
|
| D | padded_batch_dataset_op.h | 31 static constexpr const char* const kParallelCopy = "parallel_copy"; variable
|
| D | padded_batch_dataset_op.cc | 42 /* static */ constexpr const char* const PaddedBatchDatasetOp::kParallelCopy; member in tensorflow::data::PaddedBatchDatasetOp
175 {{kParallelCopy, parallel_copy}, in AsGraphDefInternal()
397 if (ctx->HasAttr(kParallelCopy)) { in PaddedBatchDatasetOp()
398 OP_REQUIRES_OK(ctx, ctx->GetAttr(kParallelCopy, ¶llel_copy_)); in PaddedBatchDatasetOp()
|
| D | padded_batch_dataset_op_test.cc | 84 {PaddedBatchDatasetOp::kParallelCopy, parallel_copy_}, in GetAttributes()
|
| D | dataset_test_base.cc | 1025 *attr_vector = {{BatchDatasetOp::kParallelCopy, parallel_copy_}, in GetAttributes()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | batch_parallelization.cc | 36 constexpr char kParallelCopy[] = "parallel_copy"; variable
53 parallel_batch.mutable_attr()->erase(kParallelCopy); in MakeParallelBatch()
|