onnxruntime/js/web/lib/wasm/options-utils.ts
Yulong Wang 896f32ec09
[js/web] support string tensor for wasm backend (#7891)
* [js/web] support string tensor for wasm backend

* disable v9/test_cast_STRING_to_FLOAT: test data is wrong

* add non-string check

* Update session-handler.ts

* Update session-handler.ts
2021-06-03 00:44:50 -07:00

31 lines
1.1 KiB
TypeScript

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
interface ExtraOptionsHandler {
(name: string, value: string): void;
}
export const iterateExtraOptions =
(options: Record<string, unknown>, prefix: string, seen: WeakSet<Record<string, unknown>>,
handler: ExtraOptionsHandler): void => {
if (typeof options == 'object' && options !== null) {
if (seen.has(options)) {
throw new Error('Circular reference in options');
} else {
seen.add(options);
}
}
Object.entries(options).forEach(([key, value]) => {
const name = (prefix) ? prefix + key : key;
if (typeof value === 'object') {
iterateExtraOptions(value as Record<string, unknown>, name + '.', seen, handler);
} else if (typeof value === 'string' || typeof value === 'number') {
handler(name, value.toString());
} else if (typeof value === 'boolean') {
handler(name, (value) ? '1' : '0');
} else {
throw new Error(`Can't handle extra config type: ${typeof value}`);
}
});
};