CustomModel

public final class CustomModel : Model

Custom model.

  • Declaration

    Swift

    public enum IOType
  • Declaration

    Swift

    public init(
      inputs: [IOType], outputs: [IOType], hint: Hint, trainable: Bool? = nil, name: String = "",
      shapeInference: @convention(c) (
        _: ccv_nnc_cmd_t, _: UnsafePointer<ccv_nnc_tensor_param_t>?, _: Int32, _: ccv_nnc_hint_t,
        _: UnsafeMutablePointer<ccv_nnc_tensor_param_t>?, _: Int32
      ) -> Void,
      execute: @convention(c) (
        _: ccv_nnc_cmd_t, _: ccv_nnc_hint_t, _: Int32,
        _: UnsafePointer<UnsafeMutablePointer<ccv_nnc_tensor_t>?>?, _: Int32,
        _: UnsafePointer<UnsafeMutablePointer<ccv_nnc_tensor_t>?>?, _: Int32, _: OpaquePointer?
      ) -> Int32
    )