Package org.robovm.apple.mlcompute

Class MLCTrainingGraph

java.lang.Object

org.robovm.rt.bro.NativeObject

org.robovm.objc.ObjCObject

org.robovm.apple.foundation.NSObject

org.robovm.apple.mlcompute.MLCGraph

org.robovm.apple.mlcompute.MLCTrainingGraph

All Implemented Interfaces:

NSObjectProtocol, ObjCProtocol

public class MLCTrainingGraph
extends MLCGraph

Since:

Available in iOS 14.0 and later.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	MLCTrainingGraph.MLCTrainingGraphPtr

Nested classes/interfaces inherited from class org.robovm.apple.mlcompute.MLCGraph

MLCGraph.MLCGraphPtr

Nested classes/interfaces inherited from class org.robovm.apple.foundation.NSObject

NSObject.Handle, NSObject.Marshaler, NSObject.NoRetainMarshaler, NSObject.NSKeyValueObserver, NSObject.NSObjectPtr, NSObject.SkipInit

Nested classes/interfaces inherited from class org.robovm.objc.ObjCObject

ObjCObject.ObjCObjectPtr, ObjCObject.Super

Field Summary

Fields inherited from class org.robovm.apple.foundation.NSObject

FLAG_NO_RETAIN

Fields inherited from class org.robovm.objc.ObjCObject

customClass

Constructor Summary

Constructors

Modifier	Constructor	Description
	MLCTrainingGraph()
	MLCTrainingGraph(NSArray<MLCGraph> graphObjects, MLCLayer lossLayer, MLCOptimizer optimizer)
protected	MLCTrainingGraph(NSObject.Handle h, long handle)
protected	MLCTrainingGraph(NSObject.SkipInit skipInit)

Method Summary

Modifier and Type	Method	Description
boolean	addInputs(NSDictionary<NSString,MLCTensor> inputs, NSDictionary<NSString,MLCTensor> lossLabels)
boolean	addInputs(NSDictionary<NSString,MLCTensor> inputs, NSDictionary<NSString,MLCTensor> lossLabels, NSDictionary<NSString,MLCTensor> lossLabelWeights)
boolean	addOutputs(NSDictionary<NSString,MLCTensor> outputs)
MLCTensor	allocateUserGradientForTensor(MLCTensor tensor)
boolean	bindOptimizerData(NSArray<MLCTensorData> data, NSArray<MLCTensorOptimizerDeviceData> deviceData, MLCTensor tensor)
boolean	compile(MLCGraphCompilationOptions options, MLCDevice device)
boolean	compile(MLCGraphCompilationOptions options, MLCDevice device, NSDictionary<NSString,MLCTensor> inputTensors, NSDictionary<NSString,MLCTensorData> inputTensorsData)
boolean	compileOptimizer(MLCOptimizer optimizer)
protected static long	create(NSArray<MLCGraph> graphObjects, MLCLayer lossLayer, MLCOptimizer optimizer)
boolean	execute(NSDictionary<NSString,MLCTensorData> inputsData, NSDictionary<NSString,MLCTensorData> lossLabelsData, NSDictionary<NSString,MLCTensorData> lossLabelWeightsData, long batchSize, MLCExecutionOptions options, VoidBlock3<MLCTensor,NSError,Double> completionHandler)
boolean	execute(NSDictionary<NSString,MLCTensorData> inputsData, NSDictionary<NSString,MLCTensorData> lossLabelsData, NSDictionary<NSString,MLCTensorData> lossLabelWeightsData, NSDictionary<NSString,MLCTensorData> outputsData, long batchSize, MLCExecutionOptions options, VoidBlock3<MLCTensor,NSError,Double> completionHandler)
boolean	executeForward(long batchSize, MLCExecutionOptions options, NSDictionary<NSString,MLCTensorData> outputsData, VoidBlock3<MLCTensor,NSError,Double> completionHandler)
boolean	executeForward(long batchSize, MLCExecutionOptions options, VoidBlock3<MLCTensor,NSError,Double> completionHandler)
boolean	executeGradient(long batchSize, MLCExecutionOptions options, NSDictionary<NSString,MLCTensorData> outputsData, VoidBlock3<MLCTensor,NSError,Double> completionHandler)
boolean	executeGradient(long batchSize, MLCExecutionOptions options, VoidBlock3<MLCTensor,NSError,Double> completionHandler)
boolean	executeOptimizerUpdate(MLCExecutionOptions options, VoidBlock3<MLCTensor,NSError,Double> completionHandler)
long	getDeviceMemorySize()
MLCOptimizer	getOptimizer()
NSData	gradientData(MLCTensor parameter, MLCLayer layer)
MLCTensor	gradientTensorForInput(MLCTensor input)
static MLCTrainingGraph	graph()
boolean	link(NSArray<MLCTrainingGraph> graphs)
NSArray<MLCTensor>	resultGradientTensorsForLayer(MLCLayer layer)
boolean	setTrainingTensorParameters(NSArray<MLCTensorParameter> parameters)
NSArray<MLCTensor>	sourceGradientTensorsForLayer(MLCLayer layer)
boolean	stopGradientForTensors(NSArray<MLCTensor> tensors)
void	synchronizeUpdates()

Methods inherited from class org.robovm.apple.mlcompute.MLCGraph

bindAndWriteData, bindAndWriteData, concatenate, gather, getDevice, getLayers, getSummarizedDOTDescription, node, node, node, node, reshape, resultTensorsForLayer, scatter, select, sourceTensorsForLayer, split, split, transpose

Methods inherited from class org.robovm.apple.foundation.NSObject

