Package org.robovm.apple.metalpsgraph

Class MPSGraph

java.lang.Object

org.robovm.rt.bro.NativeObject

org.robovm.objc.ObjCObject

org.robovm.apple.foundation.NSObject

org.robovm.apple.metalpsgraph.MPSGraph

All Implemented Interfaces:

NSObjectProtocol, ObjCProtocol

public class MPSGraph
extends NSObject

Since:

Available in iOS 14.0 and later.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	MPSGraph.MPSGraphPtr

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
	MPSGraph()
protected	MPSGraph(NSObject.Handle h, long handle)
protected	MPSGraph(NSObject.SkipInit skipInit)

Method Summary

Modifier and Type	Method	Description
MPSGraphTensor	absolute(MPSGraphTensor tensor, String name)
MPSGraphTensor	acos(MPSGraphTensor tensor, String name)
MPSGraphTensor	acosh(MPSGraphTensor tensor, String name)
MPSGraphTensor	addition(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
MPSGraphOperation	applyStochasticGradientDescent(MPSGraphTensor learningRateTensor, MPSGraphVariableOp variable, MPSGraphTensor gradientTensor, String name)
MPSGraphTensor	asin(MPSGraphTensor tensor, String name)
MPSGraphTensor	asinh(MPSGraphTensor tensor, String name)
MPSGraphOperation	assignVariable(MPSGraphTensor variable, MPSGraphTensor tensor, String name)
MPSGraphTensor	atan(MPSGraphTensor tensor, String name)
MPSGraphTensor	atan2(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
MPSGraphTensor	atanh(MPSGraphTensor tensor, String name)
MPSGraphTensor	avgPooling2D(MPSGraphTensor source, MPSGraphPooling2DOpDescriptor descriptor, String name)
MPSGraphTensor	avgPooling2D(MPSGraphTensor gradient, MPSGraphTensor source, MPSGraphPooling2DOpDescriptor descriptor, String name)
MPSGraphTensor	avgPooling4D(MPSGraphTensor source, MPSGraphPooling4DOpDescriptor descriptor, String name)
MPSGraphTensor	avgPooling4DGradient(MPSGraphTensor gradient, MPSGraphTensor source, MPSGraphPooling4DOpDescriptor descriptor, String name)
MPSGraphTensor	bandPart(MPSGraphTensor inputTensor, long numLower, long numUpper, String name)
MPSGraphTensor	bandPart(MPSGraphTensor inputTensor, MPSGraphTensor numLowerTensor, MPSGraphTensor numUpperTensor, String name)
MPSGraphTensor	broadcastTensor(MPSGraphTensor tensor, NSArray<NSNumber> shape, String name)
MPSGraphTensor	broadcastTensor(MPSGraphTensor tensor, MPSGraphTensor shapeTensor, String name)
MPSGraphTensor	castTensor(MPSGraphTensor tensor, MPSDataType type, String name)
MPSGraphTensor	ceil(MPSGraphTensor tensor, String name)
MPSGraphTensor	clamp(MPSGraphTensor tensor, MPSGraphTensor minValueTensor, MPSGraphTensor maxValueTensor, String name)
MPSGraphExecutable	compile(MPSGraphDevice device, NSDictionary<MPSGraphTensor,MPSGraphShapedType> feeds, NSArray<MPSGraphTensor> targetTensors, NSArray<MPSGraphOperation> targetOperations, MPSGraphCompilationDescriptor compilationDescriptor)
MPSGraphTensor	concatTensor(MPSGraphTensor tensor, MPSGraphTensor tensor2, long dimensionIndex, String name)
MPSGraphTensor	concatTensors(NSArray<MPSGraphTensor> tensors, long dimensionIndex, boolean interleave, String name)
MPSGraphTensor	concatTensors(NSArray<MPSGraphTensor> tensors, long dimensionIndex, String name)
MPSGraphTensor	constant(double scalar, NSArray<NSNumber> shape, MPSDataType dataType)
MPSGraphTensor	constant(double scalar, MPSDataType dataType)
MPSGraphTensor	constant(NSData data, NSArray<NSNumber> shape, MPSDataType dataType)
NSArray<MPSGraphTensor>	controlDependency(NSArray<MPSGraphOperation> operations, Block0<NSArray<MPSGraphTensor>> dependentBlock, String name)
MPSGraphTensor	convolution2D(MPSGraphTensor source, MPSGraphTensor weights, MPSGraphConvolution2DOpDescriptor descriptor, String name)
MPSGraphTensor	convolution2DDataGradient(MPSGraphTensor incomingGradient, MPSGraphTensor weights, NSArray<NSNumber> outputShape, MPSGraphConvolution2DOpDescriptor forwardConvolutionDescriptor, String name)
MPSGraphTensor	convolution2DDataGradient(MPSGraphTensor gradient, MPSGraphTensor weights, MPSGraphTensor outputShapeTensor, MPSGraphConvolution2DOpDescriptor forwardConvolutionDescriptor, String name)
MPSGraphTensor	convolution2DWeightsGradient(MPSGraphTensor incomingGradient, MPSGraphTensor source, NSArray<NSNumber> outputShape, MPSGraphConvolution2DOpDescriptor forwardConvolutionDescriptor, String name)
MPSGraphTensor	convolution2DWeightsGradient(MPSGraphTensor gradient, MPSGraphTensor source, MPSGraphTensor outputShapeTensor, MPSGraphConvolution2DOpDescriptor forwardConvolutionDescriptor, String name)
MPSGraphTensor	convolutionTranspose2D(MPSGraphTensor source, MPSGraphTensor weights, NSArray<NSNumber> outputShape, MPSGraphConvolution2DOpDescriptor descriptor, String name)
MPSGraphTensor	convolutionTranspose2D(MPSGraphTensor source, MPSGraphTensor weights, MPSGraphTensor outputShape, MPSGraphConvolution2DOpDescriptor descriptor, String name)
MPSGraphTensor	convolutionTranspose2DDataGradient(MPSGraphTensor incomingGradient, MPSGraphTensor weights, NSArray<NSNumber> outputShape, MPSGraphConvolution2DOpDescriptor forwardConvolutionDescriptor, String name)
MPSGraphTensor	convolutionTranspose2DDataGradient(MPSGraphTensor incomingGradient, MPSGraphTensor weights, MPSGraphTensor outputShape, MPSGraphConvolution2DOpDescriptor forwardConvolutionDescriptor, String name)
MPSGraphTensor	convolutionTranspose2DWeightsGradient(MPSGraphTensor incomingGradientTensor, MPSGraphTensor source, NSArray<NSNumber> outputShape, MPSGraphConvolution2DOpDescriptor forwardConvolutionDescriptor, String name)
MPSGraphTensor	convolutionTranspose2DWeightsGradient(MPSGraphTensor incomingGradientTensor, MPSGraphTensor source, MPSGraphTensor outputShape, MPSGraphConvolution2DOpDescriptor forwardConvolutionDescriptor, String name)
MPSGraphTensor	coordinateAlongAxis(long axis, NSArray<NSNumber> shape, String name)
MPSGraphTensor	coordinateAlongAxis(long axis, MPSGraphTensor shapeTensor, String name)
MPSGraphTensor	coordinateAlongAxisTensor(MPSGraphTensor axisTensor, NSArray<NSNumber> shape, String name)
MPSGraphTensor	coordinateAlongAxisTensor(MPSGraphTensor axisTensor, MPSGraphTensor shapeTensor, String name)
MPSGraphTensor	cos(MPSGraphTensor tensor, String name)
MPSGraphTensor	cosh(MPSGraphTensor tensor, String name)
MPSGraphTensor	depthToSpace2DTensor(MPSGraphTensor tensor, long widthAxis, long heightAxis, long depthAxis, long blockSize, boolean usePixelShuffleOrder, String name)
MPSGraphTensor	depthToSpace2DTensor(MPSGraphTensor tensor, MPSGraphTensor widthAxisTensor, MPSGraphTensor heightAxisTensor, MPSGraphTensor depthAxisTensor, long blockSize, boolean usePixelShuffleOrder, String name)
MPSGraphTensor	depthwiseConvolution2D(MPSGraphTensor source, MPSGraphTensor weights, MPSGraphDepthwiseConvolution2DOpDescriptor descriptor, String name)
MPSGraphTensor	depthwiseConvolution2DDataGradient(MPSGraphTensor incomingGradient, MPSGraphTensor weights, NSArray<NSNumber> outputShape, MPSGraphDepthwiseConvolution2DOpDescriptor descriptor, String name)
MPSGraphTensor	depthwiseConvolution2DWeightsGradient(MPSGraphTensor incomingGradient, MPSGraphTensor source, NSArray<NSNumber> outputShape, MPSGraphDepthwiseConvolution2DOpDescriptor descriptor, String name)
MPSGraphTensor	depthwiseConvolution3D(MPSGraphTensor source, MPSGraphTensor weights, MPSGraphDepthwiseConvolution3DOpDescriptor descriptor, String name)
MPSGraphTensor	depthwiseConvolution3DDataGradient(MPSGraphTensor incomingGradient, MPSGraphTensor weights, NSArray<NSNumber> outputShape, MPSGraphDepthwiseConvolution3DOpDescriptor descriptor, String name)
MPSGraphTensor	depthwiseConvolution3DWeightsGradient(MPSGraphTensor incomingGradient, MPSGraphTensor source, NSArray<NSNumber> outputShape, MPSGraphDepthwiseConvolution3DOpDescriptor descriptor, String name)
MPSGraphTensor	division(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
MPSGraphTensor	divisionNoNaN(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
NSArray<MPSGraphTensor>	doIf(MPSGraphTensor predicateTensor, Block0<NSArray<MPSGraphTensor>> thenBlock, Block0<NSArray<MPSGraphTensor>> elseBlock, String name)
MPSGraphTensor	dropout(MPSGraphTensor tensor, double rate, String name)
MPSGraphTensor	dropout(MPSGraphTensor tensor, MPSGraphTensor rate, String name)
void	encodeToCommandBuffer(MPSCommandBuffer commandBuffer, NSDictionary<MPSGraphTensor,MPSGraphTensorData> feeds, NSArray<MPSGraphOperation> targetOperations, NSDictionary<MPSGraphTensor,MPSGraphTensorData> resultsDictionary, MPSGraphExecutionDescriptor executionDescriptor)
NSDictionary<MPSGraphTensor,MPSGraphTensorData>	encodeToCommandBuffer(MPSCommandBuffer commandBuffer, NSDictionary<MPSGraphTensor,MPSGraphTensorData> feeds, NSArray<MPSGraphTensor> targetTensors, NSArray<MPSGraphOperation> targetOperations, MPSGraphExecutionDescriptor executionDescriptor)
MPSGraphTensor	equal(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
MPSGraphTensor	erf(MPSGraphTensor tensor, String name)
MPSGraphTensor	expandDimsOfTensor(MPSGraphTensor tensor, long axis, String name)
MPSGraphTensor	expandDimsOfTensor(MPSGraphTensor tensor, NSArray<NSNumber> axes, String name)
MPSGraphTensor	expandDimsOfTensor(MPSGraphTensor tensor, MPSGraphTensor axesTensor, String name)
MPSGraphTensor	exponent(MPSGraphTensor tensor, String name)
MPSGraphTensor	exponentBase10(MPSGraphTensor tensor, String name)
MPSGraphTensor	exponentBase2(MPSGraphTensor tensor, String name)
MPSGraphTensor	flatten2DTensor(MPSGraphTensor tensor, long axis, String name)
MPSGraphTensor	flatten2DTensor(MPSGraphTensor tensor, MPSGraphTensor axisTensor, String name)
MPSGraphTensor	floor(MPSGraphTensor tensor, String name)
MPSGraphTensor	floorModulo(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
NSArray<MPSGraphTensor>	forLoop(MPSGraphTensor numberOfIterations, NSArray<MPSGraphTensor> initialBodyArguments, Block2<MPSGraphTensor,NSArray<MPSGraphTensor>,NSArray<MPSGraphTensor>> body, String name)
NSArray<MPSGraphTensor>	forLoop(MPSGraphTensor lowerBound, MPSGraphTensor upperBound, MPSGraphTensor step, NSArray<MPSGraphTensor> initialBodyArguments, Block2<MPSGraphTensor,NSArray<MPSGraphTensor>,NSArray<MPSGraphTensor>> body, String name)
MPSGraphTensor	gather(MPSGraphTensor updatesTensor, MPSGraphTensor indicesTensor, long axis, long batchDimensions, String name)
MPSGraphTensor	gatherAlongAxis(long axis, MPSGraphTensor updatesTensor, MPSGraphTensor indicesTensor, String name)
MPSGraphTensor	gatherAlongAxisTensor(MPSGraphTensor axisTensor, MPSGraphTensor updatesTensor, MPSGraphTensor indicesTensor, String name)
MPSGraphTensor	gatherND(MPSGraphTensor updatesTensor, MPSGraphTensor indicesTensor, long batchDimensions, String name)
MPSGraphOptions	getOptions()
NSArray<MPSGraphTensor>	getPlaceholderTensors()
NSDictionary<MPSGraphTensor,MPSGraphTensor>	gradientForPrimaryTensor(MPSGraphTensor primaryTensor, NSArray<MPSGraphTensor> tensors, String name)
MPSGraphTensor	greaterThan(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
MPSGraphTensor	greaterThanOrEqual(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
MPSGraphTensor	identity(MPSGraphTensor tensor, String name)
MPSGraphTensor	isFinite(MPSGraphTensor tensor, String name)
MPSGraphTensor	isInfinite(MPSGraphTensor tensor, String name)
MPSGraphTensor	isNaN(MPSGraphTensor tensor, String name)
MPSGraphTensor	L2NormPooling4D(MPSGraphTensor source, MPSGraphPooling4DOpDescriptor descriptor, String name)
MPSGraphTensor	L2NormPooling4DGradient(MPSGraphTensor gradient, MPSGraphTensor source, MPSGraphPooling4DOpDescriptor descriptor, String name)
MPSGraphTensor	leakyReLU(MPSGraphTensor tensor, double alpha, String name)
MPSGraphTensor	leakyReLU(MPSGraphTensor tensor, MPSGraphTensor alphaTensor, String name)
MPSGraphTensor	leakyReLU(MPSGraphTensor gradient, MPSGraphTensor source, MPSGraphTensor alphaTensor, String name)
MPSGraphTensor	lessThan(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
MPSGraphTensor	lessThanOrEqual(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
MPSGraphTensor	logarithm(MPSGraphTensor tensor, String name)
MPSGraphTensor	logarithmBase10(MPSGraphTensor tensor, String name)
MPSGraphTensor	logarithmBase2(MPSGraphTensor tensor, String name)
MPSGraphTensor	logicalAND(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
MPSGraphTensor	logicalNAND(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
MPSGraphTensor	logicalNOR(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
MPSGraphTensor	logicalOR(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
MPSGraphTensor	logicalXNOR(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
MPSGraphTensor	logicalXOR(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
MPSGraphTensor	matrixMultiplication(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
MPSGraphTensor	maximum(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
MPSGraphTensor	maximumWithNaNPropagation(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
MPSGraphTensor	maxPooling2D(MPSGraphTensor source, MPSGraphPooling2DOpDescriptor descriptor, String name)
MPSGraphTensor	maxPooling2DGradient(MPSGraphTensor gradient, MPSGraphTensor source, MPSGraphPooling2DOpDescriptor descriptor, String name)
NSArray<MPSGraphTensor>	maxPooling2DReturnIndices(MPSGraphTensor source, MPSGraphPooling2DOpDescriptor descriptor, String name)
MPSGraphTensor	maxPooling4D(MPSGraphTensor source, MPSGraphPooling4DOpDescriptor descriptor, String name)
MPSGraphTensor	maxPooling4DGradient(MPSGraphTensor gradient, MPSGraphTensor source, MPSGraphPooling4DOpDescriptor descriptor, String name)
NSArray<MPSGraphTensor>	maxPooling4DReturnIndices(MPSGraphTensor source, MPSGraphPooling4DOpDescriptor descriptor, String name)
MPSGraphTensor	mean(MPSGraphTensor tensor, NSArray<NSNumber> axes, String name)
MPSGraphTensor	minimum(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
MPSGraphTensor	minimumWithNaNPropagation(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
MPSGraphTensor	modulo(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
MPSGraphTensor	multiplication(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
MPSGraphTensor	negative(MPSGraphTensor tensor, String name)
MPSGraphTensor	normalization(MPSGraphTensor tensor, MPSGraphTensor mean, MPSGraphTensor variance, MPSGraphTensor gamma, MPSGraphTensor beta, float epsilon, String name)
MPSGraphTensor	normalizationBetaGradient(MPSGraphTensor incomingGradientTensor, MPSGraphTensor sourceTensor, NSArray<NSNumber> axes, String name)
MPSGraphTensor	normalizationGammaGradient(MPSGraphTensor incomingGradientTensor, MPSGraphTensor sourceTensor, MPSGraphTensor meanTensor, MPSGraphTensor varianceTensor, NSArray<NSNumber> axes, float epsilon, String name)
MPSGraphTensor	normalizationGradient(MPSGraphTensor incomingGradientTensor, MPSGraphTensor sourceTensor, MPSGraphTensor meanTensor, MPSGraphTensor varianceTensor, MPSGraphTensor gamma, MPSGraphTensor gammaGradient, MPSGraphTensor betaGradient, NSArray<NSNumber> axes, float epsilon, String name)
MPSGraphTensor	not(MPSGraphTensor tensor, String name)
MPSGraphTensor	notEqual(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
MPSGraphTensor	oneHot(MPSGraphTensor indicesTensor, long depth, long axis, String name)
MPSGraphTensor	oneHot(MPSGraphTensor indicesTensor, long depth, long axis, MPSDataType dataType, double onValue, double offValue, String name)
MPSGraphTensor	oneHot(MPSGraphTensor indicesTensor, long depth, long axis, MPSDataType dataType, String name)
MPSGraphTensor	oneHot(MPSGraphTensor indicesTensor, long depth, String name)
MPSGraphTensor	oneHot(MPSGraphTensor indicesTensor, long depth, MPSDataType dataType, double onValue, double offValue, String name)
MPSGraphTensor	oneHot(MPSGraphTensor indicesTensor, long depth, MPSDataType dataType, String name)
MPSGraphTensor	padGradient(MPSGraphTensor incomingGradientTensor, MPSGraphTensor sourceTensor, MPSGraphPaddingMode paddingMode, NSArray<NSNumber> leftPadding, NSArray<NSNumber> rightPadding, String name)
MPSGraphTensor	padTensor(MPSGraphTensor tensor, MPSGraphPaddingMode paddingMode, NSArray<NSNumber> leftPadding, NSArray<NSNumber> rightPadding, double constantValue, String name)
MPSGraphTensor	placeholder(NSArray<NSNumber> shape, String name)
MPSGraphTensor	placeholder(NSArray<NSNumber> shape, MPSDataType dataType, String name)
MPSGraphTensor	power(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
MPSGraphTensor	randomPhiloxStateTensor(long counterLow, long counterHigh, long key, String name)
MPSGraphTensor	randomPhiloxStateTensor(long seed, String name)
MPSGraphTensor	randomTensor(NSArray<NSNumber> shape, MPSGraphRandomOpDescriptor descriptor, long seed, String name)
MPSGraphTensor	randomTensor(NSArray<NSNumber> shape, MPSGraphRandomOpDescriptor descriptor, String name)
NSArray<MPSGraphTensor>	randomTensor(NSArray<NSNumber> shape, MPSGraphRandomOpDescriptor descriptor, MPSGraphTensor state, String name)
MPSGraphTensor	randomTensor(MPSGraphTensor shapeTensor, MPSGraphRandomOpDescriptor descriptor, long seed, String name)
MPSGraphTensor	randomTensor(MPSGraphTensor shapeTensor, MPSGraphRandomOpDescriptor descriptor, String name)
NSArray<MPSGraphTensor>	randomTensor(MPSGraphTensor shapeTensor, MPSGraphRandomOpDescriptor descriptor, MPSGraphTensor state, String name)
MPSGraphTensor	randomUniformTensor(NSArray<NSNumber> shape, long seed, String name)
MPSGraphTensor	randomUniformTensor(NSArray<NSNumber> shape, String name)
NSArray<MPSGraphTensor>	randomUniformTensor(NSArray<NSNumber> shape, MPSGraphTensor state, String name)
MPSGraphTensor	randomUniformTensor(MPSGraphTensor shapeTensor, long seed, String name)
MPSGraphTensor	randomUniformTensor(MPSGraphTensor shapeTensor, String name)
NSArray<MPSGraphTensor>	randomUniformTensor(MPSGraphTensor shapeTensor, MPSGraphTensor state, String name)
MPSGraphTensor	readVariable(MPSGraphTensor variable, String name)
MPSGraphTensor	reciprocal(MPSGraphTensor tensor, String name)
MPSGraphTensor	reductionAnd(MPSGraphTensor tensor, long axis, String name)
MPSGraphTensor	reductionAnd(MPSGraphTensor tensor, NSArray<NSNumber> axes, String name)
MPSGraphTensor	reductionArgMaximum(MPSGraphTensor tensor, long axis, String name)
MPSGraphTensor	reductionArgMinimum(MPSGraphTensor tensor, long axis, String name)
MPSGraphTensor	reductionMaximum(MPSGraphTensor tensor, long axis, String name)
MPSGraphTensor	reductionMaximum(MPSGraphTensor tensor, NSArray<NSNumber> axes, String name)
MPSGraphTensor	reductionMaximumPropagateNaN(MPSGraphTensor tensor, long axis, String name)
MPSGraphTensor	reductionMaximumPropagateNaN(MPSGraphTensor tensor, NSArray<NSNumber> axes, String name)
MPSGraphTensor	reductionMinimum(MPSGraphTensor tensor, long axis, String name)
MPSGraphTensor	reductionMinimum(MPSGraphTensor tensor, NSArray<NSNumber> axes, String name)
MPSGraphTensor	reductionMinimumPropagateNa(MPSGraphTensor tensor, long axis, String name)
MPSGraphTensor	reductionMinimumPropagateNaN(MPSGraphTensor tensor, NSArray<NSNumber> axes, String name)
MPSGraphTensor	reductionOr(MPSGraphTensor tensor, long axis, String name)
MPSGraphTensor	reductionOr(MPSGraphTensor tensor, NSArray<NSNumber> axes, String name)
MPSGraphTensor	reductionProduct(MPSGraphTensor tensor, long axis, String name)
MPSGraphTensor	reductionProduct(MPSGraphTensor tensor, NSArray<NSNumber> axes, String name)
MPSGraphTensor	reductionSum(MPSGraphTensor tensor, long axis, String name)
MPSGraphTensor	reductionSum(MPSGraphTensor tensor, NSArray<NSNumber> axes, String name)
MPSGraphTensor	reLU(MPSGraphTensor tensor, String name)
MPSGraphTensor	reLUGradient(MPSGraphTensor gradient, MPSGraphTensor source, String name)
MPSGraphTensor	reshapeTensor(MPSGraphTensor tensor, NSArray<NSNumber> shape, String name)
MPSGraphTensor	reshapeTensor(MPSGraphTensor tensor, MPSGraphTensor shapeTensor, String name)
MPSGraphTensor	resize(MPSGraphTensor gradient, MPSGraphTensor input, MPSGraphResizeMode mode, boolean centerResult, boolean alignCorners, MPSGraphTensorNamedDataLayout layout, String name)
MPSGraphTensor	resizeTensor(MPSGraphTensor imagesTensor, NSArray<NSNumber> size, MPSGraphResizeMode mode, boolean centerResult, boolean alignCorners, MPSGraphTensorNamedDataLayout layout, String name)
MPSGraphTensor	resizeTensor(MPSGraphTensor imagesTensor, MPSGraphTensor size, MPSGraphResizeMode mode, boolean centerResult, boolean alignCorners, MPSGraphTensorNamedDataLayout layout, String name)
MPSGraphTensor	reverseSquareRoot(MPSGraphTensor tensor, String name)
MPSGraphTensor	reverseTensor(MPSGraphTensor tensor, String name)
MPSGraphTensor	reverseTensor(MPSGraphTensor tensor, NSArray<NSNumber> axes, String name)
MPSGraphTensor	reverseTensor(MPSGraphTensor tensor, MPSGraphTensor axesTensor, String name)
MPSGraphTensor	rint(MPSGraphTensor tensor, String name)
MPSGraphTensor	round(MPSGraphTensor tensor, String name)
NSDictionary<MPSGraphTensor,MPSGraphTensorData>	run(NSDictionary<MPSGraphTensor,MPSGraphTensorData> feeds, NSArray<MPSGraphTensor> targetTensors, NSArray<MPSGraphOperation> targetOperations)
void	run(MTLCommandQueue commandQueue, NSDictionary<MPSGraphTensor,MPSGraphTensorData> feeds, NSArray<MPSGraphOperation> targetOperations, NSDictionary<MPSGraphTensor,MPSGraphTensorData> resultsDictionary)
NSDictionary<MPSGraphTensor,MPSGraphTensorData>	run(MTLCommandQueue commandQueue, NSDictionary<MPSGraphTensor,MPSGraphTensorData> feeds, NSArray<MPSGraphTensor> targetTensors, NSArray<MPSGraphOperation> targetOperations)
NSDictionary<MPSGraphTensor,MPSGraphTensorData>	runAsync(NSDictionary<MPSGraphTensor,MPSGraphTensorData> feeds, NSArray<MPSGraphTensor> targetTensors, NSArray<MPSGraphOperation> targetOperations, MPSGraphExecutionDescriptor executionDescriptor)
void	runAsync(MTLCommandQueue commandQueue, NSDictionary<MPSGraphTensor,MPSGraphTensorData> feeds, NSArray<MPSGraphOperation> targetOperations, NSDictionary<MPSGraphTensor,MPSGraphTensorData> resultsDictionary, MPSGraphExecutionDescriptor executionDescriptor)
NSDictionary<MPSGraphTensor,MPSGraphTensorData>	runAsync(MTLCommandQueue commandQueue, NSDictionary<MPSGraphTensor,MPSGraphTensorData> feeds, NSArray<MPSGraphTensor> targetTensors, NSArray<MPSGraphOperation> targetOperations, MPSGraphExecutionDescriptor executionDescriptor)
MPSGraphTensor	scatter(MPSGraphTensor updatesTensor, MPSGraphTensor indicesTensor, NSArray<NSNumber> shape, long axis, MPSGraphScatterMode mode, String name)
MPSGraphTensor	scatter(MPSGraphTensor dataTensor, MPSGraphTensor updatesTensor, MPSGraphTensor indicesTensor, long axis, MPSGraphScatterMode mode, String name)
MPSGraphTensor	scatterAlongAxis(long axis, MPSGraphTensor updatesTensor, MPSGraphTensor indicesTensor, NSArray<NSNumber> shape, MPSGraphScatterMode mode, String name)
MPSGraphTensor	scatterAlongAxis(long axis, MPSGraphTensor dataTensor, MPSGraphTensor updatesTensor, MPSGraphTensor indicesTensor, MPSGraphScatterMode mode, String name)
MPSGraphTensor	scatterAlongAxisTensor(MPSGraphTensor axisTensor, MPSGraphTensor updatesTensor, MPSGraphTensor indicesTensor, NSArray<NSNumber> shape, MPSGraphScatterMode mode, String name)
MPSGraphTensor	scatterAlongAxisTensor(MPSGraphTensor axisTensor, MPSGraphTensor dataTensor, MPSGraphTensor updatesTensor, MPSGraphTensor indicesTensor, MPSGraphScatterMode mode, String name)
MPSGraphTensor	scatterND(MPSGraphTensor updatesTensor, MPSGraphTensor indicesTensor, NSArray<NSNumber> shape, long batchDimensions, String name)
MPSGraphTensor	scatterND(MPSGraphTensor updatesTensor, MPSGraphTensor indicesTensor, NSArray<NSNumber> shape, long batchDimensions, MPSGraphScatterMode mode, String name)
MPSGraphTensor	scatterND(MPSGraphTensor dataTensor, MPSGraphTensor updatesTensor, MPSGraphTensor indicesTensor, long batchDimensions, MPSGraphScatterMode mode, String name)
MPSGraphTensor	select(MPSGraphTensor predicateTensor, MPSGraphTensor truePredicateTensor, MPSGraphTensor falseSelectTensor, String name)
void	setOptions(MPSGraphOptions v)
MPSGraphTensor	shapeOfTensor(MPSGraphTensor tensor, String name)
MPSGraphTensor	sigmoid(MPSGraphTensor tensor, String name)
MPSGraphTensor	sigmoidGradient(MPSGraphTensor gradient, MPSGraphTensor source, String name)
MPSGraphTensor	sign(MPSGraphTensor tensor, String name)
MPSGraphTensor	signbit(MPSGraphTensor tensor, String name)
MPSGraphTensor	sin(MPSGraphTensor tensor, String name)
MPSGraphTensor	sinh(MPSGraphTensor tensor, String name)
MPSGraphTensor	sliceGradientTensor(MPSGraphTensor inputGradientTensor, MPSGraphTensor fwdInShapeTensor, NSArray<NSNumber> starts, NSArray<NSNumber> ends, NSArray<NSNumber> strides, int startMask, int endMask, int squeezeMask, String name)
MPSGraphTensor	sliceGradientTensor(MPSGraphTensor inputGradientTensor, MPSGraphTensor fwdInShapeTensor, NSArray<NSNumber> starts, NSArray<NSNumber> ends, NSArray<NSNumber> strides, String name)
MPSGraphTensor	sliceTensor(MPSGraphTensor tensor, long dimensionIndex, long start, long length, String name)
MPSGraphTensor	sliceTensor(MPSGraphTensor tensor, NSArray<NSNumber> starts, NSArray<NSNumber> ends, NSArray<NSNumber> strides, int startMask, int endMask, int squeezeMask, String name)
MPSGraphTensor	sliceTensor(MPSGraphTensor tensor, NSArray<NSNumber> starts, NSArray<NSNumber> ends, NSArray<NSNumber> strides, String name)
MPSGraphTensor	softMax(MPSGraphTensor tensor, long axis, String name)
MPSGraphTensor	softMaxCrossEntropy(MPSGraphTensor sourceTensor, MPSGraphTensor labelsTensor, long axis, MPSGraphLossReductionType reductionType, String name)
MPSGraphTensor	softMaxCrossEntropyGradient(MPSGraphTensor gradientTensor, MPSGraphTensor sourceTensor, MPSGraphTensor labelsTensor, long axis, MPSGraphLossReductionType reductionType, String name)
MPSGraphTensor	softMaxGradient(MPSGraphTensor gradient, MPSGraphTensor source, long axis, String name)
MPSGraphTensor	spaceToDepth2DTensor(MPSGraphTensor tensor, long widthAxis, long heightAxis, long depthAxis, long blockSize, boolean usePixelShuffleOrder, String name)
MPSGraphTensor	spaceToDepth2DTensor(MPSGraphTensor tensor, MPSGraphTensor widthAxisTensor, MPSGraphTensor heightAxisTensor, MPSGraphTensor depthAxisTensor, long blockSize, boolean usePixelShuffleOrder, String name)
MPSGraphTensor	sparseTensor(MPSGraphCreateSparseOpDescriptor sparseDescriptor, NSArray<MPSGraphTensor> inputTensorArray, NSArray<NSNumber> shape, String name)
MPSGraphTensor	sparseTensor(MPSGraphSparseStorageType sparseStorageType, NSArray<MPSGraphTensor> inputTensorArray, NSArray<NSNumber> shape, MPSDataType dataType, String name)
NSArray<MPSGraphTensor>	splitTensor(MPSGraphTensor tensor, long numSplits, long axis, String name)
NSArray<MPSGraphTensor>	splitTensor(MPSGraphTensor tensor, NSArray<NSNumber> splitSizes, long axis, String name)
NSArray<MPSGraphTensor>	splitTensor(MPSGraphTensor tensor, MPSGraphTensor splitSizesTensor, long axis, String name)
MPSGraphTensor	square(MPSGraphTensor tensor, String name)
MPSGraphTensor	squareRoot(MPSGraphTensor tensor, String name)
MPSGraphTensor	squeezeTensor(MPSGraphTensor tensor, long axis, String name)
MPSGraphTensor	squeezeTensor(MPSGraphTensor tensor, String name)
MPSGraphTensor	squeezeTensor(MPSGraphTensor tensor, NSArray<NSNumber> axes, String name)
MPSGraphTensor	squeezeTensor(MPSGraphTensor tensor, MPSGraphTensor axesTensor, String name)
MPSGraphTensor	stackTensors(NSArray<MPSGraphTensor> inputTensors, long axis, String name)
MPSGraphTensor	stencil(MPSGraphTensor source, MPSGraphTensor weights, MPSGraphStencilOpDescriptor descriptor, String name)
MPSGraphTensor	stochasticGradientDescent(MPSGraphTensor learningRateTensor, MPSGraphTensor valuesTensor, MPSGraphTensor gradientTensor, String name)
MPSGraphTensor	subtraction(MPSGraphTensor primaryTensor, MPSGraphTensor secondaryTensor, String name)
MPSGraphTensor	tan(MPSGraphTensor tensor, String name)
MPSGraphTensor	tanh(MPSGraphTensor tensor, String name)
MPSGraphTensor	tileGradient(MPSGraphTensor incomingGradientTensor, MPSGraphTensor sourceTensor, NSArray<NSNumber> multiplier, String name)
MPSGraphTensor	tileTensor(MPSGraphTensor tensor, NSArray<NSNumber> multiplier, String name)
NSArray<MPSGraphTensor>	topK(MPSGraphTensor source, long k, String name)
MPSGraphTensor	topK(MPSGraphTensor gradient, MPSGraphTensor source, long k, String name)
NSArray<MPSGraphTensor>	topK(MPSGraphTensor source, MPSGraphTensor kTensor, String name)
MPSGraphTensor	topK(MPSGraphTensor gradient, MPSGraphTensor source, MPSGraphTensor kTensor, String name)
MPSGraphTensor	transposeTensor(MPSGraphTensor tensor, long dimensionIndex, long dimensionIndex2, String name)
MPSGraphTensor	variable(NSData data, NSArray<NSNumber> shape, MPSDataType dataType, String name)
MPSGraphTensor	variance(MPSGraphTensor tensor, NSArray<NSNumber> axes, String name)
MPSGraphTensor	variance(MPSGraphTensor tensor, MPSGraphTensor meanTensor, NSArray<NSNumber> axes, String name)
NSArray<MPSGraphTensor>	whileLoop(NSArray<MPSGraphTensor> initialInputs, Block2<NSArray<MPSGraphTensor>,NSMutableArray<MPSGraphTensor>,MPSGraphTensor> before, Block1<NSArray<MPSGraphTensor>,NSArray<MPSGraphTensor>> after, String name)

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

MPSGraph

public MPSGraph()

MPSGraph

protected MPSGraph(NSObject.Handle h,
long handle)

MPSGraph

protected MPSGraph(NSObject.SkipInit skipInit)

Method Details

getOptions

public MPSGraphOptions getOptions()

setOptions

public void setOptions(MPSGraphOptions v)

getPlaceholderTensors

public NSArray<MPSGraphTensor> getPlaceholderTensors()

compile

public MPSGraphExecutable compile(MPSGraphDevice device,
NSDictionary<MPSGraphTensor,MPSGraphShapedType> feeds,
NSArray<MPSGraphTensor> targetTensors,
NSArray<MPSGraphOperation> targetOperations,
MPSGraphCompilationDescriptor compilationDescriptor)

Since:

Available in iOS 15.0 and later.

run

public NSDictionary<MPSGraphTensor,MPSGraphTensorData> run(NSDictionary<MPSGraphTensor,MPSGraphTensorData> feeds,
NSArray<MPSGraphTensor> targetTensors,
NSArray<MPSGraphOperation> targetOperations)

run

public NSDictionary<MPSGraphTensor,MPSGraphTensorData> run(MTLCommandQueue commandQueue,
NSDictionary<MPSGraphTensor,MPSGraphTensorData> feeds,
NSArray<MPSGraphTensor> targetTensors,
NSArray<MPSGraphOperation> targetOperations)

run

public void run(MTLCommandQueue commandQueue,
NSDictionary<MPSGraphTensor,MPSGraphTensorData> feeds,
NSArray<MPSGraphOperation> targetOperations,
NSDictionary<MPSGraphTensor,MPSGraphTensorData> resultsDictionary)

runAsync

public NSDictionary<MPSGraphTensor,MPSGraphTensorData> runAsync(NSDictionary<MPSGraphTensor,MPSGraphTensorData> feeds,
NSArray<MPSGraphTensor> targetTensors,
NSArray<MPSGraphOperation> targetOperations,
MPSGraphExecutionDescriptor executionDescriptor)

runAsync

public NSDictionary<MPSGraphTensor,MPSGraphTensorData> runAsync(MTLCommandQueue commandQueue,
NSDictionary<MPSGraphTensor,MPSGraphTensorData> feeds,
NSArray<MPSGraphTensor> targetTensors,
NSArray<MPSGraphOperation> targetOperations,
MPSGraphExecutionDescriptor executionDescriptor)

runAsync

public void runAsync(MTLCommandQueue commandQueue,
NSDictionary<MPSGraphTensor,MPSGraphTensorData> feeds,
NSArray<MPSGraphOperation> targetOperations,
NSDictionary<MPSGraphTensor,MPSGraphTensorData> resultsDictionary,
MPSGraphExecutionDescriptor executionDescriptor)

encodeToCommandBuffer

public NSDictionary<MPSGraphTensor,MPSGraphTensorData> encodeToCommandBuffer(MPSCommandBuffer commandBuffer,
NSDictionary<MPSGraphTensor,MPSGraphTensorData> feeds,
NSArray<MPSGraphTensor> targetTensors,
NSArray<MPSGraphOperation> targetOperations,
MPSGraphExecutionDescriptor executionDescriptor)

encodeToCommandBuffer

public void encodeToCommandBuffer(MPSCommandBuffer commandBuffer,
NSDictionary<MPSGraphTensor,MPSGraphTensorData> feeds,
NSArray<MPSGraphOperation> targetOperations,
NSDictionary<MPSGraphTensor,MPSGraphTensorData> resultsDictionary,
MPSGraphExecutionDescriptor executionDescriptor)

gradientForPrimaryTensor

public NSDictionary<MPSGraphTensor,MPSGraphTensor> gradientForPrimaryTensor(MPSGraphTensor primaryTensor,
NSArray<MPSGraphTensor> tensors,
String name)

reLU

public MPSGraphTensor reLU(MPSGraphTensor tensor,
String name)

reLUGradient

public MPSGraphTensor reLUGradient(MPSGraphTensor gradient,
MPSGraphTensor source,
String name)

sigmoid

public MPSGraphTensor sigmoid(MPSGraphTensor tensor,
String name)

sigmoidGradient

public MPSGraphTensor sigmoidGradient(MPSGraphTensor gradient,
MPSGraphTensor source,
String name)

softMax

public MPSGraphTensor softMax(MPSGraphTensor tensor,
long axis,
String name)

softMaxGradient

public MPSGraphTensor softMaxGradient(MPSGraphTensor gradient,
MPSGraphTensor source,
long axis,
String name)

leakyReLU

public MPSGraphTensor leakyReLU(MPSGraphTensor tensor,
double alpha,
String name)

Since:

Available in iOS 15.0 and later.

leakyReLU

public MPSGraphTensor leakyReLU(MPSGraphTensor tensor,
MPSGraphTensor alphaTensor,
String name)

Since:

Available in iOS 15.0 and later.

leakyReLU

public MPSGraphTensor leakyReLU(MPSGraphTensor gradient,
MPSGraphTensor source,
MPSGraphTensor alphaTensor,
String name)

Since:

Available in iOS 15.0 and later.

identity

public MPSGraphTensor identity(MPSGraphTensor tensor,
String name)

exponent

public MPSGraphTensor exponent(MPSGraphTensor tensor,
String name)

exponentBase2

public MPSGraphTensor exponentBase2(MPSGraphTensor tensor,
String name)

exponentBase10

public MPSGraphTensor exponentBase10(MPSGraphTensor tensor,
String name)

logarithm

public MPSGraphTensor logarithm(MPSGraphTensor tensor,
String name)

logarithmBase2

public MPSGraphTensor logarithmBase2(MPSGraphTensor tensor,
String name)

logarithmBase10

public MPSGraphTensor logarithmBase10(MPSGraphTensor tensor,
String name)

square

public MPSGraphTensor square(MPSGraphTensor tensor,
String name)

squareRoot

public MPSGraphTensor squareRoot(MPSGraphTensor tensor,
String name)

reverseSquareRoot

public MPSGraphTensor reverseSquareRoot(MPSGraphTensor tensor,
String name)

reciprocal

public MPSGraphTensor reciprocal(MPSGraphTensor tensor,
String name)

absolute

public MPSGraphTensor absolute(MPSGraphTensor tensor,
String name)

negative

public MPSGraphTensor negative(MPSGraphTensor tensor,
String name)

sign

public MPSGraphTensor sign(MPSGraphTensor tensor,
String name)

signbit

public MPSGraphTensor signbit(MPSGraphTensor tensor,
String name)

ceil

public MPSGraphTensor ceil(MPSGraphTensor tensor,
String name)

floor

public MPSGraphTensor floor(MPSGraphTensor tensor,
String name)

round

public MPSGraphTensor round(MPSGraphTensor tensor,
String name)

rint

public MPSGraphTensor rint(MPSGraphTensor tensor,
String name)

sin

public MPSGraphTensor sin(MPSGraphTensor tensor,
String name)

cos

public MPSGraphTensor cos(MPSGraphTensor tensor,
String name)

tan

public MPSGraphTensor tan(MPSGraphTensor tensor,
String name)

sinh

public MPSGraphTensor sinh(MPSGraphTensor tensor,
String name)

cosh

public MPSGraphTensor cosh(MPSGraphTensor tensor,
String name)

tanh

public MPSGraphTensor tanh(MPSGraphTensor tensor,
String name)

asin

public MPSGraphTensor asin(MPSGraphTensor tensor,
String name)

acos

public MPSGraphTensor acos(MPSGraphTensor tensor,
String name)

atan

public MPSGraphTensor atan(MPSGraphTensor tensor,
String name)

asinh

public MPSGraphTensor asinh(MPSGraphTensor tensor,
String name)

acosh

public MPSGraphTensor acosh(MPSGraphTensor tensor,
String name)

atanh

public MPSGraphTensor atanh(MPSGraphTensor tensor,
String name)

not

public MPSGraphTensor not(MPSGraphTensor tensor,
String name)

isInfinite

public MPSGraphTensor isInfinite(MPSGraphTensor tensor,
String name)

isFinite

public MPSGraphTensor isFinite(MPSGraphTensor tensor,
String name)

isNaN

public MPSGraphTensor isNaN(MPSGraphTensor tensor,
String name)

erf

public MPSGraphTensor erf(MPSGraphTensor tensor,
String name)

addition

public MPSGraphTensor addition(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

subtraction

public MPSGraphTensor subtraction(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

multiplication

public MPSGraphTensor multiplication(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

division

public MPSGraphTensor division(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

modulo

public MPSGraphTensor modulo(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

power

public MPSGraphTensor power(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

minimum

public MPSGraphTensor minimum(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

maximum

public MPSGraphTensor maximum(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

minimumWithNaNPropagation

public MPSGraphTensor minimumWithNaNPropagation(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

Since:

Available in iOS 15.0 and later.

maximumWithNaNPropagation

public MPSGraphTensor maximumWithNaNPropagation(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

Since:

Available in iOS 15.0 and later.

equal

public MPSGraphTensor equal(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

notEqual

public MPSGraphTensor notEqual(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

lessThan

public MPSGraphTensor lessThan(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

lessThanOrEqual

public MPSGraphTensor lessThanOrEqual(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

greaterThan

public MPSGraphTensor greaterThan(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

greaterThanOrEqual

public MPSGraphTensor greaterThanOrEqual(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

logicalAND

public MPSGraphTensor logicalAND(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

logicalOR

public MPSGraphTensor logicalOR(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

logicalNAND

public MPSGraphTensor logicalNAND(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

logicalNOR

public MPSGraphTensor logicalNOR(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

logicalXOR

public MPSGraphTensor logicalXOR(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

logicalXNOR

public MPSGraphTensor logicalXNOR(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

atan2

public MPSGraphTensor atan2(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

select

public MPSGraphTensor select(MPSGraphTensor predicateTensor,
MPSGraphTensor truePredicateTensor,
MPSGraphTensor falseSelectTensor,
String name)

clamp

public MPSGraphTensor clamp(MPSGraphTensor tensor,
MPSGraphTensor minValueTensor,
MPSGraphTensor maxValueTensor,
String name)

divisionNoNaN

public MPSGraphTensor divisionNoNaN(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

floorModulo

public MPSGraphTensor floorModulo(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

convolution2D

public MPSGraphTensor convolution2D(MPSGraphTensor source,
MPSGraphTensor weights,
MPSGraphConvolution2DOpDescriptor descriptor,
String name)

convolution2DDataGradient

public MPSGraphTensor convolution2DDataGradient(MPSGraphTensor incomingGradient,
MPSGraphTensor weights,
NSArray<NSNumber> outputShape,
MPSGraphConvolution2DOpDescriptor forwardConvolutionDescriptor,
String name)

convolution2DDataGradient

public MPSGraphTensor convolution2DDataGradient(MPSGraphTensor gradient,
MPSGraphTensor weights,
MPSGraphTensor outputShapeTensor,
MPSGraphConvolution2DOpDescriptor forwardConvolutionDescriptor,
String name)

Since:

Available in iOS 15.0 and later.

convolution2DWeightsGradient

public MPSGraphTensor convolution2DWeightsGradient(MPSGraphTensor incomingGradient,
MPSGraphTensor source,
NSArray<NSNumber> outputShape,
MPSGraphConvolution2DOpDescriptor forwardConvolutionDescriptor,
String name)

convolution2DWeightsGradient

public MPSGraphTensor convolution2DWeightsGradient(MPSGraphTensor gradient,
MPSGraphTensor source,
MPSGraphTensor outputShapeTensor,
MPSGraphConvolution2DOpDescriptor forwardConvolutionDescriptor,
String name)

Since:

Available in iOS 15.0 and later.

convolutionTranspose2D

public MPSGraphTensor convolutionTranspose2D(MPSGraphTensor source,
MPSGraphTensor weights,
NSArray<NSNumber> outputShape,
MPSGraphConvolution2DOpDescriptor descriptor,
String name)

convolutionTranspose2D

public MPSGraphTensor convolutionTranspose2D(MPSGraphTensor source,
MPSGraphTensor weights,
MPSGraphTensor outputShape,
MPSGraphConvolution2DOpDescriptor descriptor,
String name)

Since:

Available in iOS 15.0 and later.

convolutionTranspose2DDataGradient

public MPSGraphTensor convolutionTranspose2DDataGradient(MPSGraphTensor incomingGradient,
MPSGraphTensor weights,
NSArray<NSNumber> outputShape,
MPSGraphConvolution2DOpDescriptor forwardConvolutionDescriptor,
String name)

convolutionTranspose2DDataGradient

public MPSGraphTensor convolutionTranspose2DDataGradient(MPSGraphTensor incomingGradient,
MPSGraphTensor weights,
MPSGraphTensor outputShape,
MPSGraphConvolution2DOpDescriptor forwardConvolutionDescriptor,
String name)

Since:

Available in iOS 15.0 and later.

convolutionTranspose2DWeightsGradient

public MPSGraphTensor convolutionTranspose2DWeightsGradient(MPSGraphTensor incomingGradientTensor,
MPSGraphTensor source,
NSArray<NSNumber> outputShape,
MPSGraphConvolution2DOpDescriptor forwardConvolutionDescriptor,
String name)

convolutionTranspose2DWeightsGradient

public MPSGraphTensor convolutionTranspose2DWeightsGradient(MPSGraphTensor incomingGradientTensor,
MPSGraphTensor source,
MPSGraphTensor outputShape,
MPSGraphConvolution2DOpDescriptor forwardConvolutionDescriptor,
String name)

Since:

Available in iOS 15.0 and later.

controlDependency

public NSArray<MPSGraphTensor> controlDependency(NSArray<MPSGraphOperation> operations,
Block0<NSArray<MPSGraphTensor>> dependentBlock,
String name)

doIf

public NSArray<MPSGraphTensor> doIf(MPSGraphTensor predicateTensor,
Block0<NSArray<MPSGraphTensor>> thenBlock,
Block0<NSArray<MPSGraphTensor>> elseBlock,
String name)

whileLoop

public NSArray<MPSGraphTensor> whileLoop(NSArray<MPSGraphTensor> initialInputs,
Block2<NSArray<MPSGraphTensor>,NSMutableArray<MPSGraphTensor>,MPSGraphTensor> before,
Block1<NSArray<MPSGraphTensor>,NSArray<MPSGraphTensor>> after,
String name)

forLoop

public NSArray<MPSGraphTensor> forLoop(MPSGraphTensor lowerBound,
MPSGraphTensor upperBound,
MPSGraphTensor step,
NSArray<MPSGraphTensor> initialBodyArguments,
Block2<MPSGraphTensor,NSArray<MPSGraphTensor>,NSArray<MPSGraphTensor>> body,
String name)

forLoop

public NSArray<MPSGraphTensor> forLoop(MPSGraphTensor numberOfIterations,
NSArray<MPSGraphTensor> initialBodyArguments,
Block2<MPSGraphTensor,NSArray<MPSGraphTensor>,NSArray<MPSGraphTensor>> body,
String name)

depthwiseConvolution2D

public MPSGraphTensor depthwiseConvolution2D(MPSGraphTensor source,
MPSGraphTensor weights,
MPSGraphDepthwiseConvolution2DOpDescriptor descriptor,
String name)

depthwiseConvolution2DDataGradient

public MPSGraphTensor depthwiseConvolution2DDataGradient(MPSGraphTensor incomingGradient,
MPSGraphTensor weights,
NSArray<NSNumber> outputShape,
MPSGraphDepthwiseConvolution2DOpDescriptor descriptor,
String name)

depthwiseConvolution2DWeightsGradient

public MPSGraphTensor depthwiseConvolution2DWeightsGradient(MPSGraphTensor incomingGradient,
MPSGraphTensor source,
NSArray<NSNumber> outputShape,
MPSGraphDepthwiseConvolution2DOpDescriptor descriptor,
String name)

depthwiseConvolution3D

public MPSGraphTensor depthwiseConvolution3D(MPSGraphTensor source,
MPSGraphTensor weights,
MPSGraphDepthwiseConvolution3DOpDescriptor descriptor,
String name)

Since:

Available in iOS 15.0 and later.

depthwiseConvolution3DDataGradient

public MPSGraphTensor depthwiseConvolution3DDataGradient(MPSGraphTensor incomingGradient,
MPSGraphTensor weights,
NSArray<NSNumber> outputShape,
MPSGraphDepthwiseConvolution3DOpDescriptor descriptor,
String name)

Since:

Available in iOS 15.0 and later.

depthwiseConvolution3DWeightsGradient

public MPSGraphTensor depthwiseConvolution3DWeightsGradient(MPSGraphTensor incomingGradient,
MPSGraphTensor source,
NSArray<NSNumber> outputShape,
MPSGraphDepthwiseConvolution3DOpDescriptor descriptor,
String name)

gatherND

public MPSGraphTensor gatherND(MPSGraphTensor updatesTensor,
MPSGraphTensor indicesTensor,
long batchDimensions,
String name)

gather

public MPSGraphTensor gather(MPSGraphTensor updatesTensor,
MPSGraphTensor indicesTensor,
long axis,
long batchDimensions,
String name)

gatherAlongAxis

public MPSGraphTensor gatherAlongAxis(long axis,
MPSGraphTensor updatesTensor,
MPSGraphTensor indicesTensor,
String name)

Since:

Available in iOS 15.4 and later.

gatherAlongAxisTensor

public MPSGraphTensor gatherAlongAxisTensor(MPSGraphTensor axisTensor,
MPSGraphTensor updatesTensor,
MPSGraphTensor indicesTensor,
String name)

Since:

Available in iOS 15.4 and later.

bandPart

public MPSGraphTensor bandPart(MPSGraphTensor inputTensor,
long numLower,
long numUpper,
String name)

bandPart

public MPSGraphTensor bandPart(MPSGraphTensor inputTensor,
MPSGraphTensor numLowerTensor,
MPSGraphTensor numUpperTensor,
String name)

softMaxCrossEntropy

public MPSGraphTensor softMaxCrossEntropy(MPSGraphTensor sourceTensor,
MPSGraphTensor labelsTensor,
long axis,
MPSGraphLossReductionType reductionType,
String name)

softMaxCrossEntropyGradient

public MPSGraphTensor softMaxCrossEntropyGradient(MPSGraphTensor gradientTensor,
MPSGraphTensor sourceTensor,
MPSGraphTensor labelsTensor,
long axis,
MPSGraphLossReductionType reductionType,
String name)

matrixMultiplication

public MPSGraphTensor matrixMultiplication(MPSGraphTensor primaryTensor,
MPSGraphTensor secondaryTensor,
String name)

sparseTensor

public MPSGraphTensor sparseTensor(MPSGraphSparseStorageType sparseStorageType,
NSArray<MPSGraphTensor> inputTensorArray,
NSArray<NSNumber> shape,
MPSDataType dataType,
String name)

Since:

Available in iOS 15.0 and later.

sparseTensor

public MPSGraphTensor sparseTensor(MPSGraphCreateSparseOpDescriptor sparseDescriptor,
NSArray<MPSGraphTensor> inputTensorArray,
NSArray<NSNumber> shape,
String name)

Since:

Available in iOS 15.0 and later.

placeholder

public MPSGraphTensor placeholder(NSArray<NSNumber> shape,
MPSDataType dataType,
String name)

placeholder

public MPSGraphTensor placeholder(NSArray<NSNumber> shape,
String name)

constant

public MPSGraphTensor constant(NSData data,
NSArray<NSNumber> shape,
MPSDataType dataType)

constant

public MPSGraphTensor constant(double scalar,
MPSDataType dataType)

constant

public MPSGraphTensor constant(double scalar,
NSArray<NSNumber> shape,
MPSDataType dataType)

variable

public MPSGraphTensor variable(NSData data,
NSArray<NSNumber> shape,
MPSDataType dataType,
String name)

readVariable

public MPSGraphTensor readVariable(MPSGraphTensor variable,
String name)

assignVariable

public MPSGraphOperation assignVariable(MPSGraphTensor variable,
MPSGraphTensor tensor,
String name)

mean

public MPSGraphTensor mean(MPSGraphTensor tensor,
NSArray<NSNumber> axes,
String name)

variance

public MPSGraphTensor variance(MPSGraphTensor tensor,
MPSGraphTensor meanTensor,
NSArray<NSNumber> axes,
String name)

variance

public MPSGraphTensor variance(MPSGraphTensor tensor,
NSArray<NSNumber> axes,
String name)

normalization

public MPSGraphTensor normalization(MPSGraphTensor tensor,
MPSGraphTensor mean,
MPSGraphTensor variance,
MPSGraphTensor gamma,
MPSGraphTensor beta,
float epsilon,
String name)

normalizationGammaGradient

public MPSGraphTensor normalizationGammaGradient(MPSGraphTensor incomingGradientTensor,
MPSGraphTensor sourceTensor,
MPSGraphTensor meanTensor,
MPSGraphTensor varianceTensor,
NSArray<NSNumber> axes,
float epsilon,
String name)

normalizationBetaGradient

public MPSGraphTensor normalizationBetaGradient(MPSGraphTensor incomingGradientTensor,
MPSGraphTensor sourceTensor,
NSArray<NSNumber> axes,
String name)

normalizationGradient

public MPSGraphTensor normalizationGradient(MPSGraphTensor incomingGradientTensor,
MPSGraphTensor sourceTensor,
MPSGraphTensor meanTensor,
MPSGraphTensor varianceTensor,
MPSGraphTensor gamma,
MPSGraphTensor gammaGradient,
MPSGraphTensor betaGradient,
NSArray<NSNumber> axes,
float epsilon,
String name)

oneHot

public MPSGraphTensor oneHot(MPSGraphTensor indicesTensor,
long depth,
long axis,
MPSDataType dataType,
double onValue,
double offValue,
String name)

oneHot

public MPSGraphTensor oneHot(MPSGraphTensor indicesTensor,
long depth,
MPSDataType dataType,
double onValue,
double offValue,
String name)

oneHot

public MPSGraphTensor oneHot(MPSGraphTensor indicesTensor,
long depth,
long axis,
MPSDataType dataType,
String name)

oneHot

public MPSGraphTensor oneHot(MPSGraphTensor indicesTensor,
long depth,
long axis,
String name)

oneHot

public MPSGraphTensor oneHot(MPSGraphTensor indicesTensor,
long depth,
MPSDataType dataType,
String name)

oneHot

public MPSGraphTensor oneHot(MPSGraphTensor indicesTensor,
long depth,
String name)

stochasticGradientDescent

public MPSGraphTensor stochasticGradientDescent(MPSGraphTensor learningRateTensor,
MPSGraphTensor valuesTensor,
MPSGraphTensor gradientTensor,
String name)

applyStochasticGradientDescent

public MPSGraphOperation applyStochasticGradientDescent(MPSGraphTensor learningRateTensor,
MPSGraphVariableOp variable,
MPSGraphTensor gradientTensor,
String name)

maxPooling2D

public MPSGraphTensor maxPooling2D(MPSGraphTensor source,
MPSGraphPooling2DOpDescriptor descriptor,
String name)

maxPooling2DReturnIndices

public NSArray<MPSGraphTensor> maxPooling2DReturnIndices(MPSGraphTensor source,
MPSGraphPooling2DOpDescriptor descriptor,
String name)

Since:

Available in iOS 15.3 and later.

maxPooling2DGradient

public MPSGraphTensor maxPooling2DGradient(MPSGraphTensor gradient,
MPSGraphTensor source,
MPSGraphPooling2DOpDescriptor descriptor,
String name)

avgPooling2D

public MPSGraphTensor avgPooling2D(MPSGraphTensor source,
MPSGraphPooling2DOpDescriptor descriptor,
String name)

avgPooling2D

public MPSGraphTensor avgPooling2D(MPSGraphTensor gradient,
MPSGraphTensor source,
MPSGraphPooling2DOpDescriptor descriptor,
String name)

maxPooling4D

public MPSGraphTensor maxPooling4D(MPSGraphTensor source,
MPSGraphPooling4DOpDescriptor descriptor,
String name)

Since:

Available in iOS 15.0 and later.

maxPooling4DReturnIndices

public NSArray<MPSGraphTensor> maxPooling4DReturnIndices(MPSGraphTensor source,
MPSGraphPooling4DOpDescriptor descriptor,
String name)

Since:

Available in iOS 15.3 and later.

maxPooling4DGradient

public MPSGraphTensor maxPooling4DGradient(MPSGraphTensor gradient,
MPSGraphTensor source,
MPSGraphPooling4DOpDescriptor descriptor,
String name)

Since:

Available in iOS 15.0 and later.

avgPooling4D

public MPSGraphTensor avgPooling4D(MPSGraphTensor source,
MPSGraphPooling4DOpDescriptor descriptor,
String name)

Since:

Available in iOS 15.0 and later.

avgPooling4DGradient

public MPSGraphTensor avgPooling4DGradient(MPSGraphTensor gradient,
MPSGraphTensor source,
MPSGraphPooling4DOpDescriptor descriptor,
String name)

Since:

Available in iOS 15.0 and later.

L2NormPooling4D

public MPSGraphTensor L2NormPooling4D(MPSGraphTensor source,
MPSGraphPooling4DOpDescriptor descriptor,
String name)

Since:

Available in iOS 15.0 and later.

L2NormPooling4DGradient

public MPSGraphTensor L2NormPooling4DGradient(MPSGraphTensor gradient,
MPSGraphTensor source,
MPSGraphPooling4DOpDescriptor descriptor,
String name)

Since:

Available in iOS 15.0 and later.

randomPhiloxStateTensor

public MPSGraphTensor randomPhiloxStateTensor(long seed,
String name)

randomPhiloxStateTensor

public MPSGraphTensor randomPhiloxStateTensor(long counterLow,
long counterHigh,
long key,
String name)

randomTensor

public MPSGraphTensor randomTensor(NSArray<NSNumber> shape,
MPSGraphRandomOpDescriptor descriptor,
String name)

randomTensor

public MPSGraphTensor randomTensor(MPSGraphTensor shapeTensor,
MPSGraphRandomOpDescriptor descriptor,
String name)

randomTensor

public MPSGraphTensor randomTensor(NSArray<NSNumber> shape,
MPSGraphRandomOpDescriptor descriptor,
long seed,
String name)

randomTensor

public MPSGraphTensor randomTensor(MPSGraphTensor shapeTensor,
MPSGraphRandomOpDescriptor descriptor,
long seed,
String name)

randomTensor

public NSArray<MPSGraphTensor> randomTensor(NSArray<NSNumber> shape,
MPSGraphRandomOpDescriptor descriptor,
MPSGraphTensor state,
String name)

randomTensor

public NSArray<MPSGraphTensor> randomTensor(MPSGraphTensor shapeTensor,
MPSGraphRandomOpDescriptor descriptor,
MPSGraphTensor state,
String name)

randomUniformTensor

public MPSGraphTensor randomUniformTensor(NSArray<NSNumber> shape,
String name)

randomUniformTensor

public MPSGraphTensor randomUniformTensor(MPSGraphTensor shapeTensor,
String name)

randomUniformTensor

public MPSGraphTensor randomUniformTensor(NSArray<NSNumber> shape,
long seed,
String name)

randomUniformTensor

public MPSGraphTensor randomUniformTensor(MPSGraphTensor shapeTensor,
long seed,
String name)

randomUniformTensor

public NSArray<MPSGraphTensor> randomUniformTensor(NSArray<NSNumber> shape,
MPSGraphTensor state,
String name)

randomUniformTensor

public NSArray<MPSGraphTensor> randomUniformTensor(MPSGraphTensor shapeTensor,
MPSGraphTensor state,
String name)

dropout

public MPSGraphTensor dropout(MPSGraphTensor tensor,
double rate,
String name)

dropout

public MPSGraphTensor dropout(MPSGraphTensor tensor,
MPSGraphTensor rate,
String name)

reductionSum

public MPSGraphTensor reductionSum(MPSGraphTensor tensor,
long axis,
String name)

reductionSum

public MPSGraphTensor reductionSum(MPSGraphTensor tensor,
NSArray<NSNumber> axes,
String name)

reductionMaximum

public MPSGraphTensor reductionMaximum(MPSGraphTensor tensor,
long axis,
String name)

reductionMaximum

public MPSGraphTensor reductionMaximum(MPSGraphTensor tensor,
NSArray<NSNumber> axes,
String name)

reductionMinimum

public MPSGraphTensor reductionMinimum(MPSGraphTensor tensor,
long axis,
String name)

reductionMinimum

public MPSGraphTensor reductionMinimum(MPSGraphTensor tensor,
NSArray<NSNumber> axes,
String name)

reductionMaximumPropagateNaN

public MPSGraphTensor reductionMaximumPropagateNaN(MPSGraphTensor tensor,
long axis,
String name)

reductionMaximumPropagateNaN

public MPSGraphTensor reductionMaximumPropagateNaN(MPSGraphTensor tensor,
NSArray<NSNumber> axes,
String name)

reductionMinimumPropagateNa

public MPSGraphTensor reductionMinimumPropagateNa(MPSGraphTensor tensor,
long axis,
String name)

reductionMinimumPropagateNaN

public MPSGraphTensor reductionMinimumPropagateNaN(MPSGraphTensor tensor,
NSArray<NSNumber> axes,
String name)

reductionProduct

public MPSGraphTensor reductionProduct(MPSGraphTensor tensor,
long axis,
String name)

reductionProduct

public MPSGraphTensor reductionProduct(MPSGraphTensor tensor,
NSArray<NSNumber> axes,
String name)

reductionArgMaximum

public MPSGraphTensor reductionArgMaximum(MPSGraphTensor tensor,
long axis,
String name)

Since:

Available in iOS 15.0 and later.

reductionArgMinimum

public MPSGraphTensor reductionArgMinimum(MPSGraphTensor tensor,
long axis,
String name)

Since:

Available in iOS 15.0 and later.

reductionAnd

public MPSGraphTensor reductionAnd(MPSGraphTensor tensor,
long axis,
String name)

Since:

Available in iOS 15.3 and later.

reductionAnd

public MPSGraphTensor reductionAnd(MPSGraphTensor tensor,
NSArray<NSNumber> axes,
String name)

Since:

Available in iOS 15.3 and later.

reductionOr

public MPSGraphTensor reductionOr(MPSGraphTensor tensor,
long axis,
String name)

Since:

Available in iOS 15.3 and later.

reductionOr

public MPSGraphTensor reductionOr(MPSGraphTensor tensor,
NSArray<NSNumber> axes,
String name)

Since:

Available in iOS 15.3 and later.

resizeTensor

public MPSGraphTensor resizeTensor(MPSGraphTensor imagesTensor,
NSArray<NSNumber> size,
MPSGraphResizeMode mode,
boolean centerResult,
boolean alignCorners,
MPSGraphTensorNamedDataLayout layout,
String name)

resizeTensor

public MPSGraphTensor resizeTensor(MPSGraphTensor imagesTensor,
MPSGraphTensor size,
MPSGraphResizeMode mode,
boolean centerResult,
boolean alignCorners,
MPSGraphTensorNamedDataLayout layout,
String name)

Since:

Available in iOS 15.0 and later.

resize

public MPSGraphTensor resize(MPSGraphTensor gradient,
MPSGraphTensor input,
MPSGraphResizeMode mode,
boolean centerResult,
boolean alignCorners,
MPSGraphTensorNamedDataLayout layout,
String name)

scatterND

public MPSGraphTensor scatterND(MPSGraphTensor updatesTensor,
MPSGraphTensor indicesTensor,
NSArray<NSNumber> shape,
long batchDimensions,
MPSGraphScatterMode mode,
String name)

scatterND

public MPSGraphTensor scatterND(MPSGraphTensor updatesTensor,
MPSGraphTensor indicesTensor,
NSArray<NSNumber> shape,
long batchDimensions,
String name)

scatterND

public MPSGraphTensor scatterND(MPSGraphTensor dataTensor,
MPSGraphTensor updatesTensor,
MPSGraphTensor indicesTensor,
long batchDimensions,
MPSGraphScatterMode mode,
String name)

Since:

Available in iOS 15.0 and later.

scatter

public MPSGraphTensor scatter(MPSGraphTensor updatesTensor,
MPSGraphTensor indicesTensor,
NSArray<NSNumber> shape,
long axis,
MPSGraphScatterMode mode,
String name)

Since:

Available in iOS 15.0 and later.

scatter

public MPSGraphTensor scatter(MPSGraphTensor dataTensor,
MPSGraphTensor updatesTensor,
MPSGraphTensor indicesTensor,
long axis,
MPSGraphScatterMode mode,
String name)

Since:

Available in iOS 15.0 and later.

scatterAlongAxis

public MPSGraphTensor scatterAlongAxis(long axis,
MPSGraphTensor updatesTensor,
MPSGraphTensor indicesTensor,
NSArray<NSNumber> shape,
MPSGraphScatterMode mode,
String name)

Since:

Available in iOS 15.4 and later.

scatterAlongAxisTensor

public MPSGraphTensor scatterAlongAxisTensor(MPSGraphTensor axisTensor,
MPSGraphTensor updatesTensor,
MPSGraphTensor indicesTensor,
NSArray<NSNumber> shape,
MPSGraphScatterMode mode,
String name)

Since:

Available in iOS 15.4 and later.

scatterAlongAxis

public MPSGraphTensor scatterAlongAxis(long axis,
MPSGraphTensor dataTensor,
MPSGraphTensor updatesTensor,
MPSGraphTensor indicesTensor,
MPSGraphScatterMode mode,
String name)

Since:

Available in iOS 15.4 and later.

scatterAlongAxisTensor

public MPSGraphTensor scatterAlongAxisTensor(MPSGraphTensor axisTensor,
MPSGraphTensor dataTensor,
MPSGraphTensor updatesTensor,
MPSGraphTensor indicesTensor,
MPSGraphScatterMode mode,
String name)

Since:

Available in iOS 15.4 and later.

stencil

public MPSGraphTensor stencil(MPSGraphTensor source,
MPSGraphTensor weights,
MPSGraphStencilOpDescriptor descriptor,
String name)

reshapeTensor

public MPSGraphTensor reshapeTensor(MPSGraphTensor tensor,
NSArray<NSNumber> shape,
String name)

reshapeTensor

public MPSGraphTensor reshapeTensor(MPSGraphTensor tensor,
MPSGraphTensor shapeTensor,
String name)

Since:

Available in iOS 15.0 and later.

transposeTensor

public MPSGraphTensor transposeTensor(MPSGraphTensor tensor,
long dimensionIndex,
long dimensionIndex2,
String name)

sliceTensor

public MPSGraphTensor sliceTensor(MPSGraphTensor tensor,
long dimensionIndex,
long start,
long length,
String name)

sliceTensor

public MPSGraphTensor sliceTensor(MPSGraphTensor tensor,
NSArray<NSNumber> starts,
NSArray<NSNumber> ends,
NSArray<NSNumber> strides,
String name)

sliceTensor

public MPSGraphTensor sliceTensor(MPSGraphTensor tensor,
NSArray<NSNumber> starts,
NSArray<NSNumber> ends,
NSArray<NSNumber> strides,
int startMask,
int endMask,
int squeezeMask,
String name)

sliceGradientTensor

public MPSGraphTensor sliceGradientTensor(MPSGraphTensor inputGradientTensor,
MPSGraphTensor fwdInShapeTensor,
NSArray<NSNumber> starts,
NSArray<NSNumber> ends,
NSArray<NSNumber> strides,
String name)

sliceGradientTensor

public MPSGraphTensor sliceGradientTensor(MPSGraphTensor inputGradientTensor,
MPSGraphTensor fwdInShapeTensor,
NSArray<NSNumber> starts,
NSArray<NSNumber> ends,
NSArray<NSNumber> strides,
int startMask,
int endMask,
int squeezeMask,
String name)

concatTensor

public MPSGraphTensor concatTensor(MPSGraphTensor tensor,
MPSGraphTensor tensor2,
long dimensionIndex,
String name)

concatTensors

public MPSGraphTensor concatTensors(NSArray<MPSGraphTensor> tensors,
long dimensionIndex,
String name)

concatTensors

public MPSGraphTensor concatTensors(NSArray<MPSGraphTensor> tensors,
long dimensionIndex,
boolean interleave,
String name)

tileTensor

public MPSGraphTensor tileTensor(MPSGraphTensor tensor,
NSArray<NSNumber> multiplier,
String name)

tileGradient

public MPSGraphTensor tileGradient(MPSGraphTensor incomingGradientTensor,
MPSGraphTensor sourceTensor,
NSArray<NSNumber> multiplier,
String name)

padTensor

public MPSGraphTensor padTensor(MPSGraphTensor tensor,
MPSGraphPaddingMode paddingMode,
NSArray<NSNumber> leftPadding,
NSArray<NSNumber> rightPadding,
double constantValue,
String name)

padGradient

public MPSGraphTensor padGradient(MPSGraphTensor incomingGradientTensor,
MPSGraphTensor sourceTensor,
MPSGraphPaddingMode paddingMode,
NSArray<NSNumber> leftPadding,
NSArray<NSNumber> rightPadding,
String name)

spaceToDepth2DTensor

public MPSGraphTensor spaceToDepth2DTensor(MPSGraphTensor tensor,
MPSGraphTensor widthAxisTensor,
MPSGraphTensor heightAxisTensor,
MPSGraphTensor depthAxisTensor,
long blockSize,
boolean usePixelShuffleOrder,
String name)

Since:

Available in iOS 15.0 and later.

depthToSpace2DTensor

public MPSGraphTensor depthToSpace2DTensor(MPSGraphTensor tensor,
MPSGraphTensor widthAxisTensor,
MPSGraphTensor heightAxisTensor,
MPSGraphTensor depthAxisTensor,
long blockSize,
boolean usePixelShuffleOrder,
String name)

Since:

Available in iOS 15.0 and later.

spaceToDepth2DTensor

public MPSGraphTensor spaceToDepth2DTensor(MPSGraphTensor tensor,
long widthAxis,
long heightAxis,
long depthAxis,
long blockSize,
boolean usePixelShuffleOrder,
String name)

Since:

Available in iOS 15.0 and later.

depthToSpace2DTensor

public MPSGraphTensor depthToSpace2DTensor(MPSGraphTensor tensor,
long widthAxis,
long heightAxis,
long depthAxis,
long blockSize,
boolean usePixelShuffleOrder,
String name)

Since:

Available in iOS 15.0 and later.

reverseTensor

public MPSGraphTensor reverseTensor(MPSGraphTensor tensor,
MPSGraphTensor axesTensor,
String name)

Since:

Available in iOS 15.0 and later.

reverseTensor

public MPSGraphTensor reverseTensor(MPSGraphTensor tensor,
NSArray<NSNumber> axes,
String name)

Since:

Available in iOS 15.0 and later.

reverseTensor

public MPSGraphTensor reverseTensor(MPSGraphTensor tensor,
String name)

Since:

Available in iOS 15.0 and later.

flatten2DTensor

public MPSGraphTensor flatten2DTensor(MPSGraphTensor tensor,
long axis,
String name)

Since:

Available in iOS 15.0 and later.

flatten2DTensor

public MPSGraphTensor flatten2DTensor(MPSGraphTensor tensor,
MPSGraphTensor axisTensor,
String name)

Since:

Available in iOS 15.0 and later.

broadcastTensor

public MPSGraphTensor broadcastTensor(MPSGraphTensor tensor,
NSArray<NSNumber> shape,
String name)

Since:

Available in iOS 15.0 and later.

broadcastTensor

public MPSGraphTensor broadcastTensor(MPSGraphTensor tensor,
MPSGraphTensor shapeTensor,
String name)

Since:

Available in iOS 15.0 and later.

shapeOfTensor

public MPSGraphTensor shapeOfTensor(MPSGraphTensor tensor,
String name)

Since:

Available in iOS 15.0 and later.

castTensor

public MPSGraphTensor castTensor(MPSGraphTensor tensor,
MPSDataType type,
String name)

Since:

Available in iOS 15.0 and later.

stackTensors

public MPSGraphTensor stackTensors(NSArray<MPSGraphTensor> inputTensors,
long axis,
String name)

Since:

Available in iOS 15.4 and later.

splitTensor

public NSArray<MPSGraphTensor> splitTensor(MPSGraphTensor tensor,
NSArray<NSNumber> splitSizes,
long axis,
String name)

Since:

Available in iOS 15.4 and later.

splitTensor

public NSArray<MPSGraphTensor> splitTensor(MPSGraphTensor tensor,
MPSGraphTensor splitSizesTensor,
long axis,
String name)

Since:

Available in iOS 15.4 and later.

splitTensor

public NSArray<MPSGraphTensor> splitTensor(MPSGraphTensor tensor,
long numSplits,
long axis,
String name)

Since:

Available in iOS 15.4 and later.

squeezeTensor

public MPSGraphTensor squeezeTensor(MPSGraphTensor tensor,
String name)

Since:

Available in iOS 15.4 and later.

squeezeTensor

public MPSGraphTensor squeezeTensor(MPSGraphTensor tensor,
long axis,
String name)

Since:

Available in iOS 15.4 and later.

squeezeTensor

public MPSGraphTensor squeezeTensor(MPSGraphTensor tensor,
NSArray<NSNumber> axes,
String name)

Since:

Available in iOS 15.4 and later.

squeezeTensor

public MPSGraphTensor squeezeTensor(MPSGraphTensor tensor,
MPSGraphTensor axesTensor,
String name)

Since:

Available in iOS 15.4 and later.

expandDimsOfTensor

public MPSGraphTensor expandDimsOfTensor(MPSGraphTensor tensor,
long axis,
String name)

Since:

Available in iOS 15.4 and later.

expandDimsOfTensor

public MPSGraphTensor expandDimsOfTensor(MPSGraphTensor tensor,
NSArray<NSNumber> axes,
String name)

Since:

Available in iOS 15.4 and later.

expandDimsOfTensor

public MPSGraphTensor expandDimsOfTensor(MPSGraphTensor tensor,
MPSGraphTensor axesTensor,
String name)

Since:

Available in iOS 15.4 and later.

coordinateAlongAxis

public MPSGraphTensor coordinateAlongAxis(long axis,
NSArray<NSNumber> shape,
String name)

Since:

Available in iOS 15.4 and later.

coordinateAlongAxisTensor

public MPSGraphTensor coordinateAlongAxisTensor(MPSGraphTensor axisTensor,
NSArray<NSNumber> shape,
String name)

Since:

Available in iOS 15.4 and later.

coordinateAlongAxis

public MPSGraphTensor coordinateAlongAxis(long axis,
MPSGraphTensor shapeTensor,
String name)

Since:

Available in iOS 15.4 and later.

coordinateAlongAxisTensor

public MPSGraphTensor coordinateAlongAxisTensor(MPSGraphTensor axisTensor,
MPSGraphTensor shapeTensor,
String name)

Since:

Available in iOS 15.4 and later.

topK

public NSArray<MPSGraphTensor> topK(MPSGraphTensor source,
long k,
String name)

topK

public NSArray<MPSGraphTensor> topK(MPSGraphTensor source,
MPSGraphTensor kTensor,
String name)

topK

public MPSGraphTensor topK(MPSGraphTensor gradient,
MPSGraphTensor source,
long k,
String name)

topK

public MPSGraphTensor topK(MPSGraphTensor gradient,
MPSGraphTensor source,
MPSGraphTensor kTensor,
String name)