@tensorflow-models/coco-ssd
Version:
Object detection model (coco-ssd) in TensorFlow.js
19 lines (18 loc) • 716 B
TypeScript
import { tensorflow } from '../data/compiled_api';
import { Graph } from './types';
export declare class OperationMapper {
private static _instance;
private opMappers;
static readonly Instance: OperationMapper;
private constructor();
private isControlFlow(node);
private isDynamicShape(node);
transformGraph(graph: tensorflow.IGraphDef): Graph;
private mapNode(node);
private getStringParam(attrs, name, def, keepCase?);
private getBoolParam(attrs, name, def);
private getNumberParam(attrs, name, def);
private getDtypeParam(attrs, name, def);
private getTensorShapeParam(attrs, name, def?);
private getNumericArrayParam(attrs, name, def);
}