Searched refs:limit_on_batch_size (Results 1 – 2 of 2) sorted by relevance
29 explicit ConvolutionSpaceToBatchConverter(int64 limit_on_batch_size = 1)30 : limit_on_batch_size_(limit_on_batch_size) {} in limit_on_batch_size_() argument
190 explicit ConvolutionVisitor(int64 limit_on_batch_size,253 ConvolutionVisitor::ConvolutionVisitor(int64 limit_on_batch_size, in ConvolutionVisitor() argument256 limit_on_batch_size_ = limit_on_batch_size; in ConvolutionVisitor()