| /external/tensorflow/tensorflow/python/data/benchmarks/ |
| D | batch_benchmark.py | 53 def _benchmark_batch_dense(self, parallel_copy, benchmark_id): argument
61 options.experimental_optimization.parallel_batch = parallel_copy
63 tag = "_parallel_copy" if parallel_copy else ""
76 self._benchmark_batch_dense(parallel_copy=False, benchmark_id=2)
77 self._benchmark_batch_dense(parallel_copy=True, benchmark_id=3)
|
| /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()
169 AttrValue parallel_copy; in AsGraphDefInternal() local
170 b->BuildAttrValue(parallel_copy_, ¶llel_copy); in AsGraphDefInternal()
173 {{kParallelCopy, parallel_copy}}, output)); in AsGraphDefInternal()
|
| D | padded_batch_dataset_op.cc | 53 bool parallel_copy, std::vector<PartialTensorShape> padded_shapes, in Dataset() argument
59 parallel_copy_(parallel_copy), in Dataset()
68 {"parallel_copy", parallel_copy ? "true" : "false"}}) { in Dataset()
165 AttrValue parallel_copy; in AsGraphDefInternal() local
166 b->BuildAttrValue(parallel_copy_, ¶llel_copy); in AsGraphDefInternal()
177 {{kParallelCopy, parallel_copy}, in AsGraphDefInternal()
|
| D | parallel_batch_dataset_op_test.cc | 33 const bool parallel_copy, in ParallelBatchDatasetParams() argument
40 parallel_copy_(parallel_copy), in ParallelBatchDatasetParams()
|
| D | parallel_batch_dataset_op.cc | 72 bool drop_remainder, bool parallel_copy, const DatasetBase* input, in Dataset() argument
84 parallel_copy_(parallel_copy), in Dataset()
93 {"parallel_copy", parallel_copy ? "true" : "false"}}) { in Dataset()
|
| D | padded_batch_dataset_op_test.cc | 31 bool drop_remainder, bool parallel_copy, in PaddedBatchDatasetParams() argument
41 parallel_copy_(parallel_copy), in PaddedBatchDatasetParams()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | BatchDatasetV2.pbtxt | 51 name: "parallel_copy"
89 name: "parallel_copy"
141 name: "parallel_copy"
206 name: "parallel_copy"
|
| D | PaddedBatchDatasetV2.pbtxt | 75 name: "parallel_copy"
128 name: "parallel_copy"
195 name: "parallel_copy"
275 name: "parallel_copy"
|
| D | ParallelBatchDataset.pbtxt | 101 name: "parallel_copy"
150 name: "parallel_copy"
213 name: "parallel_copy"
289 name: "parallel_copy"
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | batch_parallelization_test.cc | 117 const bool parallel_copy = GetParam(); in TEST_P() local
130 parallel_copy), in TEST_P()
151 EXPECT_EQ(parallel_batch.attr().at("parallel_copy").b(), parallel_copy); in TEST_P()
|
| D | graph_test_utils.cc | 31 bool parallel_copy) { in MakeBatchV2Node() argument
36 {{"parallel_copy", parallel_copy}, in MakeBatchV2Node()
|
| D | graph_test_utils.h | 30 bool 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 | 29 $batch_size, $drop_remainder, $parallel_copy, $batch_output_types,
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/tf_to_tfrt_data/ |
| D | batch.mlir | 21 …remainder) {output_shapes = [#tf_type.shape<>], output_types = [i64], parallel_copy = false, metad…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | tf_data_fuse_map_and_batch.mlir | 23 parallel_copy = false,
|
| D | tf_data_fuse_pmap_and_batch.mlir | 23 parallel_copy = false,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
| D | tf-data-pipeline.pbtxt | 172 key: "parallel_copy"
|
| /external/tensorflow/tensorflow/core/ir/importexport/tests/roundtrip/ |
| D | test25.pbtxt | 680 key: "parallel_copy"
940 key: "parallel_copy"
|
| /external/tensorflow/tensorflow/core/data/ |
| D | dataset_utils.h | 310 bool parallel_copy,
|
| D | dataset_utils.cc | 742 bool parallel_copy, in CopyBatch() argument
788 if (parallel_copy && first_element.AllocatedBytes() > (1 << 15)) { in CopyBatch()
|
| D | dataset_test_base.h | 226 bool drop_remainder, bool parallel_copy, in BatchDatasetParams() argument
234 parallel_copy_(parallel_copy) { in BatchDatasetParams()
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 369 …ch_size\', \'drop_remainder\', \'output_types\', \'output_shapes\', \'parallel_copy\', \'metadata\…
2857 …hapes\', \'padding_values\', \'drop_remainder\', \'output_shapes\', \'parallel_copy\', \'metadata\…
2869 …l_calls\', \'drop_remainder\', \'output_types\', \'output_shapes\', \'parallel_copy\', \'determini…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 369 …ch_size\', \'drop_remainder\', \'output_types\', \'output_shapes\', \'parallel_copy\', \'metadata\…
2857 …hapes\', \'padding_values\', \'drop_remainder\', \'output_shapes\', \'parallel_copy\', \'metadata\…
2869 …l_calls\', \'drop_remainder\', \'output_types\', \'output_shapes\', \'parallel_copy\', \'determini…
|