lamp-onnx/lamp.onnx

lamp.onnx

package lamp.onnx

Type members

Classlikes

case class Converted(node: NodeProto, constants: Seq[TensorProto])
object DefaultOpSet extends DefaultOpSet1
trait DefaultOpSet1 extends OpSet
trait NameMap
trait OpSet
object Ops
case class VariableInfo(variable: Variable, name: String, input: Boolean, docString: String)

Value members

Concrete methods

def serialize(output: Variable, domain: String, modelDocString: String, opset: OpSet)(infoFun: PartialFunction[Variable, VariableInfo]): ModelProto
def serializeToFile(file: File, output: Variable, domain: String, modelDocString: String, opset: OpSet)(infoFun: PartialFunction[Variable, VariableInfo]): Unit