pub struct MPSGraphPooling2DOpDescriptor { /* private fields */ }
MPSGraphCore
and MPSGraphPoolingOps
only.Expand description
The class that defines the parameters for a 2D pooling operation.
Use this descriptor with the following methods:
MPSGraph/maxPooling2DWithSourceTensor:descriptor:name:
MPSGraph/maxPooling2DReturnIndicesWithSourceTensor:descriptor:name:
MPSGraph/maxPooling2DGradientWithGradientTensor:sourceTensor:descriptor:name:
MPSGraph/maxPooling2DGradientWithGradientTensor:indicesTensor:outputShape:descriptor:name:
MPSGraph/maxPooling2DGradientWithGradientTensor:indicesTensor:outputShapeTensor:descriptor:name:
MPSGraph/avgPooling2DWithSourceTensor:descriptor:name:
MPSGraph/avgPooling2DGradientWithGradientTensor:sourceTensor:descriptor:name:
See also Apple’s documentation
Implementations§
Source§impl MPSGraphPooling2DOpDescriptor
impl MPSGraphPooling2DOpDescriptor
Sourcepub unsafe fn kernelWidth(&self) -> NSUInteger
pub unsafe fn kernelWidth(&self) -> NSUInteger
Defines the pooling window size for the width dimension.
Sourcepub unsafe fn setKernelWidth(&self, kernel_width: NSUInteger)
pub unsafe fn setKernelWidth(&self, kernel_width: NSUInteger)
Setter for kernelWidth
.
Sourcepub unsafe fn kernelHeight(&self) -> NSUInteger
pub unsafe fn kernelHeight(&self) -> NSUInteger
Defines the pooling window size for the height dimension.
Sourcepub unsafe fn setKernelHeight(&self, kernel_height: NSUInteger)
pub unsafe fn setKernelHeight(&self, kernel_height: NSUInteger)
Setter for kernelHeight
.
Sourcepub unsafe fn strideInX(&self) -> NSUInteger
pub unsafe fn strideInX(&self) -> NSUInteger
Defines the stride for the width dimension.
Default value: 1.
Sourcepub unsafe fn setStrideInX(&self, stride_in_x: NSUInteger)
pub unsafe fn setStrideInX(&self, stride_in_x: NSUInteger)
Setter for strideInX
.
Sourcepub unsafe fn strideInY(&self) -> NSUInteger
pub unsafe fn strideInY(&self) -> NSUInteger
Defines the stride for the height dimension.
Default value: 1.
Sourcepub unsafe fn setStrideInY(&self, stride_in_y: NSUInteger)
pub unsafe fn setStrideInY(&self, stride_in_y: NSUInteger)
Setter for strideInY
.
Sourcepub unsafe fn dilationRateInX(&self) -> NSUInteger
pub unsafe fn dilationRateInX(&self) -> NSUInteger
Defines the dilation rate for the width dimension.
Default value: 1.
Sourcepub unsafe fn setDilationRateInX(&self, dilation_rate_in_x: NSUInteger)
pub unsafe fn setDilationRateInX(&self, dilation_rate_in_x: NSUInteger)
Setter for dilationRateInX
.
Sourcepub unsafe fn dilationRateInY(&self) -> NSUInteger
pub unsafe fn dilationRateInY(&self) -> NSUInteger
Defines the dilation rate for the height dimension.
Default value: 1.
Sourcepub unsafe fn setDilationRateInY(&self, dilation_rate_in_y: NSUInteger)
pub unsafe fn setDilationRateInY(&self, dilation_rate_in_y: NSUInteger)
Setter for dilationRateInY
.
Sourcepub unsafe fn paddingLeft(&self) -> NSUInteger
pub unsafe fn paddingLeft(&self) -> NSUInteger
Defines the explicit padding value for the width dimension to add before the data.
Default value: 0.
Sourcepub unsafe fn setPaddingLeft(&self, padding_left: NSUInteger)
pub unsafe fn setPaddingLeft(&self, padding_left: NSUInteger)
Setter for paddingLeft
.
Sourcepub unsafe fn paddingRight(&self) -> NSUInteger
pub unsafe fn paddingRight(&self) -> NSUInteger
Defines the explicit padding value for the width dimension to add after the data.
Default value: 0.
Sourcepub unsafe fn setPaddingRight(&self, padding_right: NSUInteger)
pub unsafe fn setPaddingRight(&self, padding_right: NSUInteger)
Setter for paddingRight
.
Sourcepub unsafe fn paddingTop(&self) -> NSUInteger
pub unsafe fn paddingTop(&self) -> NSUInteger
Defines the explicit padding value for the height dimension to add before the data.
Default value: 0.
Sourcepub unsafe fn setPaddingTop(&self, padding_top: NSUInteger)
pub unsafe fn setPaddingTop(&self, padding_top: NSUInteger)
Setter for paddingTop
.
Sourcepub unsafe fn paddingBottom(&self) -> NSUInteger
pub unsafe fn paddingBottom(&self) -> NSUInteger
Defines the explicit padding value for the height dimension to add after the data.
Default value: 0.
Sourcepub unsafe fn setPaddingBottom(&self, padding_bottom: NSUInteger)
pub unsafe fn setPaddingBottom(&self, padding_bottom: NSUInteger)
Setter for paddingBottom
.
Sourcepub unsafe fn paddingStyle(&self) -> MPSGraphPaddingStyle
pub unsafe fn paddingStyle(&self) -> MPSGraphPaddingStyle
Defines what kind of padding graph applies to the operation.
Default value: MPSGraphPaddingStyleExplicit
.
Sourcepub unsafe fn setPaddingStyle(&self, padding_style: MPSGraphPaddingStyle)
pub unsafe fn setPaddingStyle(&self, padding_style: MPSGraphPaddingStyle)
Setter for paddingStyle
.
Sourcepub unsafe fn dataLayout(&self) -> MPSGraphTensorNamedDataLayout
pub unsafe fn dataLayout(&self) -> MPSGraphTensorNamedDataLayout
Defines the data layout of the input data in the forward pass. See: MPSGraphTensorNamedDataLayout
.
Sourcepub unsafe fn setDataLayout(&self, data_layout: MPSGraphTensorNamedDataLayout)
pub unsafe fn setDataLayout(&self, data_layout: MPSGraphTensorNamedDataLayout)
Setter for dataLayout
.
Sourcepub unsafe fn returnIndicesMode(&self) -> MPSGraphPoolingReturnIndicesMode
pub unsafe fn returnIndicesMode(&self) -> MPSGraphPoolingReturnIndicesMode
Defines the mode for returned indices of maximum values within each pooling window.
Use this in conjunction with MPSGraph/maxPooling2DReturnIndicesWithSourceTensor:descriptor:name:
API.
If returnIndicesMode = MPSGraphPoolingReturnIndicesNone
then only the first result
MPSGraph returns from MPSGraph/maxPooling2DReturnIndicesWithSourceTensor:descriptor:name:
will be valid and using the second result will assert.
Default value: MPSGraphPoolingReturnIndicesNone
.
Sourcepub unsafe fn setReturnIndicesMode(
&self,
return_indices_mode: MPSGraphPoolingReturnIndicesMode,
)
pub unsafe fn setReturnIndicesMode( &self, return_indices_mode: MPSGraphPoolingReturnIndicesMode, )
Setter for returnIndicesMode
.
Sourcepub unsafe fn returnIndicesDataType(&self) -> MPSDataType
Available on crate feature objc2-metal-performance-shaders
only.
pub unsafe fn returnIndicesDataType(&self) -> MPSDataType
objc2-metal-performance-shaders
only.Defines the data type for returned indices.
Use this in conjunction with MPSGraph/maxPooling2DReturnIndicesWithSourceTensor:descriptor:name:
API.
Currently MPSGraph supports the following datatypes: MPSDataTypeInt32
.
Default value: MPSDataTypeInt32
.
Sourcepub unsafe fn setReturnIndicesDataType(
&self,
return_indices_data_type: MPSDataType,
)
Available on crate feature objc2-metal-performance-shaders
only.
pub unsafe fn setReturnIndicesDataType( &self, return_indices_data_type: MPSDataType, )
objc2-metal-performance-shaders
only.Setter for returnIndicesDataType
.
Sourcepub unsafe fn ceilMode(&self) -> bool
pub unsafe fn ceilMode(&self) -> bool
Affects how the graph computes the output size.
if set to YES
then output size is computed by rounding up instead of down when dividing input size by stride.
Default value: NO
.
Sourcepub unsafe fn setCeilMode(&self, ceil_mode: bool)
pub unsafe fn setCeilMode(&self, ceil_mode: bool)
Setter for ceilMode
.
Sourcepub unsafe fn includeZeroPadToAverage(&self) -> bool
pub unsafe fn includeZeroPadToAverage(&self) -> bool
Defines a mode for average pooling, where samples outside the input tensor count as zeroes in the average computation.
Otherwise the result is sum over samples divided by number of samples that didn’t come from padding.
Default value: NO
.
Sourcepub unsafe fn setIncludeZeroPadToAverage(
&self,
include_zero_pad_to_average: bool,
)
pub unsafe fn setIncludeZeroPadToAverage( &self, include_zero_pad_to_average: bool, )
Setter for includeZeroPadToAverage
.
Sourcepub unsafe fn descriptorWithKernelWidth_kernelHeight_strideInX_strideInY_dilationRateInX_dilationRateInY_paddingLeft_paddingRight_paddingTop_paddingBottom_paddingStyle_dataLayout(
kernel_width: NSUInteger,
kernel_height: NSUInteger,
stride_in_x: NSUInteger,
stride_in_y: NSUInteger,
dilation_rate_in_x: NSUInteger,
dilation_rate_in_y: NSUInteger,
padding_left: NSUInteger,
padding_right: NSUInteger,
padding_top: NSUInteger,
padding_bottom: NSUInteger,
padding_style: MPSGraphPaddingStyle,
data_layout: MPSGraphTensorNamedDataLayout,
) -> Option<Retained<Self>>
pub unsafe fn descriptorWithKernelWidth_kernelHeight_strideInX_strideInY_dilationRateInX_dilationRateInY_paddingLeft_paddingRight_paddingTop_paddingBottom_paddingStyle_dataLayout( kernel_width: NSUInteger, kernel_height: NSUInteger, stride_in_x: NSUInteger, stride_in_y: NSUInteger, dilation_rate_in_x: NSUInteger, dilation_rate_in_y: NSUInteger, padding_left: NSUInteger, padding_right: NSUInteger, padding_top: NSUInteger, padding_bottom: NSUInteger, padding_style: MPSGraphPaddingStyle, data_layout: MPSGraphTensorNamedDataLayout, ) -> Option<Retained<Self>>
Creates a 2D pooling descriptor with given values.
- Parameters:
- kernelWidth: See
kernelWidth
property. - kernelHeight: See
kernelHeight
property. - strideInX: See
strideInX
property. - strideInY: See
strideInY
property. - dilationRateInX: See
dilationRateInX
property. - dilationRateInY: See
dilationRateInY
property. - paddingLeft: See
paddingLeft
property. - paddingRight: See
paddingRight
property. - paddingTop: See
paddingTop
property. - paddingBottom: See
paddingBottom
property. - paddingStyle: See
paddingStyle
property. - dataLayout: See
dataLayout
property. - Returns: The descriptor on autoreleasepool.
Sourcepub unsafe fn descriptorWithKernelWidth_kernelHeight_strideInX_strideInY_paddingStyle_dataLayout(
kernel_width: NSUInteger,
kernel_height: NSUInteger,
stride_in_x: NSUInteger,
stride_in_y: NSUInteger,
padding_style: MPSGraphPaddingStyle,
data_layout: MPSGraphTensorNamedDataLayout,
) -> Option<Retained<Self>>
pub unsafe fn descriptorWithKernelWidth_kernelHeight_strideInX_strideInY_paddingStyle_dataLayout( kernel_width: NSUInteger, kernel_height: NSUInteger, stride_in_x: NSUInteger, stride_in_y: NSUInteger, padding_style: MPSGraphPaddingStyle, data_layout: MPSGraphTensorNamedDataLayout, ) -> Option<Retained<Self>>
Creates a 2D pooling descriptor with given values.
- Parameters:
- kernelWidth: See
kernelWidth
property. - kernelHeight: See `kernelHeight`` property.
- strideInX: See
strideInX
property. - strideInY: See
strideInY
property. - paddingStyle: See
paddingStyle
property. - dataLayout: See
dataLayout
property. - Returns: The descriptor on autoreleasepool.
Sourcepub unsafe fn setExplicitPaddingWithPaddingLeft_paddingRight_paddingTop_paddingBottom(
&self,
padding_left: NSUInteger,
padding_right: NSUInteger,
padding_top: NSUInteger,
padding_bottom: NSUInteger,
)
pub unsafe fn setExplicitPaddingWithPaddingLeft_paddingRight_paddingTop_paddingBottom( &self, padding_left: NSUInteger, padding_right: NSUInteger, padding_top: NSUInteger, padding_bottom: NSUInteger, )
Sets the explicit padding values and sets padding style to explicit.
- Parameters:
- paddingLeft: See
paddingLeft
property. - paddingRight: See
paddingRight
property. - paddingTop: See
paddingTop
property. - paddingBottom: See
paddingBottom
property.
Methods from Deref<Target = NSObject>§
Sourcepub fn doesNotRecognizeSelector(&self, sel: Sel) -> !
pub fn doesNotRecognizeSelector(&self, sel: Sel) -> !
Handle messages the object doesn’t recognize.
See Apple’s documentation for details.
Methods from Deref<Target = AnyObject>§
Sourcepub fn class(&self) -> &'static AnyClass
pub fn class(&self) -> &'static AnyClass
Dynamically find the class of this object.
§Panics
May panic if the object is invalid (which may be the case for objects
returned from unavailable init
/new
methods).
§Example
Check that an instance of NSObject
has the precise class NSObject
.
use objc2::ClassType;
use objc2::runtime::NSObject;
let obj = NSObject::new();
assert_eq!(obj.class(), NSObject::class());
Sourcepub unsafe fn get_ivar<T>(&self, name: &str) -> &Twhere
T: Encode,
👎Deprecated: this is difficult to use correctly, use Ivar::load
instead.
pub unsafe fn get_ivar<T>(&self, name: &str) -> &Twhere
T: Encode,
Ivar::load
instead.Use Ivar::load
instead.
§Safety
The object must have an instance variable with the given name, and it
must be of type T
.
See Ivar::load_ptr
for details surrounding this.
Sourcepub fn downcast_ref<T>(&self) -> Option<&T>where
T: DowncastTarget,
pub fn downcast_ref<T>(&self) -> Option<&T>where
T: DowncastTarget,
Attempt to downcast the object to a class of type T
.
This is the reference-variant. Use Retained::downcast
if you want
to convert a retained object to another type.
§Mutable classes
Some classes have immutable and mutable variants, such as NSString
and NSMutableString
.
When some Objective-C API signature says it gives you an immutable class, it generally expects you to not mutate that, even though it may technically be mutable “under the hood”.
So using this method to convert a NSString
to a NSMutableString
,
while not unsound, is generally frowned upon unless you created the
string yourself, or the API explicitly documents the string to be
mutable.
See Apple’s documentation on mutability and on
isKindOfClass:
for more details.
§Generic classes
Objective-C generics are called “lightweight generics”, and that’s because they aren’t exposed in the runtime. This makes it impossible to safely downcast to generic collections, so this is disallowed by this method.
You can, however, safely downcast to generic collections where all the
type-parameters are AnyObject
.
§Panics
This works internally by calling isKindOfClass:
. That means that the
object must have the instance method of that name, and an exception
will be thrown (if CoreFoundation is linked) or the process will abort
if that is not the case. In the vast majority of cases, you don’t need
to worry about this, since both root objects NSObject
and
NSProxy
implement this method.
§Examples
Cast an NSString
back and forth from NSObject
.
use objc2::rc::Retained;
use objc2_foundation::{NSObject, NSString};
let obj: Retained<NSObject> = NSString::new().into_super();
let string = obj.downcast_ref::<NSString>().unwrap();
// Or with `downcast`, if we do not need the object afterwards
let string = obj.downcast::<NSString>().unwrap();
Try (and fail) to cast an NSObject
to an NSString
.
use objc2_foundation::{NSObject, NSString};
let obj = NSObject::new();
assert!(obj.downcast_ref::<NSString>().is_none());
Try to cast to an array of strings.
use objc2_foundation::{NSArray, NSObject, NSString};
let arr = NSArray::from_retained_slice(&[NSObject::new()]);
// This is invalid and doesn't type check.
let arr = arr.downcast_ref::<NSArray<NSString>>();
This fails to compile, since it would require enumerating over the array to ensure that each element is of the desired type, which is a performance pitfall.
Downcast when processing each element instead.
use objc2_foundation::{NSArray, NSObject, NSString};
let arr = NSArray::from_retained_slice(&[NSObject::new()]);
for elem in arr {
if let Some(data) = elem.downcast_ref::<NSString>() {
// handle `data`
}
}
Trait Implementations§
Source§impl AsRef<MPSGraphObject> for MPSGraphPooling2DOpDescriptor
impl AsRef<MPSGraphObject> for MPSGraphPooling2DOpDescriptor
Source§fn as_ref(&self) -> &MPSGraphObject
fn as_ref(&self) -> &MPSGraphObject
Source§impl Borrow<MPSGraphObject> for MPSGraphPooling2DOpDescriptor
impl Borrow<MPSGraphObject> for MPSGraphPooling2DOpDescriptor
Source§fn borrow(&self) -> &MPSGraphObject
fn borrow(&self) -> &MPSGraphObject
Source§impl ClassType for MPSGraphPooling2DOpDescriptor
impl ClassType for MPSGraphPooling2DOpDescriptor
Source§const NAME: &'static str = "MPSGraphPooling2DOpDescriptor"
const NAME: &'static str = "MPSGraphPooling2DOpDescriptor"
Source§type Super = MPSGraphObject
type Super = MPSGraphObject
Source§type ThreadKind = <<MPSGraphPooling2DOpDescriptor as ClassType>::Super as ClassType>::ThreadKind
type ThreadKind = <<MPSGraphPooling2DOpDescriptor as ClassType>::Super as ClassType>::ThreadKind
Source§impl CopyingHelper for MPSGraphPooling2DOpDescriptor
impl CopyingHelper for MPSGraphPooling2DOpDescriptor
Source§type Result = MPSGraphPooling2DOpDescriptor
type Result = MPSGraphPooling2DOpDescriptor
Self
if the type has no
immutable counterpart. Read moreSource§impl Hash for MPSGraphPooling2DOpDescriptor
impl Hash for MPSGraphPooling2DOpDescriptor
Source§impl NSCopying for MPSGraphPooling2DOpDescriptor
impl NSCopying for MPSGraphPooling2DOpDescriptor
Source§impl NSObjectProtocol for MPSGraphPooling2DOpDescriptor
impl NSObjectProtocol for MPSGraphPooling2DOpDescriptor
Source§fn isEqual(&self, other: Option<&AnyObject>) -> bool
fn isEqual(&self, other: Option<&AnyObject>) -> bool
Source§fn hash(&self) -> usize
fn hash(&self) -> usize
Source§fn isKindOfClass(&self, cls: &AnyClass) -> bool
fn isKindOfClass(&self, cls: &AnyClass) -> bool
Source§fn is_kind_of<T>(&self) -> bool
fn is_kind_of<T>(&self) -> bool
isKindOfClass
directly, or cast your objects with AnyObject::downcast_ref