Searched defs:parallel_copy (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | batch_dataset_op.cc | 51 bool parallel_copy, const DatasetBase* input, int op_version) in Dataset()
169 AttrValue parallel_copy; in AsGraphDefInternal() local
|
| H A D | padded_batch_dataset_op.cc | 53 bool parallel_copy, std::vector<PartialTensorShape> padded_shapes, in Dataset()
165 AttrValue parallel_copy; in AsGraphDefInternal() local
|
| H A D | parallel_batch_dataset_op_test.cc | 33 const bool parallel_copy, in ParallelBatchDatasetParams()
|
| H A D | padded_batch_dataset_op_test.cc | 31 bool drop_remainder, bool parallel_copy, in PaddedBatchDatasetParams()
|
| H A D | parallel_batch_dataset_op.cc | 72 bool drop_remainder, bool parallel_copy, const DatasetBase* input, in Dataset()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/benchmarks/ |
| H A D | batch_benchmark.py | 53 def _benchmark_batch_dense(self, parallel_copy, benchmark_id): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| H A D | batch_parallelization_test.cc | 117 const bool parallel_copy = GetParam(); in TEST_P() local
|
| H A D | graph_test_utils.cc | 31 bool parallel_copy) { in MakeBatchV2Node()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | dataset_test_base.h | 226 bool drop_remainder, bool parallel_copy, in BatchDatasetParams()
|
| H A D | dataset_utils.cc | 741 bool parallel_copy, in CopyBatch()
|