@protocol MTLDevice, MTLLibrary, MTLCommandQueue, MTLComputePipelineState;
@interface Backwarp : NSObject {
id<MTLDevice> _device;
id<MTLCommandQueue> _commandQueue;
id<MTLLibrary> _mtlLibrary;
id<MTLComputePipelineState> _backwarpKernel;
id<MTLComputePipelineState> _backwarpLossKernel;
id<MTLComputePipelineState> _flowUpscaleKernel;
id<MTLComputePipelineState> _backwarpLossWithFlowMagnitudeKernel;
id<MTLComputePipelineState> _subsampleFlowKernel;
id<MTLComputePipelineState> _subsampleErrorKernel;
id<MTLComputePipelineState> _upscaleErrorKernel;
id<MTLComputePipelineState> _subsampleInputKernel;
id<MTLComputePipelineState> _padTextureKernel;
id<MTLComputePipelineState> _reverseFlowKernel;
id<MTLComputePipelineState> _flowSplattingWarpKernel;
id<MTLComputePipelineState> _flowReshuffleKernel;
BOOL _interleaved;
}
- (void)dealloc;
- (void).cxx_destruct;
- (void)setupMetal;
- (void)calcBackwarpLoss:(id)a0 second:(id)a1 flow:(id)a2 timeScale:(float)a3 destination:(id)a4;
- (void)copyTextureWithPaddingSource:(id)a0 destination:(id)a1;
- (void)encodeBackwarpLossToCommandBuffer:(id)a0 first:(id)a1 second:(id)a2 flow:(id)a3 timeScale:(float)a4 destination:(id)a5;
- (void)encodeBackwarpLossWithFlowMagnitudeToCommandBuffer:(id)a0 first:(id)a1 second:(id)a2 flow:(id)a3 timeScale:(float)a4 gamma:(float)a5 protectionThreshold:(float)a6 destination:(id)a7;
- (void)encodeFlowSplattingWarpToCommandBuffer:(id)a0 source:(id)a1 flow:(id)a2 timeScale:(float)a3 destination:(id)a4;
- (void)encodePaddingTextureToCommandBuffer:(id)a0 source:(id)a1 destination:(id)a2;
- (void)encodeReShuffleFlowToCommandBuffer:(id)a0 shuffledFlow:(id)a1 previousFlow:(id)a2 destination:(id)a3;
- (void)encodeReverseFlowToCommandBuffer:(id)a0 source:(id)a1 destination:(id)a2;
- (void)encodeSubsampleErrorToCommandBuffer:(id)a0 source:(id)a1 destination:(id)a2;
- (void)encodeSubsampleFlowToCommandBuffer:(id)a0 source:(id)a1 destination:(id)a2;
- (void)encodeSubsampleInputToCommandBufferr:(id)a0 source:(id)a1 destination:(id)a2;
- (void)encodeSubsampleToCommandBufferr:(id)a0 source:(id)a1 destination:(id)a2 kernel:(id)a3;
- (void)encodeToCommandBuffer:(id)a0 source:(id)a1 flow:(id)a2 destination:(id)a3 upscaledFlow:(id)a4;
- (void)encodeUpscaleErrorToCommandBuffer:(id)a0 source:(id)a1 destination:(id)a2;
- (void)encodeUpscaleFlowToCommandBuffer:(id)a0 source:(id)a1 destination:(id)a2;
- (id)initWithDevice:(id)a0 interleaved:(BOOL)a1;
- (void)reverseFlowWithSource:(id)a0 destination:(id)a1;
- (void)upscaleFlow:(id)a0 destination:(id)a1;
- (void)warpImage:(id)a0 to:(id)a1 withFlow:(id)a2 upscaledFlow:(id)a3;
@end