Backend half
This commit is contained in:
Generated
Vendored
+20
@@ -0,0 +1,20 @@
|
||||
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
|
||||
};
|
||||
Reference in New Issue
Block a user