Searched defs:allowed_batch_sizes (Results 1 – 4 of 4) sorted by relevance
119 std::vector<int32> allowed_batch_sizes) in BatchResourceBase()130 std::vector<int32> allowed_batch_sizes) in BatchResourceBase()
111 void RecordBatchParamAllowedBatchSizes(const string& allowed_batch_sizes, in RecordBatchParamAllowedBatchSizes()229 int32 max_enqueued_batches, const std::vector<int32>& allowed_batch_sizes, in GetBatcherQueueOptions()261 const std::vector<int32>& allowed_batch_sizes) { in GetAdaptiveBatcherQueueOptions()
35 allowed_batch_sizes=None, argument
88 const std::vector<int32>& allowed_batch_sizes, in Create()111 int32 max_enqueued_batches, const std::vector<int32>& allowed_batch_sizes, in Create()133 std::vector<int32> allowed_batch_sizes) in BatchResource()145 std::vector<int32> allowed_batch_sizes) in BatchResource()