1 #import <ATen/ATen.h> 2 #import <CoreML/CoreML.h> 3 4 NS_ASSUME_NONNULL_BEGIN 5 6 @interface PTMCoreMLFeatureProvider : NSObject<MLFeatureProvider> 7 8 - (instancetype)initWithFeatureNames:(NSSet<NSString*>*)featureNames; 9 10 - (void)clearInputTensors; 11 12 - (void)setInputTensor:(const at::Tensor&)tensor forFeatureName:(NSString*)name; 13 14 @end 15 16 NS_ASSUME_NONNULL_END 17