Lib.rs

Cargo Features

[dependencies]
objc2-ml-compute = { version = "0.2.2", default-features = false, features = ["std", "alloc", "all", "bitflags", "block2", "objc2-metal", "MLCActivationDescriptor", "MLCActivationLayer", "MLCAdamWOptimizer", "MLCComparisonLayer", "MLCConcatenationLayer", "MLCConvolutionDescriptor", "MLCDefines", "MLCDevice", "MLCDropoutLayer", "MLCEmbeddingDescriptor", "MLCGatherLayer", "MLCGramMatrixLayer", "MLCGraph", "MLCGroupNormalizationLayer", "MLCInferenceGraph", "MLCInstanceNormalizationLayer", "MLCLSTMDescriptor", "MLCLSTMLayer", "MLCLayer", "MLCLayerNormalizationLayer", "MLCLossDescriptor", "MLCLossLayer", "MLCMatMulDescriptor", "MLCMatMulLayer", "MLCMultiheadAttentionDescriptor", "MLCMultiheadAttentionLayer", "MLCOptimizerDescriptor", "MLCPaddingLayer", "MLCPlatform", "MLCPoolingDescriptor", "MLCPoolingLayer", "MLCRMSPropOptimizer", "MLCReshapeLayer", "MLCSGDOptimizer", "MLCSelectionLayer", "MLCSliceLayer", "MLCSoftmaxLayer", "MLCSplitLayer", "MLCTensor", "MLCTensorData", "MLCTensorDescriptor", "MLCTensorOptimizerDeviceData", "MLCTensorParameter", "MLCTrainingGraph", "MLCTransposeLayer", "MLCTypes", "MLCUpsampleLayer", "MLCYOLOLossDescriptor", "MLCYOLOLossLayer"] }

default = std

The std feature is set by default whenever objc2-ml-compute is added without default-features = false somewhere in the dependency tree.

std default = alloc

Currently not possible to turn off, put here for forwards compatibility.

Enables std of optional bitflags, optional block2, objc2, objc2-foundation, and optional objc2-metal

alloc std

Enables alloc of optional block2, objc2, objc2-foundation, and optional objc2-metal

all = MLCActivationDescriptor, MLCActivationLayer, MLCAdamOptimizer, MLCAdamWOptimizer, MLCArithmeticLayer, MLCBatchNormalizationLayer, MLCComparisonLayer, MLCConcatenationLayer, MLCConvolutionDescriptor, MLCConvolutionLayer, MLCDefines, MLCDevice, MLCDropoutLayer, MLCEmbeddingDescriptor, MLCEmbeddingLayer, MLCFullyConnectedLayer, MLCGatherLayer, MLCGramMatrixLayer, MLCGraph, MLCGroupNormalizationLayer, MLCInferenceGraph, MLCInstanceNormalizationLayer, MLCLSTMDescriptor, MLCLSTMLayer, MLCLayer, MLCLayerNormalizationLayer, MLCLossDescriptor, MLCLossLayer, MLCMatMulDescriptor, MLCMatMulLayer, MLCMultiheadAttentionDescriptor, MLCMultiheadAttentionLayer, MLCOptimizer, MLCOptimizerDescriptor, MLCPaddingLayer, MLCPlatform, MLCPoolingDescriptor, MLCPoolingLayer, MLCRMSPropOptimizer, MLCReductionLayer, MLCReshapeLayer, MLCSGDOptimizer, MLCScatterLayer, MLCSelectionLayer, MLCSliceLayer, MLCSoftmaxLayer, MLCSplitLayer, MLCTensor, MLCTensorData, MLCTensorDescriptor, MLCTensorOptimizerDeviceData, MLCTensorParameter, MLCTrainingGraph, MLCTransposeLayer, MLCTypes, MLCUpsampleLayer, MLCYOLOLossDescriptor, MLCYOLOLossLayer, bitflags, block2, objc2-metal

bitflags MLCTypes? all?

Enables bitflags, bitflags of objc2-foundation and optional objc2-metal

