Searched defs:allowed_batch_sizes (Results 1 – 5 of 5) sorted by relevance
132 std::vector<int32> allowed_batch_sizes) in BatchResourceBase()143 std::vector<int32> allowed_batch_sizes) in BatchResourceBase()
174 void RecordBatchParamAllowedBatchSizes(const string& allowed_batch_sizes, in RecordBatchParamAllowedBatchSizes()296 const std::vector<int32>& allowed_batch_sizes, in GetBatcherQueueOptions()328 const std::vector<int32>& allowed_batch_sizes) { in GetAdaptiveBatcherQueueOptions()
36 allowed_batch_sizes=None, argument
76 ArrayRef<int32_t> allowed_batch_sizes, in Create()116 ArrayRef<int32_t> allowed_batch_sizes) in FallbackBatchResource()
110 const std::vector<int32>& allowed_batch_sizes, in Create()134 const std::vector<int32>& allowed_batch_sizes, in Create()156 std::vector<int32> allowed_batch_sizes) in BatchResource()168 std::vector<int32> allowed_batch_sizes) in BatchResource()