Home
last modified time | relevance | path

Searched defs:parallel_batch (Results 1 – 5 of 5) 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
87 auto* parallel_batch = in OptimizeAndCollectStats() local
H A Dbatch_parallelization_test.cc145 NodeDef parallel_batch = output.node( in TEST_P() local
/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/ops/
H A Doptions.py378 parallel_batch = options_lib.create_option( variable in OptimizationOptions
/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.h2249 inline bool OptimizationOptions::parallel_batch() const { in parallel_batch() function