Searched defs:num_batch_threads (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/inference/ |
| D | batch_op_rewriter_test.cc | 46 GraphDef* graph, int num_batch_threads, in AddBatchOp()
49 NodeDef* batch_op) { in __anonaa493bee0202()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | batch_ops.py | 33 def batch_function(num_batch_threads, argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 78 void RecordBatchParamNumBatchThreads(int64_t num_batch_threads, in RecordBatchParamNumBatchThreads()
97 const string& thread_name, int num_batch_threads) { in GetOrCreateBatchThreadsPool()
106 static Status Create(int32_t num_batch_threads, in Create()
451 int32_t num_batch_threads) { in SetAdaptiveBatchSchedulerOptions()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | runtime_fallback_batch_tf_opkernels.cc | 73 static Status Create(int32_t num_batch_threads, int32_t max_batch_size, in Create()
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_resource_base.cc | 294 int32_t num_batch_threads, int32_t max_batch_size, in GetBatcherQueueOptions()
|
| D | shared_batch_scheduler_test.cc | 91 int num_batch_threads, Env* env = Env::Default()) { in CreateSharedBatchScheduler()
|