get-operation-shape-names.js 595 Bytes
function getOperationShapeNames(model) {
    var operationShapeNames = [];
    var operations = model.operations;

    for (var operationName of Object.keys(operations)) {
        var operation = operations[operationName];
        if (operation.input && operation.input.shape) {
            operationShapeNames.push(operation.input.shape);
        }
        if (operation.output && operation.output.shape) {
            operationShapeNames.push(operation.output.shape);
        }
    }

    return operationShapeNames;
};

module.exports = {
    getOperationShapeNames: getOperationShapeNames
};