block2 all?

Enables block2, block2 of objc2-foundation and optional objc2-metal

Affects MLCTypes::MLCGraphCompletionHandler…

objc2-metal all?

Enables objc2-metal

MLCActivationDescriptor all?

Enables NSObject of objc2-foundation

MLCActivationLayer all?

MLCAdamOptimizer MLCAdamWOptimizer all?

Enables NSObject of objc2-foundation

MLCArithmeticLayer MLCBatchNormalizationLayer MLCComparisonLayer all?

MLCConcatenationLayer all?

MLCConvolutionDescriptor all?

Enables NSArray, NSObject and NSValue of objc2-foundation

MLCConvolutionLayer MLCDefines all?

MLCDevice all?

Enables NSArray and NSObject of objc2-foundation, MTLDevice of optional objc2-metal

MLCDropoutLayer all?

MLCEmbeddingDescriptor all?

Enables NSObject and NSValue of objc2-foundation

MLCEmbeddingLayer MLCFullyConnectedLayer MLCGatherLayer all?

MLCGramMatrixLayer all?

MLCGraph all?

Enables NSArray, NSDictionary, NSString and NSValue of objc2-foundation

MLCGroupNormalizationLayer all?

MLCInferenceGraph all?

Enables NSArray, NSDate, NSDictionary, NSError and NSString of objc2-foundation

MLCInstanceNormalizationLayer all?

MLCLSTMDescriptor all?

Enables NSObject of objc2-foundation

MLCLSTMLayer all?

Enables NSArray of objc2-foundation

MLCLayer all?

Enables NSString of objc2-foundation

MLCLayerNormalizationLayer all?

Enables NSArray and NSValue of objc2-foundation

MLCLossDescriptor all?

Enables NSObject of objc2-foundation

MLCLossLayer all?

MLCMatMulDescriptor all?

Enables NSObject of objc2-foundation

MLCMatMulLayer all?

MLCMultiheadAttentionDescriptor all?

Enables NSObject of objc2-foundation

MLCMultiheadAttentionLayer all?

Enables NSArray of objc2-foundation

MLCOptimizer MLCOptimizerDescriptor all?

Enables NSObject of objc2-foundation

MLCPaddingLayer all?

Enables NSArray, NSObject and NSValue of objc2-foundation

MLCPlatform all?

Enables NSValue of objc2-foundation

MLCPoolingDescriptor all?

Enables NSArray, NSObject and NSValue of objc2-foundation

MLCPoolingLayer all?

MLCRMSPropOptimizer all?

Enables NSObject of objc2-foundation

MLCReductionLayer MLCReshapeLayer all?

Enables NSArray and NSValue of objc2-foundation

MLCSGDOptimizer all?

Enables NSObject of objc2-foundation

MLCScatterLayer MLCSelectionLayer all?

MLCSliceLayer all?

Enables NSArray and NSValue of objc2-foundation

MLCSoftmaxLayer all?

MLCSplitLayer all?

Enables NSArray and NSValue of objc2-foundation

MLCTensor all?

Enables NSArray, NSData, NSObject, NSString and NSValue of objc2-foundation

Affects MLCTypes::MLCGraphCompletionHandler…

MLCTensorData all?

MLCTensorDescriptor all?

Enables NSArray, NSObject and NSValue of objc2-foundation

MLCTensorOptimizerDeviceData all?

Enables NSObject of objc2-foundation

MLCTensorParameter all?

Enables NSArray of objc2-foundation

MLCTrainingGraph all?

Enables NSArray, NSData, NSDate, NSDictionary, NSError and NSString of objc2-foundation

MLCTransposeLayer all?

Enables NSArray and NSValue of objc2-foundation

MLCTypes all? = bitflags

Enables NSDate, NSError and NSString of objc2-foundation

MLCUpsampleLayer all?

Enables NSArray and NSValue of objc2-foundation

MLCYOLOLossDescriptor all?

Enables NSData and NSObject of objc2-foundation

MLCYOLOLossLayer all?