| /external/tensorflow/tensorflow/python/ops/ |
| D | batch_ops.py | 35 enable_large_batch_splitting=True): argument
111 enable_large_batch_splitting=enable_large_batch_splitting,
|
| D | batch_ops_test.py | 481 enable_large_batch_splitting=True,
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | basic_batch_scheduler.h | 225 bool enable_large_batch_splitting = false; member
330 shared_scheduler_queue_options.enable_large_batch_splitting = in Create()
331 options.enable_large_batch_splitting; in Create()
|
| D | shared_batch_scheduler.h | 183 bool enable_large_batch_splitting = false; member
379 if (options.enable_large_batch_splitting) { in GetMaxExecutionBatchSize()
556 if ((!rewrite_options.enable_large_batch_splitting) && in AddQueue()
592 if (options.enable_large_batch_splitting && in AddQueueAfterRewritingOptions()
597 options.enable_large_batch_splitting); in AddQueueAfterRewritingOptions()
600 if (options.enable_lazy_split && (!options.enable_large_batch_splitting)) { in AddQueueAfterRewritingOptions()
606 if (options.enable_large_batch_splitting && in AddQueueAfterRewritingOptions()
612 options.enable_large_batch_splitting, options.input_batch_size_limit, in AddQueueAfterRewritingOptions()
879 const bool large_batch_splitting = options_.enable_large_batch_splitting; in ScheduleWithoutOrEagerSplit()
992 if (options_.enable_large_batch_splitting) { in BatchTaskExceedQueueCapacity()
|
| D | batch_resource_base.h | 153 bool enable_large_batch_splitting);
157 int32_t max_enqueued_batches, bool enable_large_batch_splitting,
|
| D | batch_resource_base.cc | 334 bool enable_large_batch_splitting) { in GetBatcherQueueOptions() argument
339 batcher_queue_options.enable_large_batch_splitting = in GetBatcherQueueOptions()
340 enable_large_batch_splitting; in GetBatcherQueueOptions()
341 if (enable_large_batch_splitting) { in GetBatcherQueueOptions()
364 int32_t max_enqueued_batches, bool enable_large_batch_splitting, in GetAdaptiveBatcherQueueOptions() argument
377 if (enable_large_batch_splitting) { in GetAdaptiveBatcherQueueOptions()
|
| D | shared_batch_scheduler_test.cc | 120 bool enable_large_batch_splitting, in CreateQueueOptions() argument
127 queue_options.enable_large_batch_splitting = enable_large_batch_splitting; in CreateQueueOptions()
129 if (enable_large_batch_splitting) { in CreateQueueOptions()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/ |
| D | batch_function_deduplicate.mlir | 10 … "", device = "/job:localhost/replica:0/task:0/device:CPU:0", enable_large_batch_splitting = true,…
25 … "", device = "/job:localhost/replica:0/task:0/device:CPU:0", enable_large_batch_splitting = true,…
39 … "", device = "/job:localhost/replica:0/task:0/device:CPU:0", enable_large_batch_splitting = true,…
|
| D | batch_function_deduplicate_failed.mlir | 7 … "", device = "/job:localhost/replica:0/task:0/device:CPU:0", enable_large_batch_splitting = true,…
16 … "", device = "/job:localhost/replica:0/task:0/device:CPU:0", enable_large_batch_splitting = true,…
|
| D | batch_function_fallback.mlir | 23 // CHECK-SAME: enable_large_batch_splitting = false
28 …atching_queue = "", container = "", device = "/device:CPU:0", enable_large_batch_splitting = false…
|
| D | lower_saved_model.mlir | 149 …atching_queue = "", container = "", device = "/device:CPU:0", enable_large_batch_splitting = false…
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | BatchFunction.pbtxt | 169 name: "enable_large_batch_splitting"
260 name: "enable_large_batch_splitting"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BatchFunction.pbtxt | 54 If enable_large_batch_splitting is false (i.e., large-input-split is not
91 name: "enable_large_batch_splitting"
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | runtime_fallback_batch_tf_opkernels.cc | 120 bool enable_large_batch_splitting, in Create() argument
133 enable_large_batch_splitting), in Create()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 140 bool enable_large_batch_splitting, in Create() argument
152 enable_large_batch_splitting), in Create()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/opdefs/ |
| D | tfrt_fallback_async.td | 457 enable_large_batch_splitting = false,
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/ir/ |
| D | tfrt_fallback_async.td | 456 enable_large_batch_splitting = false,
|
| /external/tensorflow/tensorflow/core/ir/importexport/tests/roundtrip/ |
| D | test13.pbtxt | 525 key: "enable_large_batch_splitting"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.pbtxt | 845 …wed_batch_sizes\', \'max_enqueued_batches\', \'autograph\', \'enable_large_batch_splitting\'], var…
|
| D | tensorflow.raw_ops.pbtxt | 385 …izes\', \'container\', \'shared_name\', \'batching_queue\', \'enable_large_batch_splitting\', \'na…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.pbtxt | 1745 …wed_batch_sizes\', \'max_enqueued_batches\', \'autograph\', \'enable_large_batch_splitting\'], var…
|
| D | tensorflow.raw_ops.pbtxt | 385 …izes\', \'container\', \'shared_name\', \'batching_queue\', \'enable_large_batch_splitting\', \'na…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 4044 name: "enable_large_batch_splitting"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 1096 DefaultValuedAttr<BoolAttr, "false">:$enable_large_batch_splitting
|