| 54 | } |
| 55 | |
| 56 | ImageBatchVarShape Conv2DVarShape(ImageBatchVarShape &input, ImageBatchVarShape &kernel, Tensor &kernel_anchor, |
| 57 | NVCVBorderType border, std::optional<Stream> pstream) |
| 58 | { |
| 59 | ImageBatchVarShape output = ImageBatchVarShape::Create(input.capacity()); |
| 60 | |
| 61 | for (int i = 0; i < input.numImages(); ++i) |
| 62 | { |
| 63 | nvcv::ImageFormat format = input[i].format(); |
| 64 | nvcv::Size2D size = input[i].size(); |
| 65 | auto image = Image::Create(size, format); |
| 66 | output.pushBack(image); |
| 67 | } |
| 68 | |
| 69 | return Conv2DVarShapeInto(output, input, kernel, kernel_anchor, border, pstream); |
| 70 | } |
| 71 | |
| 72 | } // namespace |
| 73 | |