pub struct LstmDescriptor { /* private fields */ }Expand description
Wraps MPSLSTMDescriptor.
Implementations§
Source§impl LstmDescriptor
impl LstmDescriptor
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 LstmDescriptor
impl LstmDescriptor
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 MPSLSTMDescriptor.
Sourcepub fn as_descriptor(&self) -> Option<RnnDescriptor>
pub fn as_descriptor(&self) -> Option<RnnDescriptor>
Wraps the corresponding MPSLSTMDescriptor conversion helper.
Sourcepub fn memory_weights_are_diagonal(&self) -> bool
pub fn memory_weights_are_diagonal(&self) -> bool
Wraps the corresponding MPSLSTMDescriptor method.
Sourcepub fn set_memory_weights_are_diagonal(&self, value: bool)
pub fn set_memory_weights_are_diagonal(&self, value: bool)
Wraps the corresponding MPSLSTMDescriptor setter.
Sourcepub fn cell_to_output_neuron_type(&self) -> usize
pub fn cell_to_output_neuron_type(&self) -> usize
Wraps the corresponding MPSLSTMDescriptor method.
Sourcepub fn set_cell_to_output_neuron_type(&self, value: usize)
pub fn set_cell_to_output_neuron_type(&self, value: usize)
Wraps the corresponding MPSLSTMDescriptor setter.
Sourcepub fn cell_to_output_neuron_param_a(&self) -> f32
pub fn cell_to_output_neuron_param_a(&self) -> f32
Wraps the corresponding MPSLSTMDescriptor method.
Sourcepub fn set_cell_to_output_neuron_param_a(&self, value: f32)
pub fn set_cell_to_output_neuron_param_a(&self, value: f32)
Wraps the corresponding MPSLSTMDescriptor setter.
Sourcepub fn cell_to_output_neuron_param_b(&self) -> f32
pub fn cell_to_output_neuron_param_b(&self) -> f32
Wraps the corresponding MPSLSTMDescriptor method.
Sourcepub fn set_cell_to_output_neuron_param_b(&self, value: f32)
pub fn set_cell_to_output_neuron_param_b(&self, value: f32)
Wraps the corresponding MPSLSTMDescriptor setter.
Sourcepub fn cell_to_output_neuron_param_c(&self) -> f32
pub fn cell_to_output_neuron_param_c(&self) -> f32
Wraps the corresponding MPSLSTMDescriptor method.
Sourcepub fn set_cell_to_output_neuron_param_c(&self, value: f32)
pub fn set_cell_to_output_neuron_param_c(&self, value: f32)
Wraps the corresponding MPSLSTMDescriptor setter.