70 lines
2.4 KiB
TypeScript
70 lines
2.4 KiB
TypeScript
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
import { Graph } from './graph';
|
|
import { OperatorImplementation, OperatorInitialization } from './operators';
|
|
|
|
export interface OpSet {
|
|
domain: string;
|
|
version: number;
|
|
}
|
|
export declare namespace OpSet {
|
|
/**
|
|
* Domain of an opset, it can be an empty string(default value, represent for ai.onnx), or 'ai.onnx.ml'
|
|
*/
|
|
type Domain = '' | 'ai.onnx.ml' | 'com.microsoft';
|
|
/**
|
|
* A resolve rule consists of 4 or 5 items: opType, opSetDomain, versionSelector, operatorImplementation and
|
|
* operatorInitialization (optional)
|
|
*/
|
|
type ResolveRule =
|
|
| [string, Domain, string, OperatorImplementation<Graph.Node>]
|
|
| [string, Domain, string, OperatorImplementation<unknown>, OperatorInitialization<unknown>];
|
|
}
|
|
|
|
export function resolveOperator(node: Graph.Node, opsets: readonly OpSet[], rules: readonly OpSet.ResolveRule[]) {
|
|
for (const rule of rules) {
|
|
const opType = rule[0];
|
|
const domain = rule[1];
|
|
const versionSelector = rule[2];
|
|
const opImpl = rule[3];
|
|
const opInit = rule[4];
|
|
|
|
if (node.opType === opType) {
|
|
// operator type matches
|
|
for (const opset of opsets) {
|
|
// opset '' and 'ai.onnx' are considered the same.
|
|
if (opset.domain === domain || (opset.domain === 'ai.onnx' && domain === '')) {
|
|
// opset domain found
|
|
if (matchSelector(opset.version, versionSelector)) {
|
|
return { opImpl, opInit };
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
throw new TypeError(
|
|
`cannot resolve operator '${node.opType}' with opsets: ${opsets
|
|
.map((set) => `${set.domain || 'ai.onnx'} v${set.version}`)
|
|
.join(', ')}`,
|
|
);
|
|
}
|
|
|
|
function matchSelector(version: number, selector: string): boolean {
|
|
if (selector.endsWith('+')) {
|
|
// minimum version match ('7+' expects version>=7)
|
|
const rangeStart = Number.parseInt(selector.substring(0, selector.length - 1), 10);
|
|
return !isNaN(rangeStart) && rangeStart <= version;
|
|
} else if (selector.split('-').length === 2) {
|
|
// range match ('6-8' expects 6<=version<=8)
|
|
const pair = selector.split('-');
|
|
const rangeStart = Number.parseInt(pair[0], 10);
|
|
const rangeEnd = Number.parseInt(pair[1], 10);
|
|
return !isNaN(rangeStart) && !isNaN(rangeEnd) && rangeStart <= version && version <= rangeEnd;
|
|
} else {
|
|
// exact match ('7' expects version===7)
|
|
return Number.parseInt(selector, 10) === version;
|
|
}
|
|
}
|