opschema

package
v0.6.17 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Aug 22, 2026 License: MIT Imports: 0 Imported by: 0

Documentation

Overview

Package opschema holds the per-op input specifications MIL ops are declared with in coremltools.

Every op class in coremltools/converters/mil/mil/ops/defs declares an InputSpec (input_type.py:32-53): an ordered map from input name to an _InputType carrying a const and an optional flag (input_type.py:145-160) and, for tensors, a type domain (input_type.py:238-282). The spec is what coremltools checks a program against before it will serialize it. The table in table.gen.go is a dump of those specs for the ops we emit; see gen/dumpops.py.

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func Known 

func Known(opset string) bool

Known reports whether schemas are recorded for opset, named as MIL text names it ("ios15" and up). Callers skip per-op checking for unknown opsets rather than reporting an error, so a test asserts the opsets we emit are known.

func Registered 

func Registered(opType string) bool

Registered reports whether the table records opType for any opset. It is false both for an op that does not exist and for one the dump does not cover, so it cannot be used to reject an op outright.

Types

type DataType 

type DataType string

DataType is a MIL builtin type name, as types.builtin_to_string renders it.

type Kind 

type Kind uint8

Kind is the _InputType subclass an input was declared with. 

const (
	KindTensor Kind = iota
	KindList
	KindListOrTensorOrDict
	KindTuple
	KindState
	KindPyFunc
	KindInternal
)

type Op 

type Op struct {
	Type    string
	Params  []Param               // declaration order
	Domains map[string][]DataType // type_domains
}

Op is the input specification of one op at one opset.

func Lookup 

func Lookup(opset, opType string) (*Op, bool)

Lookup returns the schema for opType at opset. ok is false if either the opset or the op is unknown.

func (*Op) Param 

func (op *Op) Param(name string) (*Param, bool)

Param returns the declared input named name.

type Param 

type Param struct {
	Name     string
	Kind     Kind
	Const    bool // must be const at compile time
	Optional bool
	DomainID string     // type domain id, "" if the spec listed types directly
	Domain   []DataType // allowed types; empty iff DomainID != ""
}

Param is one declared input of an op.

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.