UNPKG

@tensorflow/tfjs-node

Version:

This repository provides native TensorFlow execution in backend JavaScript applications under the Node.js runtime, accelerated by the TensorFlow C binary under the hood. It provides the same API as [TensorFlow.js](https://js.tensorflow.org/api/latest/).

39 lines (38 loc) 1.48 kB
"use strict"; /** * @license * Copyright 2020 Google LLC. All Rights Reserved. * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. * ============================================================================= */ Object.defineProperty(exports, "__esModule", { value: true }); exports.realConfig = void 0; var tfjs_1 = require("@tensorflow/tfjs"); var nodejs_kernel_backend_1 = require("../nodejs_kernel_backend"); exports.realConfig = { kernelName: tfjs_1.Real, backendName: 'tensorflow', kernelFunc: function (args) { var input = args.inputs.input; var backend = args.backend; var opAttrs = [ (0, nodejs_kernel_backend_1.createTensorsTypeOpAttr)('T', input), { name: 'Tout', type: backend.binding.TF_ATTR_TYPE, value: backend.binding.TF_FLOAT } ]; var inputs = [input]; return backend.executeSingleOutput(tfjs_1.Real, opAttrs, inputs); } };