| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | batch_dataset_op.cc | 51 bool parallel_copy, const DatasetBase* input, int op_version) in Dataset() argument
61 parallel_copy_(parallel_copy), in Dataset()
68 {"parallel_copy", parallel_copy ? "true" : "false"}}) { in Dataset()
139 AttrValue parallel_copy; in AsGraphDefInternal() local
140 b->BuildAttrValue(parallel_copy_, ¶llel_copy); in AsGraphDefInternal()
143 {{kParallelCopy, parallel_copy}}, output)); in AsGraphDefInternal()
|
| D | padded_batch_dataset_op.cc | 52 bool parallel_copy, std::vector<PartialTensorShape> padded_shapes, in Dataset() argument
58 parallel_copy_(parallel_copy), in Dataset()
163 AttrValue parallel_copy; in AsGraphDefInternal() local
164 b->BuildAttrValue(parallel_copy_, ¶llel_copy); in AsGraphDefInternal()
175 {{kParallelCopy, parallel_copy}, in AsGraphDefInternal()
|
| D | dataset_utils.cc | 847 Status CopyBatch(bool parallel_copy, IteratorContext* ctx, in CopyBatch() argument
882 if (TF_PREDICT_FALSE(parallel_copy)) { in CopyBatch()
896 if (TF_PREDICT_FALSE(parallel_copy)) { in CopyBatch()
910 if (TF_PREDICT_FALSE(parallel_copy)) { in CopyBatch()
|
| D | padded_batch_dataset_op_test.cc | 31 bool drop_remainder, bool parallel_copy, in PaddedBatchDatasetParams() argument
41 parallel_copy_(parallel_copy), in PaddedBatchDatasetParams()
|
| D | dataset_utils.h | 329 Status CopyBatch(bool parallel_copy, IteratorContext* ctx,
|
| D | dataset_test_base.h | 209 bool drop_remainder, bool parallel_copy, in BatchDatasetParams() argument
217 parallel_copy_(parallel_copy) { in BatchDatasetParams()
|
| /external/tensorflow/tensorflow/python/data/benchmarks/ |
| D | batch_benchmark.py | 55 for parallel_copy in [True, False]:
61 options.experimental_optimization.parallel_batch = parallel_copy
63 tag = "_parallel" if parallel_copy else ""
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | BatchDatasetV2.pbtxt | 51 name: "parallel_copy"
|
| D | PaddedBatchDatasetV2.pbtxt | 75 name: "parallel_copy"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | BatchDatasetV2.pbtxt | 51 name: "parallel_copy"
|
| D | PaddedBatchDatasetV2.pbtxt | 75 name: "parallel_copy"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tf_data_optimization.td | 27 $batch_size, $drop_remainder, $parallel_copy, $batch_output_types,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | tf_data_fuse_map_and_batch.mlir | 19 …1) {device = "", output_shapes = [#tf.shape<>], output_types = [i32], parallel_copy = false} : (te…
|
| D | tf_data_fuse_pmap_and_batch.mlir | 19 …1) {device = "", output_shapes = [#tf.shape<>], output_types = [i32], parallel_copy = false} : (te…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
| D | tf-data-pipeline.pbtxt | 172 key: "parallel_copy"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops.td | 1083 DefaultValuedAttr<BoolAttr, "false">:$parallel_copy,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 333 …ch_size\', \'drop_remainder\', \'output_types\', \'output_shapes\', \'parallel_copy\', \'name\'], …
2789 …hapes\', \'padding_values\', \'drop_remainder\', \'output_shapes\', \'parallel_copy\', \'name\'], …
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 333 …ch_size\', \'drop_remainder\', \'output_types\', \'output_shapes\', \'parallel_copy\', \'name\'], …
2789 …hapes\', \'padding_values\', \'drop_remainder\', \'output_shapes\', \'parallel_copy\', \'name\'], …
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 3468 name: "parallel_copy"
27623 name: "parallel_copy"
|