BatchNormInfo
public struct BatchNormInfo
Corresponding to BatchNormInfo
in metal file
-
Undocumented
Declaration
Swift
var channelPosition: MetalShort
-
Undocumented
Declaration
Swift
var channels: MetalInt
-
Undocumented
Declaration
Swift
var inputWidth: MetalInt
-
Undocumented
Declaration
Swift
var inputHeight: MetalInt
-
Undocumented
Declaration
Swift
var epsilon: MetalFloat
-
Undocumented
Declaration
Swift
public static func makeBatchNormInfo(channelOrder: TensorChannelOrder, epsilon: Float, input: Tensor) -> BatchNormInfo