@tensorflow-models/coco-ssd
Version:
Object detection model (coco-ssd) in TensorFlow.js
58 lines • 2.74 kB
JavaScript
Object.defineProperty(exports, "__esModule", { value: true });
var arithmetic = require("./executors/arithmetic_executor");
var basicMath = require("./executors/basic_math_executor");
var control = require("./executors/control_executor");
var convolution = require("./executors/convolution_executor");
var creation = require("./executors/creation_executor");
var dynamic = require("./executors/dynamic_executor");
var evaluation = require("./executors/evaluation_executor");
var graph = require("./executors/graph_executor");
var image = require("./executors/image_executor");
var logical = require("./executors/logical_executor");
var matrices = require("./executors/matrices_executor");
var normalization = require("./executors/normalization_executor");
var reduction = require("./executors/reduction_executor");
var sliceJoin = require("./executors/slice_join_executor");
var spectral = require("./executors/spectral_executor");
var transformation = require("./executors/transformation_executor");
function executeOp(node, tensorMap, context) {
switch (node.category) {
case 'arithmetic':
return arithmetic.executeOp(node, tensorMap, context);
case 'basic_math':
return basicMath.executeOp(node, tensorMap, context);
case 'control':
return control.executeOp(node, tensorMap, context);
case 'convolution':
return convolution.executeOp(node, tensorMap, context);
case 'creation':
return creation.executeOp(node, tensorMap, context);
case 'dynamic':
return dynamic.executeOp(node, tensorMap, context);
case 'evaluation':
return evaluation.executeOp(node, tensorMap, context);
case 'image':
return image.executeOp(node, tensorMap, context);
case 'graph':
return graph.executeOp(node, tensorMap, context);
case 'logical':
return logical.executeOp(node, tensorMap, context);
case 'matrices':
return matrices.executeOp(node, tensorMap, context);
case 'normalization':
return normalization.executeOp(node, tensorMap, context);
case 'reduction':
return reduction.executeOp(node, tensorMap, context);
case 'slice_join':
return sliceJoin.executeOp(node, tensorMap, context);
case 'spectral':
return spectral.executeOp(node, tensorMap, context);
case 'transformation':
return transformation.executeOp(node, tensorMap, context);
default:
throw TypeError("Node type " + node.op + " is not implemented");
}
}
exports.executeOp = executeOp;
//# sourceMappingURL=operation_executor.js.map
;