objc2-ml-compute 0.3.2

//! This file has been automatically generated by `objc2`'s `header-translator`.
//! DO NOT EDIT
use core::ptr::NonNull;
use objc2::__framework_prelude::*;
use objc2_foundation::*;

use crate::*;

extern_class!(
    /// A graph of layers that can be used to build a training or inference graph
    ///
    /// See also [Apple's documentation](https://developer.apple.com/documentation/mlcompute/mlcgraph?language=objc)
    #[unsafe(super(NSObject))]
    #[derive(Debug, PartialEq, Eq, Hash)]
    #[deprecated]
    pub struct MLCGraph;
);

extern_conformance!(
    unsafe impl NSObjectProtocol for MLCGraph {}
);

impl MLCGraph {
    extern_methods!(
        #[cfg(feature = "MLCDevice")]
        /// The device to be used when compiling and executing a graph
        #[deprecated]
        #[unsafe(method(device))]
        #[unsafe(method_family = none)]
        pub unsafe fn device(&self) -> Option<Retained<MLCDevice>>;

        #[cfg(feature = "MLCLayer")]
        /// Layers in the graph
        #[deprecated]
        #[unsafe(method(layers))]
        #[unsafe(method_family = none)]
        pub unsafe fn layers(&self) -> Retained<NSArray<MLCLayer>>;

        /// Creates a new graph.
        ///
        /// Returns: A new graph.
        #[deprecated]
        #[unsafe(method(graph))]
        #[unsafe(method_family = none)]
        pub unsafe fn graph() -> Retained<Self>;

        /// A DOT representation of the graph.
        ///
        /// For more info on the DOT language, refer to https://en.wikipedia.org/wiki/DOT_(graph_description_language).
        /// Edges that have a dashed lines are those that have stop gradients, while those with solid lines don't.
        #[deprecated]
        #[unsafe(method(summarizedDOTDescription))]
        #[unsafe(method_family = none)]
        pub unsafe fn summarizedDOTDescription(&self) -> Retained<NSString>;

        #[cfg(all(feature = "MLCLayer", feature = "MLCTensor"))]
        /// Add a layer to the graph
        ///
        /// Parameter `layer`: The layer
        ///
        /// Parameter `source`: The source tensor
        ///
        /// Returns: A result tensor
        #[deprecated]
        #[unsafe(method(nodeWithLayer:source:))]
        #[unsafe(method_family = none)]
        pub unsafe fn nodeWithLayer_source(
            &self,
            layer: &MLCLayer,
            source: &MLCTensor,
        ) -> Option<Retained<MLCTensor>>;

        #[cfg(all(feature = "MLCLayer", feature = "MLCTensor"))]
        /// Add a layer to the graph
        ///
        /// Parameter `layer`: The layer
        ///
        /// Parameter `sources`: A list of source tensors
        ///
        /// For variable length sequences of LSTMs/RNNs layers, create an MLCTensor of sortedSequenceLengths and pass it as the last index (i.e. index 2 or 4) of sources. This tensor must of be type MLCDataTypeInt32.
        ///
        /// Returns: A result tensor
        #[deprecated]
        #[unsafe(method(nodeWithLayer:sources:))]
        #[unsafe(method_family = none)]
        pub unsafe fn nodeWithLayer_sources(
            &self,
            layer: &MLCLayer,
            sources: &NSArray<MLCTensor>,
        ) -> Option<Retained<MLCTensor>>;

        #[cfg(all(feature = "MLCLayer", feature = "MLCTensor"))]
        /// Add a layer to the graph
        ///
        /// Parameter `layer`: The layer
        ///
        /// Parameter `sources`: A list of source tensors
        ///
        /// Parameter `disableUpdate`: A flag to indicate if optimizer update should be disabled for this layer
        ///
        /// For variable length sequences of LSTMs/RNNs layers, create an MLCTensor of sortedSequenceLengths and pass it as the last index (i.e. index 2 or 4) of sources. This tensor must of be type MLCDataTypeInt32.
        ///
        /// Returns: A result tensor
        #[deprecated]
        #[unsafe(method(nodeWithLayer:sources:disableUpdate:))]
        #[unsafe(method_family = none)]
        pub unsafe fn nodeWithLayer_sources_disableUpdate(
            &self,
            layer: &MLCLayer,
            sources: &NSArray<MLCTensor>,
            disable_update: bool,
        ) -> Option<Retained<MLCTensor>>;

        #[cfg(all(feature = "MLCLayer", feature = "MLCTensor"))]
        /// Add a loss layer to the graph
        ///
        /// Parameter `layer`: The loss layer
        ///
        /// Parameter `lossLabels`: The loss labels tensor
        ///
        /// For variable length sequences of LSTMs/RNNs layers, create an MLCTensor of sortedSequenceLengths and pass it as the last index (i.e. index 2 or 4) of sources. This tensor must of be type MLCDataTypeInt32.
        ///
        /// Returns: A result tensor
        #[deprecated]
        #[unsafe(method(nodeWithLayer:sources:lossLabels:))]
        #[unsafe(method_family = none)]
        pub unsafe fn nodeWithLayer_sources_lossLabels(
            &self,
            layer: &MLCLayer,
            sources: &NSArray<MLCTensor>,
            loss_labels: &NSArray<MLCTensor>,
        ) -> Option<Retained<MLCTensor>>;

        #[cfg(feature = "MLCTensor")]
        /// Add a split layer to the graph
        ///
        /// Parameter `source`: The source tensor
        ///
        /// Parameter `splitCount`: The number of splits
        ///
        /// Parameter `dimension`: The dimension to split the source tensor
        ///
        /// Returns: A result tensor
        #[deprecated]
        #[unsafe(method(splitWithSource:splitCount:dimension:))]
        #[unsafe(method_family = none)]
        pub unsafe fn splitWithSource_splitCount_dimension(
            &self,
            source: &MLCTensor,
            split_count: NSUInteger,
            dimension: NSUInteger,
        ) -> Option<Retained<NSArray<MLCTensor>>>;

        #[cfg(feature = "MLCTensor")]
        /// Add a split layer to the graph
        ///
        /// Parameter `source`: The source tensor
        ///
        /// Parameter `splitSectionLengths`: The lengths of each split section
        ///
        /// Parameter `dimension`: The dimension to split the source tensor
        ///
        /// Returns: A result tensor
        #[deprecated]
        #[unsafe(method(splitWithSource:splitSectionLengths:dimension:))]
        #[unsafe(method_family = none)]
        pub unsafe fn splitWithSource_splitSectionLengths_dimension(
            &self,
            source: &MLCTensor,
            split_section_lengths: &NSArray<NSNumber>,
            dimension: NSUInteger,
        ) -> Option<Retained<NSArray<MLCTensor>>>;

        #[cfg(feature = "MLCTensor")]
        /// Add a concat layer to the graph
        ///
        /// Parameter `sources`: The source tensors to concatenate
        ///
        /// Parameter `dimension`: The concatenation dimension
        ///
        /// Returns: A result tensor
        #[deprecated]
        #[unsafe(method(concatenateWithSources:dimension:))]
        #[unsafe(method_family = none)]
        pub unsafe fn concatenateWithSources_dimension(
            &self,
            sources: &NSArray<MLCTensor>,
            dimension: NSUInteger,
        ) -> Option<Retained<MLCTensor>>;

        #[cfg(feature = "MLCTensor")]
        /// Add a reshape layer to the graph
        ///
        /// Parameter `shape`: An array representing the shape of result tensor
        ///
        /// Parameter `source`: The source tensor
        ///
        /// Returns: A result tensor
        #[deprecated]
        #[unsafe(method(reshapeWithShape:source:))]
        #[unsafe(method_family = none)]
        pub unsafe fn reshapeWithShape_source(
            &self,
            shape: &NSArray<NSNumber>,
            source: &MLCTensor,
        ) -> Option<Retained<MLCTensor>>;

        #[cfg(feature = "MLCTensor")]
        /// Add a transpose layer to the graph
        ///
        /// Parameter `dimensions`: NSArray
        /// <NSNumber
        /// *> representing the desired ordering of dimensions
        /// The dimensions array specifies the input axis source for each output axis, such that the
        /// K'th element in the dimensions array specifies the input axis source for the K'th axis in the
        /// output.  The batch dimension which is typically axis 0 cannot be transposed.
        ///
        /// Returns: A result tensor
        #[deprecated]
        #[unsafe(method(transposeWithDimensions:source:))]
        #[unsafe(method_family = none)]
        pub unsafe fn transposeWithDimensions_source(
            &self,
            dimensions: &NSArray<NSNumber>,
            source: &MLCTensor,
        ) -> Option<Retained<MLCTensor>>;

        #[cfg(feature = "MLCTensor")]
        /// Add a select layer to the graph
        ///
        /// Parameter `sources`: The source tensors
        ///
        /// Parameter `condition`: The condition mask
        ///
        /// Returns: A result tensor
        #[unsafe(method(selectWithSources:condition:))]
        #[unsafe(method_family = none)]
        pub unsafe fn selectWithSources_condition(
            &self,
            sources: &NSArray<MLCTensor>,
            condition: &MLCTensor,
        ) -> Option<Retained<MLCTensor>>;

        #[cfg(all(feature = "MLCTensor", feature = "MLCTypes"))]
        /// Add a scatter layer to the graph
        ///
        /// Parameter `dimension`: The dimension along which to index
        ///
        /// Parameter `source`: The updates to use with scattering with index positions specified in indices to result tensor
        ///
        /// Parameter `indices`: The index of elements to scatter
        ///
        /// Parameter `copyFrom`: The source tensor whose data is  to be first copied to the result tensor
        ///
        /// Parameter `reductionType`: The reduction type applied for all values in source tensor that are scattered to a specific location in the result tensor.
        /// Must be: MLCReductionTypeNone or MLCReductionTypeSum.
        ///
        /// Returns: A result tensor
        #[unsafe(method(scatterWithDimension:source:indices:copyFrom:reductionType:))]
        #[unsafe(method_family = none)]
        pub unsafe fn scatterWithDimension_source_indices_copyFrom_reductionType(
            &self,
            dimension: NSUInteger,
            source: &MLCTensor,
            indices: &MLCTensor,
            copy_from: &MLCTensor,
            reduction_type: MLCReductionType,
        ) -> Option<Retained<MLCTensor>>;

        #[cfg(feature = "MLCTensor")]
        /// Add a gather layer to the graph
        ///
        /// Parameter `dimension`: The dimension along which to index
        ///
        /// Parameter `source`: The source tensor
        ///
        /// Parameter `indices`: The index of elements to gather
        ///
        /// Returns: A result tensor
        #[unsafe(method(gatherWithDimension:source:indices:))]
        #[unsafe(method_family = none)]
        pub unsafe fn gatherWithDimension_source_indices(
            &self,
            dimension: NSUInteger,
            source: &MLCTensor,
            indices: &MLCTensor,
        ) -> Option<Retained<MLCTensor>>;

        #[cfg(all(
            feature = "MLCDevice",
            feature = "MLCTensor",
            feature = "MLCTensorData"
        ))]
        /// Associates data with input tensors. If the device is GPU, also copies the data to the device memory.
        /// Returns true if the data is successfully associated with input tensors.
        ///
        /// This function should be used if you execute the forward, gradient and optimizer updates independently.
        /// Before the forward pass is executed, the inputs should be written to device memory.  Similarly, before the
        /// gradient pass is executed, the inputs (typically the initial gradient tensor) should be written to device
        /// memory.  The caller must guarantee the lifetime of the underlying memory of each value of
        /// `inputsData`for the entirety of each corresponding input tensor's lifetime.
        ///
        /// Parameter `inputsData`: The input data to use to write to device memory
        ///
        /// Parameter `inputTensors`: The list of tensors to perform writes on
        ///
        /// Parameter `device`: The device
        ///
        /// Parameter `batchSize`: The batch size.  This should be set to the actual batch size that may be used when we execute
        /// the graph and can be a value less than or equal to the batch size specified in the tensor.
        /// If set to 0, we use batch size specified in the tensor.
        ///
        /// Parameter `synchronous`: Whether to execute the copy to the device synchronously.  For performance, asynchronous
        /// execution is recommended.
        ///
        /// Returns: A Boolean value indicating whether the data is successfully associated with the tensor.
        #[deprecated]
        #[unsafe(method(bindAndWriteData:forInputs:toDevice:batchSize:synchronous:))]
        #[unsafe(method_family = none)]
        pub unsafe fn bindAndWriteData_forInputs_toDevice_batchSize_synchronous(
            &self,
            inputs_data: &NSDictionary<NSString, MLCTensorData>,
            input_tensors: &NSDictionary<NSString, MLCTensor>,
            device: &MLCDevice,
            batch_size: NSUInteger,
            synchronous: bool,
        ) -> bool;

        #[cfg(all(
            feature = "MLCDevice",
            feature = "MLCTensor",
            feature = "MLCTensorData"
        ))]
        /// Associates data with input tensors. If the device is GPU, also copies the data to the device memory.
        /// Returns true if the data is successfully associated with input tensors.
        ///
        /// This function should be used if you execute the forward, gradient and optimizer updates independently.
        /// Before the forward pass is executed, the inputs should be written to device memory.  Similarly, before the
        /// gradient pass is executed, the inputs (typically the initial gradient tensor) should be written to device
        /// memory.  The caller must guarantee the lifetime of the underlying memory of each value of
        /// `inputsData`for the entirety of each corresponding input tensor's lifetime.
        ///
        /// Parameter `inputsData`: The input data to use to write to device memory
        ///
        /// Parameter `inputTensors`: The list of tensors to perform writes on
        ///
        /// Parameter `device`: The device
        ///
        /// Parameter `synchronous`: Whether to execute the copy to the device synchronously.  For performance, asynchronous
        /// execution is recommended.
        ///
        /// Returns: A Boolean value indicating whether the data is successfully associated with the tensor.
        #[deprecated]
        #[unsafe(method(bindAndWriteData:forInputs:toDevice:synchronous:))]
        #[unsafe(method_family = none)]
        pub unsafe fn bindAndWriteData_forInputs_toDevice_synchronous(
            &self,
            inputs_data: &NSDictionary<NSString, MLCTensorData>,
            input_tensors: &NSDictionary<NSString, MLCTensor>,
            device: &MLCDevice,
            synchronous: bool,
        ) -> bool;

        #[cfg(all(feature = "MLCLayer", feature = "MLCTensor"))]
        /// Get the source tensors for a layer in the training graph
        ///
        /// Parameter `layer`: A layer in the training graph
        ///
        /// Returns: A list of tensors
        #[deprecated]
        #[unsafe(method(sourceTensorsForLayer:))]
        #[unsafe(method_family = none)]
        pub unsafe fn sourceTensorsForLayer(
            &self,
            layer: &MLCLayer,
        ) -> Retained<NSArray<MLCTensor>>;

        #[cfg(all(feature = "MLCLayer", feature = "MLCTensor"))]
        /// Get the result tensors for a layer in the training graph
        ///
        /// Parameter `layer`: A layer in the training graph
        ///
        /// Returns: A list of tensors
        #[deprecated]
        #[unsafe(method(resultTensorsForLayer:))]
        #[unsafe(method_family = none)]
        pub unsafe fn resultTensorsForLayer(
            &self,
            layer: &MLCLayer,
        ) -> Retained<NSArray<MLCTensor>>;
    );
}

/// Methods declared on superclass `NSObject`.
impl MLCGraph {
    extern_methods!(
        #[unsafe(method(init))]
        #[unsafe(method_family = init)]
        pub unsafe fn init(this: Allocated<Self>) -> Retained<Self>;

        #[unsafe(method(new))]
        #[unsafe(method_family = new)]
        pub unsafe fn new() -> Retained<Self>;
    );
}