Searched refs:parallel_batch (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | batch_parallelization.cc | 44 NodeDef parallel_batch = graph->graph()->node(index); in MakeParallelBatch() local
46 ¶llel_batch); in MakeParallelBatch()
47 parallel_batch.set_op(kParallelBatchDataset); in MakeParallelBatch()
50 string drop_remainder_name = parallel_batch.input(2); in MakeParallelBatch()
51 parallel_batch.set_input(2, num_parallel_calls->name()); in MakeParallelBatch()
52 parallel_batch.add_input(drop_remainder_name); in MakeParallelBatch()
53 parallel_batch.mutable_attr()->erase(kParallelCopy); in MakeParallelBatch()
55 return parallel_batch; in MakeParallelBatch()
90 auto* parallel_batch = in OptimizeAndCollectStats() local
93 graph.UpdateFanouts(batch_node->name(), parallel_batch->name())); in OptimizeAndCollectStats()
|
| D | batch_parallelization_test.cc | 139 NodeDef parallel_batch = output.node( in TEST() local
141 EXPECT_EQ(parallel_batch.input_size(), 4); in TEST()
142 EXPECT_EQ(parallel_batch.input(0), "range"); in TEST()
143 EXPECT_EQ(parallel_batch.input(1), "batch_size"); in TEST()
144 EXPECT_EQ(parallel_batch.input(3), "drop_remainder"); in TEST()
145 EXPECT_TRUE(parallel_batch.attr().find("parallel_copy") == in TEST()
146 parallel_batch.attr().end()); in TEST()
149 graph_utils::FindGraphNodeWithName(parallel_batch.input(2), output)); in TEST()
|
| D | BUILD | 35 ":parallel_batch",
816 name = "parallel_batch",
817 srcs = ["parallel_batch.cc"],
818 hdrs = ["parallel_batch.h"],
836 ":parallel_batch",
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | optimization_options.py | 204 parallel_batch = options.create_option( variable in OptimizationOptions
378 if self.parallel_batch is not None:
379 pb.parallel_batch = self.parallel_batch
416 self.parallel_batch = pb.parallel_batch
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.experimental.-optimization-options.pbtxt | 63 name: "parallel_batch"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.experimental.-optimization-options.pbtxt | 63 name: "parallel_batch"
|
| /external/tensorflow/tensorflow/python/data/benchmarks/ |
| D | batch_benchmark.py | 61 options.experimental_optimization.parallel_batch = parallel_copy
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | options_test.py | 157 options.experimental_optimization.parallel_batch = True
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | optimize_dataset_test.py | 465 options.experimental_optimization.parallel_batch = True
513 options.experimental_optimization.parallel_batch = False
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | dataset_options.proto | 116 bool parallel_batch = 15; field
|