Function getOpsForConfig

tfjs/tools/custom_module/model_parser.ts:25–42

(
    // tslint:disable-next-line: no-any
    config: CustomTFJSBundleConfig, kernelToOps: any)

import {bail} from './util';

export function getOpsForConfig(
    // tslint:disable-next-line: no-any
    config: CustomTFJSBundleConfig, kernelToOps: any) {
  // This will return a list of ops used by the model.json(s) passed in.
  const results: Set<string> = new Set();
  let modelJson;
  for (const modelJsonPath of config.models) {
    try {
      modelJson = JSON.parse(fs.readFileSync(modelJsonPath, 'utf-8'));
    } catch (e) {
      bail(`Error loading JSON file ${modelJsonPath}`);
    }

    const ops = getOps(modelJson, kernelToOps);
    ops.forEach((op: string) => results.add(op));
  }
  return Array.from(results);
}

export function getOps(
    // tslint:disable-next-line: no-any