pub struct GruDescriptor { /* private fields */ }Expand description
Wraps MPSGRUDescriptor.
Implementations§
Source§impl GruDescriptor
impl GruDescriptor
Source§impl GruDescriptor
impl GruDescriptor
Sourcepub fn input_feature_channels(&self) -> usize
pub fn input_feature_channels(&self) -> usize
Wraps the corresponding Metal Performance Shaders method.
Sourcepub fn set_input_feature_channels(&self, value: usize)
pub fn set_input_feature_channels(&self, value: usize)
Wraps the corresponding Metal Performance Shaders method.
Sourcepub fn output_feature_channels(&self) -> usize
pub fn output_feature_channels(&self) -> usize
Wraps the corresponding Metal Performance Shaders method.
Sourcepub fn set_output_feature_channels(&self, value: usize)
pub fn set_output_feature_channels(&self, value: usize)
Wraps the corresponding Metal Performance Shaders method.
Sourcepub fn use_layer_input_unit_transform_mode(&self) -> bool
pub fn use_layer_input_unit_transform_mode(&self) -> bool
Wraps the corresponding Metal Performance Shaders method.
Sourcepub fn set_use_layer_input_unit_transform_mode(&self, value: bool)
pub fn set_use_layer_input_unit_transform_mode(&self, value: bool)
Wraps the corresponding Metal Performance Shaders method.
Sourcepub fn use_float32_weights(&self) -> bool
pub fn use_float32_weights(&self) -> bool
Wraps the corresponding Metal Performance Shaders method.
Sourcepub fn set_use_float32_weights(&self, value: bool)
pub fn set_use_float32_weights(&self, value: bool)
Wraps the corresponding Metal Performance Shaders method.
Sourcepub fn layer_sequence_direction(&self) -> usize
pub fn layer_sequence_direction(&self) -> usize
Wraps the corresponding Metal Performance Shaders method.
Sourcepub fn set_layer_sequence_direction(&self, value: usize)
pub fn set_layer_sequence_direction(&self, value: usize)
Wraps the corresponding Metal Performance Shaders method.
Source§impl GruDescriptor
impl GruDescriptor
Sourcepub fn new(
input_feature_channels: usize,
output_feature_channels: usize,
) -> Option<Self>
pub fn new( input_feature_channels: usize, output_feature_channels: usize, ) -> Option<Self>
Wraps a constructor on MPSGRUDescriptor.
Sourcepub fn as_descriptor(&self) -> Option<RnnDescriptor>
pub fn as_descriptor(&self) -> Option<RnnDescriptor>
Wraps the corresponding MPSGRUDescriptor conversion helper.
Sourcepub fn gate_pnorm_value(&self) -> f32
pub fn gate_pnorm_value(&self) -> f32
Wraps the corresponding MPSGRUDescriptor method.
Sourcepub fn set_gate_pnorm_value(&self, value: f32)
pub fn set_gate_pnorm_value(&self, value: f32)
Wraps the corresponding MPSGRUDescriptor setter.
Sourcepub fn flip_output_gates(&self) -> bool
pub fn flip_output_gates(&self) -> bool
Wraps the corresponding MPSGRUDescriptor method.
Sourcepub fn set_flip_output_gates(&self, value: bool)
pub fn set_flip_output_gates(&self, value: bool)
Wraps the corresponding MPSGRUDescriptor setter.