Searched defs:num_batch_threads (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/inference/ |
| H A D | batch_op_rewriter_test.cc | 46 GraphDef* graph, int num_batch_threads, in AddBatchOp()
49 NodeDef* batch_op) { in __anon11d085b80202()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | batch_kernels.cc | 82 void RecordBatchParamNumBatchThreads(int64_t num_batch_threads, in RecordBatchParamNumBatchThreads()
133 static Status Create(int32_t num_batch_threads, in Create()
471 OpKernelConstruction* c, int32_t num_batch_threads) { in SetAdaptiveBatchSchedulerOptions()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| H A D | runtime_fallback_batch_tf_opkernels.cc | 115 static Status Create(int32_t num_batch_threads, int32_t max_batch_size, in Create()
429 OpKernelConstruction* c, int32_t num_batch_threads) { in SetAdaptiveBatchSchedulerOptions()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | batch_ops.py | 29 def batch_function(num_batch_threads, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/ |
| H A D | shared_batch_scheduler_test.cc | 92 int num_batch_threads, Env* env = Env::Default()) { in CreateSharedBatchScheduler()
|
| H A D | batch_resource_base.cc | 331 int32_t num_batch_threads, int32_t max_batch_size, in GetBatcherQueueOptions()
|