Searched defs:next_batch_size (Results 1 – 7 of 7) sorted by relevance
257 inline UnpoolingOperatorTester& next_batch_size(size_t next_batch_size) { in next_batch_size() argument263 inline size_t next_batch_size() const { in next_batch_size() function
307 inline ArgmaxPoolingOperatorTester& next_batch_size(size_t next_batch_size) { in next_batch_size() function313 inline size_t next_batch_size() const { in next_batch_size() function
184 inline ResizeBilinearOperatorTester& next_batch_size(size_t next_batch_size) { in next_batch_size() function190 inline size_t next_batch_size() const { in next_batch_size() function
368 inline AveragePoolingOperatorTester& next_batch_size(size_t next_batch_size) { in next_batch_size() argument374 inline size_t next_batch_size() const { in next_batch_size() function
410 inline MaxPoolingOperatorTester& next_batch_size(size_t next_batch_size) { in next_batch_size() function416 inline size_t next_batch_size() const { in next_batch_size() function
380 inline DeconvolutionOperatorTester& next_batch_size(size_t next_batch_size) { in next_batch_size() argument386 inline size_t next_batch_size() const { in next_batch_size() function
439 inline ConvolutionOperatorTester& next_batch_size(size_t next_batch_size) { in next_batch_size() function445 inline size_t next_batch_size() const { in next_batch_size() function