addKeyValueObserver, addKeyValueObserver, afterMarshaled, alloc, autorelease, autorelease, awakeFromNib, conformsToProtocol, copy, description, didChangeValue, didChangeValue, didChangeValues, doDispose, equals, forceSkipInit, getAutoContentAccessingProxy, getClassForCoder, getClassForKeyedArchiver, getKeyValueCoder, getObservationInfo, hash, hashCode, init, isEqual, isKindOfClass, isMemberOfClass, mutableCopy, performSelector, performSelector, performSelector, performSelector, performSelector, performSelector, performSelector, performSelectorInBackground, performSelectorOnMainThread, performSelectorOnMainThread, performSelectorV, performSelectorV, performSelectorV, release, release, removeKeyValueObserver, removeKeyValueObservers, respondsToSelector, retain, retain, retainCount, setObservationInfo, toString, willChangeValue, willChangeValue, willChangeValues

Methods inherited from class org.robovm.objc.ObjCObject

addStrongRef, dispose, dispose, finalize, getAssociatedObject, getObjCClass, getPeerObject, getSuper, initObject, logRetainRelease, removeStrongRef, retainCustomObjectFromCb, setAssociatedObject, toObjCObject, toObjCObject, updateStrongRef

Methods inherited from class org.robovm.rt.bro.NativeObject

as, getHandle, setHandle

Methods inherited from class java.lang.Object

clone, getClass, notify, notifyAll, wait, wait, wait

Constructor Details

MLCTrainingGraph

public MLCTrainingGraph()

MLCTrainingGraph

protected MLCTrainingGraph(NSObject.Handle h, long handle)

MLCTrainingGraph

protected MLCTrainingGraph(NSObject.SkipInit skipInit)

MLCTrainingGraph

public MLCTrainingGraph(NSArray<MLCGraph> graphObjects, MLCLayer lossLayer, MLCOptimizer optimizer)

Method Details

getOptimizer

public MLCOptimizer getOptimizer()

getDeviceMemorySize

public long getDeviceMemorySize()

addInputs

public boolean addInputs(NSDictionary<NSString,MLCTensor> inputs, NSDictionary<NSString,MLCTensor> lossLabels)

addInputs

public boolean addInputs(NSDictionary<NSString,MLCTensor> inputs, NSDictionary<NSString,MLCTensor> lossLabels, NSDictionary<NSString,MLCTensor> lossLabelWeights)

addOutputs

public boolean addOutputs(NSDictionary<NSString,MLCTensor> outputs)

stopGradientForTensors

public boolean stopGradientForTensors(NSArray<MLCTensor> tensors)

compile

public boolean compile(MLCGraphCompilationOptions options, MLCDevice device)

compile

public boolean compile(MLCGraphCompilationOptions options, MLCDevice device, NSDictionary<NSString,MLCTensor> inputTensors, NSDictionary<NSString,MLCTensorData> inputTensorsData)

Since:

Available in iOS 14.5 and later.

compileOptimizer

public boolean compileOptimizer(MLCOptimizer optimizer)

link

public boolean link(NSArray<MLCTrainingGraph> graphs)

gradientTensorForInput

public MLCTensor gradientTensorForInput(MLCTensor input)

sourceGradientTensorsForLayer

public NSArray<MLCTensor> sourceGradientTensorsForLayer(MLCLayer layer)

resultGradientTensorsForLayer

public NSArray<MLCTensor> resultGradientTensorsForLayer(MLCLayer layer)

gradientData

public NSData gradientData(MLCTensor parameter, MLCLayer layer)

allocateUserGradientForTensor

public MLCTensor allocateUserGradientForTensor(MLCTensor tensor)

execute

public boolean execute(NSDictionary<NSString,MLCTensorData> inputsData, NSDictionary<NSString,MLCTensorData> lossLabelsData, NSDictionary<NSString,MLCTensorData> lossLabelWeightsData, long batchSize, MLCExecutionOptions options, VoidBlock3<MLCTensor,NSError,Double> completionHandler)

execute

public boolean execute(NSDictionary<NSString,MLCTensorData> inputsData, NSDictionary<NSString,MLCTensorData> lossLabelsData, NSDictionary<NSString,MLCTensorData> lossLabelWeightsData, NSDictionary<NSString,MLCTensorData> outputsData, long batchSize, MLCExecutionOptions options, VoidBlock3<MLCTensor,NSError,Double> completionHandler)

executeForward

public boolean executeForward(long batchSize, MLCExecutionOptions options, VoidBlock3<MLCTensor,NSError,Double> completionHandler)

executeForward

public boolean executeForward(long batchSize, MLCExecutionOptions options, NSDictionary<NSString,MLCTensorData> outputsData, VoidBlock3<MLCTensor,NSError,Double> completionHandler)

executeGradient

public boolean executeGradient(long batchSize, MLCExecutionOptions options, VoidBlock3<MLCTensor,NSError,Double> completionHandler)

executeGradient

public boolean executeGradient(long batchSize, MLCExecutionOptions options, NSDictionary<NSString,MLCTensorData> outputsData, VoidBlock3<MLCTensor,NSError,Double> completionHandler)

executeOptimizerUpdate

public boolean executeOptimizerUpdate(MLCExecutionOptions options, VoidBlock3<MLCTensor,NSError,Double> completionHandler)

synchronizeUpdates

public void synchronizeUpdates()

setTrainingTensorParameters

public boolean setTrainingTensorParameters(NSArray<MLCTensorParameter> parameters)

bindOptimizerData

public boolean bindOptimizerData(NSArray<MLCTensorData> data, NSArray<MLCTensorOptimizerDeviceData> deviceData, MLCTensor tensor)

create

protected static long create(NSArray<MLCGraph> graphObjects, MLCLayer lossLayer, MLCOptimizer optimizer)

graph

public static MLCTrainingGraph graph()