pub unsafe trait MLCustomLayer {
// Provided methods
unsafe fn initWithParameterDictionary_error(
this: Allocated<Self>,
parameters: &NSDictionary<NSString, AnyObject>,
) -> Result<Retained<Self>, Retained<NSError>>
where Self: Sized + Message { ... }
unsafe fn setWeightData_error(
&self,
weights: &NSArray<NSData>,
) -> Result<(), Retained<NSError>>
where Self: Sized + Message { ... }
unsafe fn outputShapesForInputShapes_error(
&self,
input_shapes: &NSArray<NSArray<NSNumber>>,
) -> Result<Retained<NSArray<NSArray<NSNumber>>>, Retained<NSError>>
where Self: Sized + Message { ... }
unsafe fn evaluateOnCPUWithInputs_outputs_error(
&self,
inputs: &NSArray<MLMultiArray>,
outputs: &NSArray<MLMultiArray>,
) -> Result<(), Retained<NSError>>
where Self: Sized + Message { ... }
unsafe fn encodeToCommandBuffer_inputs_outputs_error(
&self,
command_buffer: &ProtocolObject<dyn MTLCommandBuffer>,
inputs: &NSArray<ProtocolObject<dyn MTLTexture>>,
outputs: &NSArray<ProtocolObject<dyn MTLTexture>>,
) -> Result<(), Retained<NSError>>
where Self: Sized + Message { ... }
}
Available on crate feature
MLCustomLayer
only.Expand description
Provided Methods§
unsafe fn initWithParameterDictionary_error( this: Allocated<Self>, parameters: &NSDictionary<NSString, AnyObject>, ) -> Result<Retained<Self>, Retained<NSError>>
unsafe fn setWeightData_error( &self, weights: &NSArray<NSData>, ) -> Result<(), Retained<NSError>>
unsafe fn outputShapesForInputShapes_error( &self, input_shapes: &NSArray<NSArray<NSNumber>>, ) -> Result<Retained<NSArray<NSArray<NSNumber>>>, Retained<NSError>>
unsafe fn evaluateOnCPUWithInputs_outputs_error( &self, inputs: &NSArray<MLMultiArray>, outputs: &NSArray<MLMultiArray>, ) -> Result<(), Retained<NSError>>
Available on crate feature
MLMultiArray
only.unsafe fn encodeToCommandBuffer_inputs_outputs_error( &self, command_buffer: &ProtocolObject<dyn MTLCommandBuffer>, inputs: &NSArray<ProtocolObject<dyn MTLTexture>>, outputs: &NSArray<ProtocolObject<dyn MTLTexture>>, ) -> Result<(), Retained<NSError>>
Available on crate feature
objc2-metal
and non-watchOS only.