Searched defs:allowed_batch_sizes (Results 1 – 5 of 5) sorted by relevance
118 ArrayRef<int32_t> allowed_batch_sizes, in Create()141 int32_t max_enqueued_batches, ArrayRef<int32_t> allowed_batch_sizes, in Create()178 ArrayRef<int32_t> allowed_batch_sizes) in FallbackBatchResource()193 ArrayRef<int32_t> allowed_batch_sizes) in FallbackBatchResource()
137 const std::vector<int32>& allowed_batch_sizes, in Create()161 const std::vector<int32>& allowed_batch_sizes, in Create()183 std::vector<int32> allowed_batch_sizes) in BatchResource()195 std::vector<int32> allowed_batch_sizes) in BatchResource()
132 std::vector<int32> allowed_batch_sizes) in BatchResourceBase()143 std::vector<int32> allowed_batch_sizes) in BatchResourceBase()
191 void RecordBatchParamAllowedBatchSizes(const string& allowed_batch_sizes, in RecordBatchParamAllowedBatchSizes()333 const std::vector<int32>& allowed_batch_sizes, in GetBatcherQueueOptions()365 const std::vector<int32>& allowed_batch_sizes) { in GetAdaptiveBatcherQueueOptions()
32 allowed_batch_sizes=None, argument