MCPcopy Create free account

hub / github.com/tensorflow/tfjs / functions

Functions6,022 in github.com/tensorflow/tfjs

Functiondiag_
* Returns a diagonal tensor with given diagonal values. * * Given a diagonal, this operation returns a tensor with the diagonal and * everything el
tfjs-core/src/ops/diag.ts:49
Functiondilation2D
(args: { inputs: Dilation2DInputs, attrs: Dilation2DAttrs, backend: WebGPUBackend })
tfjs-backend-webgpu/src/kernels/Dilation2D.ts:23
Functiondilation2D
(args: { inputs: Dilation2DInputs, attrs: Dilation2DAttrs, backend: MathBackendWebGL })
tfjs-backend-webgl/src/kernels/Dilation2D.ts:24
Functiondilation2D
(args: { inputs: Dilation2DInputs, attrs: Dilation2DAttrs, backend: BackendWasm, })
tfjs-backend-wasm/src/kernels/Dilation2D.ts:53
Functiondilation2DBackpropFilter
(args: { inputs: Dilation2DBackpropFilterInputs, attrs: Dilation2DAttrs, backend: WebGPUBackend })
tfjs-backend-webgpu/src/kernels/Dilation2DBackpropFilter.ts:24
Functiondilation2DBackpropFilter
(args: { inputs: {x: Tensor4D, filter: Tensor3D, dy: Tensor4D}, attrs: Dilation2DAttrs, backend: Backend
tfjs-backend-wasm/src/kernels/Dilation2DBackpropFilter.ts:57
Functiondilation2DBackpropInput
(args: { inputs: Dilation2DBackpropInputInputs, attrs: Dilation2DAttrs, backend: WebGPUBackend })
tfjs-backend-webgpu/src/kernels/Dilation2DBackpropInput.ts:24
Functiondilation2DBackpropInput
(args: { inputs: {x: Tensor4D, filter: Tensor3D, dy: Tensor4D}, attrs: Dilation2DAttrs, backend: Backend
tfjs-backend-wasm/src/kernels/Dilation2DBackpropInput.ts:57
Functiondilation2d_
* Computes the grayscale dilation over the input `x`. * * @param x The input tensor, rank 3 or rank 4 of shape * `[batch, height, width, depth]
tfjs-core/src/ops/dilation2d.ts:61
FunctiondirToTarget
* Get the bazel target from a tfjs package's directory. * * @param dir The tfjs package's directory. * @returns The bazel target to build the tfjs
link-package/build_deps.ts:196
FunctiondisableDeprecationWarnings
()
tfjs-core/src/globals.ts:56
Functiondispose
* Attempt to dispose a LayersModel's weights. * * This method decrease the reference count of the LayersModel object by 1. * * A LayersMod
tfjs-layers/src/engine/container.ts:517
Functiondispose
* Attempt to dispose layer's weights. * * This method decreases the reference count of the Layer object by 1. * * A Layer is reference-cou
tfjs-layers/src/engine/topology.ts:1557
Functiondispose
* Dispose the variables (if any) owned by this optimizer instance.
tfjs-core/src/optimizers/optimizer.ts:127
Methoddispose
* Dispose this LayersVariable instance from memory.
tfjs-layers/src/variables.ts:112
Methoddispose
()
tfjs-layers/src/engine/input_layer.ts:126
Methoddispose
()
tfjs-tfdf/src/tfdf_model.ts:143
Methoddispose
()
tfjs-backend-webgpu/src/texture_manager.ts:106
Methoddispose
()
tfjs-backend-webgpu/src/buffer_manager.ts:98
Methoddispose
()
tfjs-core/src/jasmine_util.ts:289
Methoddispose
()
tfjs-core/src/tensor.ts:599
Methoddispose
()
tfjs-core/src/optimizers/adam_optimizer.ts:125
Methoddispose
()
tfjs-core/src/optimizers/rmsprop_optimizer.ts:147
Methoddispose
()
tfjs-core/src/optimizers/adamax_optimizer.ts:125
Methoddispose
()
tfjs-core/src/optimizers/adadelta_optimizer.ts:108
Methoddispose
()
tfjs-core/src/optimizers/momentum_optimizer.ts:90
Methoddispose
()
tfjs-core/src/optimizers/adagrad_optimizer.ts:89
Methoddispose
()
tfjs-node/src/nodejs_kernel_backend.ts:216
Methoddispose
()
tfjs-backend-webgl/src/texture_manager.ts:162
Methoddispose
()
tfjs-backend-webgl/src/backend_webgl.ts:1042
Methoddispose
* Dispose `ResourceManager`, including its hashTables and tensors in them.
tfjs-converter/src/executor/resource_manager.ts:62
Methoddispose
()
tfjs-converter/demo/control_flow/loop_model.js:32
MethoddisposeData
* Dispose the memory if the dataId has 0 refCount. Return true if the memory * is released or delayed in this backend, false if there are still
tfjs-backend-webgpu/src/backend_webgpu.ts:183
MethoddisposeData
(dataId: object, force?: boolean)
tfjs-core/src/backends/backend.ts:125
MethoddisposeData
* Dispose the memory if the dataId has 0 refCount. Return true if the memory * is released, false otherwise. * @param dataId * @oaram force O
tfjs-node/src/nodejs_kernel_backend.ts:240
MethoddisposeData
* Dispose the memory if the dataId has 0 refCount. Return true if the memory * is released, false otherwise. * @param dataId * @oaram force O
tfjs-backend-wasm/src/backend_wasm.ts:145
FunctiondisposeFunc
(backend: KernelBackend)
tfjs-core/src/kernel_registry_test.ts:173
FunctiondisposeVariables
()
tfjs-core/src/globals.ts:76
FunctiondisposeWeights
* Dispose the weight variables that this Layer instance holds. * * @returns {number} Number of disposed variables.
tfjs-layers/src/engine/topology.ts:1516
FunctiondistSquared
(a: FlatVector, b: FlatVector)
tfjs-core/src/util_base.ts:127
MethoddivNoNan
(a: Tensor, b: Tensor)
tfjs-node/src/nodejs_kernel_backend.ts:316
FunctiondivNoNan_
* Divides two `tf.Tensor`s element-wise, A / B. Supports broadcasting. Return 0 * if denominator is 0. * * * ```js * const a = tf.tensor1d([1, 4,
tfjs-core/src/ops/div_no_nan.ts:59
Functiondiv_
* Divides two `tf.Tensor`s element-wise, A / B. Supports broadcasting. * * ```js * const a = tf.tensor1d([1, 4, 9, 16]); * const b = tf.tensor1d([
tfjs-core/src/ops/div.ts:53
Methoddivide
(a: Tensor, b: Tensor)
tfjs-node/src/nodejs_kernel_backend.ts:310
MethoddoMovingAverage
(variable: LayerVariable, value: Tensor, momentum: number)
tfjs-layers/src/layers/normalization.ts:376
Functiondot
( a: Tensor, b: Tensor, activation?: tfc.fused.Activation, bias?: Tensor)
tfjs-layers/src/backend/tfjs_backend.ts:389
Functiondot_
* Computes the dot product of two matrices and/or vectors, `t1` and `t2`. * * ```js * const a = tf.tensor1d([1, 2]); * const b = tf.tensor2d([[1,
tfjs-core/src/ops/dot.ts:44
Functiondouble_module_fn
()
tfjs-converter/python/test_pip_package.py:135
FunctiondownloadAndUnpackResource
* Downloads and unpacks a given tarball or zip file at a given path. * @param {string} uri The path of the compressed file to download and extract.
tfjs-node/scripts/resources.js:39
FunctiondownloadByteEncodedFloatMatrixFromOutputTexture
( gl: WebGLRenderingContext, rows: number, columns: number, textureConfig: TextureConfig)
tfjs-backend-webgl/src/gpgpu_util.ts:298
FunctiondownloadFloat32MatrixFromBuffer
( gl: WebGLRenderingContext, buffer: WebGLBuffer, size: number)
tfjs-backend-webgl/src/gpgpu_util.ts:284
FunctiondownloadMatrixFromPackedOutputTexture
( gl: WebGLRenderingContext, physicalRows: number, physicalCols: number)
tfjs-backend-webgl/src/gpgpu_util.ts:336
FunctiondownloadPackedMatrixFromBuffer
( gl: WebGLRenderingContext, buffer: WebGLBuffer, batch: number, rows: number, cols: number, physicalR
tfjs-backend-webgl/src/gpgpu_util.ts:319
MethoddownloadPackedMatrixFromBuffer
( buffer: WebGLBuffer, batch: number, rows: number, columns: number, physicalRows: number, physica
tfjs-backend-webgl/src/gpgpu_context.ts:260
Functiondraw
( args: {inputs: DrawInputs, backend: MathBackendCPU, attrs: DrawAttrs})
tfjs-backend-cpu/src/kernels/Draw.ts:23
Functiondraw
( args: {inputs: DrawInputs, backend: WebGPUBackend, attrs: DrawAttrs})
tfjs-backend-webgpu/src/kernels/Draw.ts:24
Functiondraw
( image: Tensor2D|Tensor3D|TensorLike, canvas: HTMLCanvasElement, options?: DrawOptions)
tfjs-core/src/ops/browser.ts:416
Functiondropout
(args: DropoutLayerArgs)
tfjs-layers/src/exports_layers.ts:533
Functiondropout
( x: Tensor, level: number, noiseShape?: number[], seed?: number)
tfjs-layers/src/backend/tfjs_backend.ts:683
Functiondropout_
* Computes dropout. * * ```js * const x = tf.tensor1d([1, 2, 2, 1]); * const rate = 0.75; * const output = tf.dropout(x, rate); * output.print()
tfjs-core/src/ops/dropout.ts:53
FunctiondroppedInputs
()
tfjs-layers/src/layers/recurrent.ts:2114
MethoddroppedInputs
()
tfjs-layers/src/layers/noise.ts:180
Functiondump
* Dump the predict results of two backends and save diffs to files. * @param model The loaded model. * @param input The actual and expected results
e2e/benchmarks/local-benchmark/dump.js:184
FunctionearlyStopping
(args?: EarlyStoppingCallbackArgs)
tfjs-layers/src/callbacks.ts:251
Functioneinsum
( args: {inputs: EinsumInputs, backend: WebGPUBackend, attrs: EinsumAttrs})
tfjs-backend-webgpu/src/kernels/Einsum.ts:27
Functioneinsum
( args: {inputs: EinsumInputs, backend: MathBackendWebGL, attrs: EinsumAttrs})
tfjs-backend-webgl/src/kernels/Einsum.ts:27
Functioneinsum_
(equation: string, ...tensors: Tensor[])
tfjs-core/src/ops/einsum.ts:103
FunctioneluGrad
(args: {inputs: EluGradInputs, backend: MathBackendCPU})
tfjs-backend-cpu/src/kernels/EluGrad.ts:23
FunctioneluGrad
(args: {inputs: EluGradInputs, backend: WebGPUBackend})
tfjs-backend-webgpu/src/kernels/EluGrad.ts:25
FunctioneluGrad
(args: {inputs: EluGradInputs, backend: MathBackendWebGL})
tfjs-backend-webgl/src/kernels/EluGrad.ts:31
FunctioneluGrad
(args: {inputs: EluGradInputs, backend: BackendWasm})
tfjs-backend-wasm/src/kernels/EluGrad.ts:32
Functionelu_
* Computes exponential linear element-wise: `x > 0 ? x : (e ^ x) - 1`. * * ```js * const x = tf.tensor1d([-1, 1, -3, 2]); * * x.elu().print(); /
tfjs-core/src/ops/elu.ts:39
Functionembedding
(args: EmbeddingLayerArgs)
tfjs-layers/src/exports_layers.ts:684
FunctionenableDebugMode
()
tfjs-core/src/globals.ts:51
FunctionenableProdMode
()
tfjs-core/src/globals.ts:33
Methodencode
* Encode the provided string into an array of bytes using the provided * encoding.
tfjs-react-native/src/platform_react_native.ts:147
Methodencode
(text: string, encoding: string)
tfjs-core/src/platforms/platform_node.ts:69
Methodencode
(text: string, encoding: string)
tfjs-core/src/platforms/platform_browser.ts:47
FunctionencodeCategoricalInputs
(inputs: Tensor|Tensor[], outputMode: OutputMode,
tfjs-layers/src/layers/preprocessing/preprocessing_utils.ts:19
FunctionencodeJpeg
( image: Tensor3D, format: ''|'grayscale'|'rgb' = '', quality = 95, progressive = false, optimizeSize
tfjs-node/src/image.ts:220
FunctionencodePng
( image: Tensor3D, compression = 1)
tfjs-node/src/image.ts:244
FunctionencodeWeights
( tensors: NamedTensorMap|NamedTensor[], group?: WeightGroup)
tfjs-core/src/io/io_utils.ts:52
MethodendTokenId
()
tfjs-layers/src/layers/nlp/models/gpt2/gpt2_tokenizer.ts:96
MethodensureLength
( input: Tensor, length: number, padValue?: string|number)
tfjs-layers/src/layers/nlp/preprocessing/start_end_packer.ts:148
FunctionensureShape_
* Checks the input tensor mathes the given shape. * * Given an input tensor, returns a new tensor with the same values as the * input tensor with s
tfjs-core/src/ops/ensure_shape.ts:50
Functionenter
(node: NodeType)
tfjs-backend-wasm/scripts/patch_wechat_webassembly.ts:43
Methodepsilon
Returns the smallest representable number.
tfjs-backend-cpu/src/backend_cpu.ts:230
Methodepsilon
()
tfjs-node/src/nodejs_kernel_backend.ts:166
Methodepsilon
Returns the smallest representable number.
tfjs-backend-webgl/src/backend_webgl.ts:1092
Functionequal_
* Returns the truth value of (a == b) element-wise. Supports broadcasting. * * ```js * const a = tf.tensor1d([1, 2, 3]); * const b = tf.tensor1d([
tfjs-core/src/ops/equal.ts:43
Functionerf_
* Computes Gauss error function of the input `tf.Tensor` element-wise: * `erf(x)` * * ```js * const x = tf.tensor1d([0, .1, -.1, .7]); * * x.erf
tfjs-core/src/ops/erf.ts:42
FunctioneuclideanNorm_
* Computes the Euclidean norm of scalar, vectors, and matrices. * * ```js * const x = tf.tensor1d([1, 2, 3, 4]); * * x.euclideanNorm().print();
tfjs-core/src/ops/euclidean_norm.ts:47
Methodevaluate
( x: Tensor|Tensor[], y: Tensor|Tensor[], args?: ModelEvaluateArgs )
tfjs-layers/src/layers/nlp/utils.ts:154
Methodexecute
* Execute the inference for the input tensors and return activation * values for specified output node names without batching. * * @param inp
tfjs-tflite/src/tflite_model.ts:176
Methodexecute
* Execute internal tensors of the model with input data feed. * @param inputs Input data feed. Must match the inputs of the model. * @param outp
tfjs-layers/src/engine/training.ts:936
Methodexecute
* Execute the inference for the input tensors and return activation * values for specified output node names without batching. * * @param inp
tfjs-tfdf/src/tfdf_model.ts:116
Methodexecute
* Execute the inference for the input tensors and return activation * values for specified output node names without batching. * * @param inp
tfjs-node/src/saved_model.ts:364
Methodexecute
* Executes the inference for given input tensors. * @param inputs Tensor map for the model inputs, keyed by the input node * names. * @param
tfjs-converter/src/executor/graph_executor.ts:231
MethodexecuteAsync
* Executes the inference for given input tensors in Async fashion. * @param inputs Tensor map for the model inputs, keyed by the input node * na
tfjs-converter/src/executor/graph_executor.ts:420
MethodexecuteFunctionAsync
( inputs: Tensor[], tensorArrayMap: TensorArrayMap, tensorListMap: TensorListMap)
tfjs-converter/src/executor/graph_executor.ts:517
← previousnext →4,101–4,200 of 6,022, ranked by callers