Home
last modified time | relevance | path

Searched defs:num_batch_threads (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/inference/
Dbatch_op_rewriter_test.cc43 void AddBatchOp(GraphDef* graph, int num_batch_threads) { in AddBatchOp()
45 NodeDef* batch_op) { in AddBatchOp()
/external/tensorflow/tensorflow/python/ops/
Dbatch_ops.py32 def batch_function(num_batch_threads, argument
/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbasic_batch_scheduler_benchmark_test.cc339 int num_batch_threads) { in RunThroughputBenchmark()
Dbatch_resource_base.cc228 int32 num_batch_threads, int32 max_batch_size, int32 batch_timeout_micros, in GetBatcherQueueOptions()
/external/tensorflow/tensorflow/core/kernels/
Dbatch_kernels.cc65 void RecordBatchParamNumBatchThreads(int64 num_batch_threads, in RecordBatchParamNumBatchThreads()
86 static Status Create(int32 num_batch_threads, int32 max_execution_batch_size, in Create()