-
Undocumented
Declaration
Swift
public typealias Device = CPU
-
Declaration
Swift
public static func fill<N>(value: N, result: Buffer<N, Device>, count: Int) where N : NumericType
-
Declaration
Swift
public static func vNeg<N>(val: Buffer<N, Device>, result: Buffer<N, Device>, count: Int) where N : NumericType
-
Declaration
Swift
public static func argmax<N>(values: Buffer<N, Device>, count: Int) -> (Int, N) where N : NumericType
-
Declaration
Swift
@_specialize(exported: false, kind: full, where N == Float) public static func fillDiagonal<N>(values: ShapedBuffer<N, CPU>, target: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
@_specialize(exported: false, kind: full, where N == Float) public static func fillDiagonal<N>(value: N, target: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
@_specialize(exported: false, kind: full, where N == Float) public static func extractDiagonal<N>(values: ShapedBuffer<N, CPU>, target: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
@_specialize(exported: false, kind: full, where N == Float) public static func gemm<N>(lhs: ShapedBuffer<N, CPU>, rhs: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>, alpha: N, beta: N, transposeFirst: Bool, transposeSecond: Bool) where N : NumericType
-
Declaration
Swift
@_specialize(exported: false, kind: full, where N == Float) public static func broadcastAdd<N>(lhs: ShapedBuffer<N, CPU>, rhs: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
@_specialize(exported: false, kind: full, where N == Float) public static func broadcastSub<N>(lhs: ShapedBuffer<N, CPU>, rhs: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
@_specialize(exported: false, kind: full, where N == Float) public static func broadcastMul<N>(lhs: ShapedBuffer<N, CPU>, rhs: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
@_specialize(exported: false, kind: full, where N == Float) public static func broadcastDiv<N>(lhs: ShapedBuffer<N, CPU>, rhs: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
@_specialize(exported: false, kind: full, where N == Float) public static func reduceSum<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>, axis: Int) where N : NumericType
-
Declaration
Swift
@_specialize(exported: false, kind: full, where N == Float) public static func reduceMax<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>, context: ShapedBuffer<Int32, CPU>?, axis: Int) where N : NumericType
-
Declaration
Swift
@_specialize(exported: false, kind: full, where N == Float) public static func reduceMin<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>, context: ShapedBuffer<Int32, CPU>?, axis: Int) where N : NumericType
-
Declaration
Swift
@_specialize(exported: false, kind: full, where N == Float) public static func reduceMean<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>, axis: Int) where N : NumericType
-
Declaration
Swift
@_specialize(exported: false, kind: full, where N == Float) public static func reduceSum<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>, axes: [Int]) where N : NumericType
-
Declaration
Swift
@_specialize(exported: false, kind: full, where N == Float) public static func reduceMean<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>, axes: [Int]) where N : NumericType
-
Declaration
Swift
@_specialize(exported: false, kind: full, where N == Float) public static func reduceMax<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>, context: ShapedBuffer<Int32, CPU>?, axes: [Int]) where N : NumericType
-
Declaration
Swift
@_specialize(exported: false, kind: full, where N == Float) public static func reduceMin<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>, context: ShapedBuffer<Int32, CPU>?, axes: [Int]) where N : NumericType
-
Declaration
Swift
public static func scatter<N>(reduced: ShapedBuffer<N, CPU>, context: ShapedBuffer<Int32, CPU>, result: ShapedBuffer<N, CPU>, axis: Int, ignoreIndex: Int32) where N : NumericType
-
Declaration
Swift
public static func gather<N>(expanded: ShapedBuffer<N, CPU>, context: ShapedBuffer<Int32, CPU>, result: ShapedBuffer<N, CPU>, axis: Int, ignoreIndex: Int32) where N : NumericType
-
Declaration
Swift
@_specialize(exported: false, kind: full, where N == Float) public static func sum<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
@_specialize(exported: false, kind: full, where N == Float) public static func mean<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
@discardableResult @_specialize(exported: false, kind: full, where N == Float) public static func max<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>) -> Int where N : NumericType
-
Declaration
Swift
@discardableResult @_specialize(exported: false, kind: full, where N == Float) public static func min<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>) -> Int where N : NumericType
-
Declaration
Swift
public static func exp<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
public static func log<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
public static func sqrt<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
public static func relu<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
public static func heaviside<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
public static func sin<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
public static func cos<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
public static func tan<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
public static func sinh<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
public static func cosh<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
public static func tanh<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
public static func max<N>(_ lhs: ShapedBuffer<N, CPU>, _ rhs: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
public static func max<N>(_ lhs: ShapedBuffer<N, CPU>, _ rhs: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>, context: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
public static func min<N>(_ lhs: ShapedBuffer<N, CPU>, _ rhs: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
public static func min<N>(_ lhs: ShapedBuffer<N, CPU>, _ rhs: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>, context: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
public static func permuteAxes<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>, arangement: [Int]) where N : NumericType
-
Declaration
Swift
public static func permuteAxesAdd<N>(values: ShapedBuffer<N, CPU>, add: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>, arangement: [Int]) where N : NumericType
-
Declaration
Swift
public static func arange<N>(lowerBound: N, upperBound: N, result: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
public static func subscriptRead<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>, index: [Int?])
-
Declaration
Swift
public static func subscriptWrite<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>, index: [Int?])
-
Declaration
Swift
public static func subscriptReadAdd<N>(values: ShapedBuffer<N, CPU>, add: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>, index: [Int?]) where N : NumericType
-
Declaration
Swift
public static func subscriptWriteAdd<N>(values: ShapedBuffer<N, CPU>, add: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>, index: [Int?]) where N : NumericType
-
Declaration
Swift
public static func stack<N>(buffers: [ShapedBuffer<N, CPU>], result: ShapedBuffer<N, CPU>, axis: Int)
-
Declaration
Swift
public static func unstackAdd<N>(stacked: ShapedBuffer<N, CPU>, add: [ShapedBuffer<N, CPU>], result: [ShapedBuffer<N, CPU>], axis: Int) where N : NumericType
-
Declaration
Swift
public static func unstack<N>(stacked: ShapedBuffer<N, CPU>, result: [ShapedBuffer<N, CPU>], axis: Int) where N : NumericType
-
Declaration
Swift
public static func reverse<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>)
-
Declaration
Swift
public static func reverseAdd<N>(values: ShapedBuffer<N, CPU>, add: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>) where N : NumericType
-
Declaration
Swift
@_specialize(exported: false, kind: full, where N == Float) public static func img2col<N>(values: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>, kernelWidth: Int, kernelHeight: Int, padding: Int, stride: Int) where N : NumericType
-
Declaration
Swift
@_specialize(exported: false, kind: full, where N == Float) public static func col2img<N>(matrix: ShapedBuffer<N, CPU>, image: ShapedBuffer<N, CPU>, kernelWidth: Int, kernelHeight: Int, padding: Int, stride: Int) where N : NumericType
-
Declaration
Swift
public static func band<N>(buffer: ShapedBuffer<N, CPU>, result: ShapedBuffer<N, CPU>, belowDiagonal: Int?, aboveDiagonal: Int?) where N : NumericType