| 52 | } |
| 53 | |
| 54 | Tensor Composite(Tensor &foreground, Tensor &background, Tensor &fgMask, int outChannels, std::optional<Stream> pstream) |
| 55 | { |
| 56 | Shape out_shape = CreateShape(foreground.shape()); |
| 57 | out_shape[out_shape.size() - 1] = outChannels; |
| 58 | |
| 59 | Tensor output = Tensor::Create(out_shape, foreground.dtype(), foreground.layout()); |
| 60 | |
| 61 | return CompositeInto(output, foreground, background, fgMask, pstream); |
| 62 | } |
| 63 | |
| 64 | ImageBatchVarShape CompositeVarShapeInto(ImageBatchVarShape &output, ImageBatchVarShape &foreground, |
| 65 | ImageBatchVarShape &background, ImageBatchVarShape &fgMask, |
nothing calls this directly
no test coverage detected