@tensorflow/tfjs-layers
Version:
TensorFlow layers API in JavaScript
41 lines (40 loc) • 1.22 kB
TypeScript
/**
* @license
* Copyright 2018 Google LLC
*
* Use of this source code is governed by an MIT-style
* license that can be found in the LICENSE file or at
* https://opensource.org/licenses/MIT.
* =============================================================================
*/
/// <amd-module name="@tensorflow/tfjs-layers/dist/logs" />
import { Scalar } from '@tensorflow/tfjs-core';
/**
* Logs in which values can be either numbers or Tensors (Scalars).
*
* Used internally.
*/
export type UnresolvedLogs = {
[key: string]: number | Scalar;
};
/**
* Turn any Scalar values in a Logs object into actual number values.
*
* @param logs The `Logs` object to be resolved in place.
*/
export declare function resolveScalarsInLogs(logs: UnresolvedLogs): Promise<void>;
/**
* Dispose all Tensors in an UnresolvedLogs object.
*
* @param logs An `UnresolvedLogs` object potentially containing `tf.Tensor`s in
* places where the values can be `tf.Tensor` or `number`.
*/
export declare function disposeTensorsInLogs(logs: UnresolvedLogs): void;
/**
* Logs in which values can only be numbers.
*
* Used when calling client-provided custom callbacks.
*/
export type Logs = {
[key: string]: number;
};