Home
last modified time | relevance | path

Searched refs:parallel_batch (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A Dbatch_parallelization.cc43 NodeDef parallel_batch = graph->graph()->node(index); in MakeParallelBatch() local
45 &parallel_batch); in MakeParallelBatch()
46 parallel_batch.set_op(kParallelBatchDataset); in MakeParallelBatch()
49 string drop_remainder_name = parallel_batch.input(2); in MakeParallelBatch()
50 parallel_batch.set_input(2, num_parallel_calls->name()); in MakeParallelBatch()
51 parallel_batch.add_input(drop_remainder_name); in MakeParallelBatch()
53 return parallel_batch; in MakeParallelBatch()
87 auto* parallel_batch = in OptimizeAndCollectStats() local
90 graph.UpdateFanouts(batch_node->name(), parallel_batch->name())); in OptimizeAndCollectStats()
H A Dbatch_parallelization_test.cc145 NodeDef parallel_batch = output.node( in TEST_P() local
147 EXPECT_EQ(parallel_batch.input_size(), 4); in TEST_P()
148 EXPECT_EQ(parallel_batch.input(0), "range"); in TEST_P()
149 EXPECT_EQ(parallel_batch.input(1), "batch_size"); in TEST_P()
150 EXPECT_EQ(parallel_batch.input(3), "drop_remainder"); in TEST_P()
151 EXPECT_EQ(parallel_batch.attr().at("parallel_copy").b(), parallel_copy); in TEST_P()
154 graph_utils::FindGraphNodeWithName(parallel_batch.input(2), output)); in TEST_P()
H A DBUILD36 ":parallel_batch",
814 name = "parallel_batch",
815 srcs = ["parallel_batch.cc"],
816 hdrs = ["parallel_batch.h"],
835 ":parallel_batch",
/aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/
H A Doptions.py378 parallel_batch = options_lib.create_option( variable in OptimizationOptions
410 if self.parallel_batch is not None:
411 pb.parallel_batch = self.parallel_batch
436 self.parallel_batch = pb.parallel_batch
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.data.experimental.-optimization-options.pbtxt43 name: "parallel_batch"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.data.experimental.-optimization-options.pbtxt43 name: "parallel_batch"
/aosp_15_r20/external/tensorflow/tensorflow/python/data/benchmarks/
H A Dbatch_benchmark.py61 options.experimental_optimization.parallel_batch = parallel_copy
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Ddataset_options.proto135 bool parallel_batch = 15; field
/aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/
H A Doptions_test.py156 options.experimental_optimization.parallel_batch = True
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Ddataset_utils.cc182 if (optimization_options.parallel_batch()) { in DefaultOptimizationGraphRewrites()
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/framework/
Ddataset_options.pb.h994 bool parallel_batch() const; in Swap()
2249 inline bool OptimizationOptions::parallel_batch() const { in parallel_batch() function
/aosp_15_r20/external/tensorflow/
H A DRELEASE.md525 * The optimization `parallel_batch` now becomes default if not disabled by