Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/tensorflow/tfjs
/ functions
Functions
6,022 in github.com/tensorflow/tfjs
⨍
Functions
6,022
◇
Types & classes
1,932
↳
Endpoints
2
Function
_convert_v3_group
Construct a weights group entry. Args: group: The HDF5 group data, possibly nested. Returns: An array of weight groups (see `write_weigh
tfjs-converter/python/tensorflowjs/converters/keras_h5_conversion.py:90
Function
_create_saved_model_v1
Create a TensorFlow V1 SavedModel for testing. Args: save_dir: directory name of where the saved model will be stored.
e2e/integration_tests/convert_predict.py:132
Function
_create_saved_model_v1_with_hashtable
Test a TF V1 model with HashTable Ops. Args: save_dir: directory name of where the saved model will be stored.
e2e/integration_tests/convert_predict.py:376
Function
_create_saved_model_v2
Test a basic TF V2 model with functions to make sure functions are inlined. Args: save_dir: directory name of where the saved model will be sto
e2e/integration_tests/convert_predict.py:179
Function
_create_saved_model_v2_complex64
Test a TF V2 model with complex dtype. Args: save_dir: directory name of where the saved model will be stored.
e2e/integration_tests/convert_predict.py:287
Function
_create_saved_model_v2_with_control_flow
Test a basic TF v2 model with control flow to inlined. Args: save_dir: directory name of where the saved model will be stored.
e2e/integration_tests/convert_predict.py:201
Function
_create_saved_model_v2_with_control_flow_v2
Test a TF V2 model with control flow v2. Args: save_dir: directory name of where the saved model will be stored.
e2e/integration_tests/convert_predict.py:307
Function
_create_saved_model_v2_with_hashtable
Test a TF V2 model with HashTable Ops. Args: save_dir: directory name of where the saved model will be stored.
e2e/integration_tests/convert_predict.py:431
Function
_create_saved_model_v2_with_tensorlist_ops
Test a TF V2 model with TensorList Ops. Args: save_dir: directory name of where the saved model will be stored.
e2e/integration_tests/convert_predict.py:349
Function
_create_saved_model_with_conv2d
Test a basic model with fusable conv2d. Args: save_dir: directory name of where the saved model will be stored.
e2e/integration_tests/convert_predict.py:230
Function
_create_saved_model_with_prelu
Test a basic model with prelu activation. Args: save_dir: directory name of where the saved model will be stored.
e2e/integration_tests/convert_predict.py:259
Function
_find_contraction_with_activation
(node, node_map)
tfjs-converter/python/tensorflowjs/converters/fuse_depthwise_conv2d.py:29
Function
_find_contraction_with_bias_and_activation
(node, node_map)
tfjs-converter/python/tensorflowjs/converters/fuse_depthwise_conv2d.py:57
Function
_fusedMatMul
(args: { inputs: _FusedMatMulInputs, attrs: _FusedMatMulAttrs, backend: MathBackendCPU })
tfjs-backend-cpu/src/kernels/_FusedMatMul.ts:26
Function
_fusedMatMul
(args: { inputs: _FusedMatMulInputs, attrs: _FusedMatMulAttrs, backend: WebGPUBackend })
tfjs-backend-webgpu/src/kernels/_FusedMatMul.ts:23
Function
_fusedMatMul
(args: { inputs: _FusedMatMulInputs, attrs: _FusedMatMulAttrs, backend: MathBackendWebGL })
tfjs-backend-webgl/src/kernels/_FusedMatMul.ts:23
Method
_keyDense
* Should be used for testing purposes only.
tfjs-layers/src/layers/nlp/multihead_attention.ts:394
Function
_logcosh
(x: number)
tfjs-layers/src/losses_test.ts:141
Method
_outputDense
* Should be used for testing purposes only.
tfjs-layers/src/layers/nlp/multihead_attention.ts:408
Method
_queryDense
* Should be used for testing purposes only.
tfjs-layers/src/layers/nlp/multihead_attention.ts:387
Function
_strip_unused_nodes
(frozen_graph, concrete_func, output_node_names)
tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2.py:896
Function
_testSetTimeout
()
tfjs-core/src/platforms/platform_browser_test.ts:104
Function
_testSetTimeoutCustom
()
tfjs-core/src/platforms/platform_browser_test.ts:132
Method
_valueDense
* Should be used for testing purposes only.
tfjs-layers/src/layers/nlp/multihead_attention.ts:401
Function
a
()
tfjs-backend-webgpu/src/backend_webgpu_test.ts:662
Function
a
()
tfjs-core/src/tensor_util_test.ts:84
Function
a
()
tfjs-core/src/debug_mode_test.ts:37
Function
a
()
tfjs-backend-webgl/src/backend_webgl_test.ts:62
Function
abs
(args: {inputs: AbsInputs, backend: MathBackendWebGL})
tfjs-backend-webgl/src/kernels/Abs.ts:27
Function
absDiffLoss
(x: Tensor, y: Tensor)
tfjs-layers/src/engine/training_test.ts:716
Function
abs_
* Computes absolute value element-wise: `abs(x)` * * ```js * const x = tf.tensor1d([-1, 2, -3, 4]); * * x.abs().print(); // or tf.abs(x) * ```
tfjs-core/src/ops/abs.ts:39
Function
absoluteDifference_
* Computes the absolute difference loss between two tensors. * * @param labels The ground truth output tensor, same dimensions as * 'predictions
tfjs-core/src/ops/losses/absolute_difference.ts:44
Function
acos_
* Computes acos of the input `tf.Tensor` element-wise: `acos(x)` * * ```js * const x = tf.tensor1d([0, 1, -1, .7]); * * x.acos().print(); // or
tfjs-core/src/ops/acos.ts:37
Function
acosh_
* Computes the inverse hyperbolic cos of the input `tf.Tensor` element-wise: * `acosh(x)` * * ```js * const x = tf.tensor1d([10, 1, 3, 5.7]); *
tfjs-core/src/ops/acosh.ts:40
Method
acquireTexture
( shapeRC: [number, number], usage: TextureUsage, isPacked: boolean)
tfjs-backend-webgl/src/texture_manager.ts:36
Function
activation
(args: ActivationLayerArgs)
tfjs-layers/src/exports_layers.ts:482
Function
add
(config?: SymbolicTensor[]|Tensor[]|LayerArgs)
tfjs-layers/src/layers/merge.ts:327
Method
addFeed
* Add a Feed to the FeedDict. * @param feed The new `Feed` to add. * @returns This `FeedDict`.
tfjs-layers/src/engine/executor.ts:113
Function
addInboundNode
* Internal method to create an inbound node for the layer. * * @param inputTensors List of input tensors. * @param outputTensors List of outp
tfjs-layers/src/engine/topology.ts:1423
Function
addLoss
* Add losses to the layer. * * The loss may potentially be conditional on some inputs tensors, * for instance activity losses are conditional
tfjs-layers/src/engine/topology.ts:1329
Function
addN
(args: {inputs: AddNInputs, backend: MathBackendCPU})
tfjs-backend-cpu/src/kernels/AddN.ts:23
Function
addN
(args: {inputs: AddNInputs, backend: WebGPUBackend})
tfjs-backend-webgpu/src/kernels/AddN.ts:25
Function
addN
(args: {inputs: AddNInputs, backend: MathBackendWebGL})
tfjs-backend-webgl/src/kernels/AddN.ts:25
Function
addN_
* Adds a list of `tf.Tensor`s element-wise, each with the same shape and dtype. * * ```js * const a = tf.tensor1d([1, 2]); * const b = tf.tensor1d
tfjs-core/src/ops/add_n.ts:40
Method
addV2_conv2d
(x)
tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2_test.py:204
Function
addWeight
* Adds a weight variable to the layer. * * @param name Name of the new weight variable. * @param shape The shape of the weight. * @param d
tfjs-layers/src/engine/topology.ts:1269
Function
add_
* Adds two `tf.Tensor`s element-wise, A + B. Supports broadcasting. * * * ```js * const a = tf.tensor1d([1, 2, 3, 4]); * const b = tf.tensor1d([1
tfjs-core/src/ops/add.ts:50
Function
addn
(args: {inputs: TensorInfo[], backend: BackendWasm})
tfjs-backend-wasm/src/kernels/AddN.ts:37
Function
all
( args: {inputs: AllInputs, attrs: AllAttrs, backend: WebGPUBackend})
tfjs-backend-webgpu/src/kernels/All.ts:23
Function
all
( args: {inputs: AllInputs, backend: MathBackendWebGL, attrs: AllAttrs})
tfjs-backend-webgl/src/kernels/All.ts:26
Function
all
(args: {backend: BackendWasm, inputs: AllInputs, attrs: AllAttrs})
tfjs-backend-wasm/src/kernels/All.ts:30
Function
all_
* Computes the logical and of elements across dimensions of a `tf.Tensor`. * * Reduces the input along the dimensions given in `axes`. Unless `keepD
tfjs-core/src/ops/all.ts:57
Function
alphaDropout
(args: AlphaDropoutArgs)
tfjs-layers/src/exports_layers.ts:1675
Method
answer
(context: string, question: string)
tfjs-tflite/src/tflite_task_library_client/bert_qa.ts:59
Function
any
( args: {inputs: AnyInputs, attrs: AnyAttrs, backend: WebGPUBackend})
tfjs-backend-webgpu/src/kernels/Any.ts:23
Function
any
( args: {inputs: AnyInputs, backend: MathBackendWebGL, attrs: AnyAttrs})
tfjs-backend-webgl/src/kernels/Any.ts:25
Function
any
(args: {backend: BackendWasm, inputs: AnyInputs, attrs: AnyAttrs})
tfjs-backend-wasm/src/kernels/Any.ts:30
Function
any_
* Computes the logical or of elements across dimensions of a `tf.Tensor`. * * Reduces the input along the dimensions given in `axes`. Unless `keepDi
tfjs-core/src/ops/any.ts:57
Method
appendMetric
(outputIndex: number, metricName: string, metricTensor: LossOrMetricFn)
tfjs-layers/src/engine/training.ts:694
Method
appendToNGram
(str: Uint8Array)
tfjs-backend-cpu/src/kernels/StringNGrams_impl.ts:90
Function
apply
( inputs: Tensor|Tensor[]|SymbolicTensor|SymbolicTensor[], kwargs?: Kwargs)
tfjs-layers/src/engine/topology.ts:947
Method
apply
(w: Tensor)
tfjs-layers/src/constraints.ts:79
Method
apply
(w: Tensor)
tfjs-layers/src/constraints.ts:121
Method
apply
(w: Tensor)
tfjs-layers/src/constraints.ts:136
Method
apply
(w: Tensor)
tfjs-layers/src/constraints.ts:198
Method
apply
(shape: Shape, dtype?: DataType)
tfjs-layers/src/initializers.ts:56
Method
apply
(shape: Shape, dtype?: DataType)
tfjs-layers/src/initializers.ts:66
Method
apply
(shape: Shape, dtype?: DataType)
tfjs-layers/src/initializers.ts:93
Method
apply
(shape: Shape, dtype?: DataType)
tfjs-layers/src/initializers.ts:130
Method
apply
(shape: Shape, dtype?: DataType)
tfjs-layers/src/initializers.ts:165
Method
apply
(shape: Shape, dtype?: DataType)
tfjs-layers/src/initializers.ts:207
Method
apply
(shape: Shape, dtype?: DataType)
tfjs-layers/src/initializers.ts:238
Method
apply
(shape: Shape, dtype?: DataType)
tfjs-layers/src/initializers.ts:332
Method
apply
(shape: Shape, dtype?: DataType)
tfjs-layers/src/initializers.ts:542
Method
apply
* Porting note: Renamed from __call__. * @param x Variable of which to calculate the regularization score.
tfjs-layers/src/regularizers.ts:73
Method
apply
(x: Tensor)
tfjs-layers/src/activations.ts:62
Method
apply
(x: Tensor)
tfjs-layers/src/activations.ts:74
Method
apply
(x: Tensor)
tfjs-layers/src/activations.ts:86
Method
apply
(x: Tensor)
tfjs-layers/src/activations.ts:96
Method
apply
(x: Tensor)
tfjs-layers/src/activations.ts:108
Method
apply
(x: Tensor)
tfjs-layers/src/activations.ts:120
Method
apply
(x: Tensor)
tfjs-layers/src/activations.ts:132
Method
apply
(x: Tensor)
tfjs-layers/src/activations.ts:144
Method
apply
(x: Tensor)
tfjs-layers/src/activations.ts:156
Method
apply
* Calculate the activation function. * * @param x Tensor. * @param axis Integer, axis along which the softmax normalization is applied. *
tfjs-layers/src/activations.ts:180
Method
apply
* Calculate the activation function of log softmax: * log( exp(x_i) / sum(exp(x)) ) * * @param x Tensor. * @param axis Integer, axis along
tfjs-layers/src/activations.ts:205
Method
apply
* Calculate the activation function. * * @param x Tensor. * @returns a Tensor of the same shape as x
tfjs-layers/src/activations.ts:223
Method
apply
* Calculate the activation function. * * @param x Tensor. * @returns a Tensor of the same shape as x
tfjs-layers/src/activations.ts:249
Method
apply
* Calculate the activation function. * * @param x Tensor. * @returns a Tensor of the same shape as x
tfjs-layers/src/activations.ts:283
Method
apply
* Calculate the activation function. * * @param x Tensor. * @param alpha Scaling factor for the sigmoid function. * @returns a Tensor of t
tfjs-layers/src/activations.ts:302
Method
apply
( inputs: Tensor|Tensor[]|SymbolicTensor|SymbolicTensor[], kwargs?: Kwargs)
tfjs-layers/src/layers/wrappers.ts:354
Method
apply
(shape: Shape, dtype?: DataType)
tfjs-layers/src/layers/recurrent.ts:1719
Method
applyDropout
(x: tfc.Tensor, mask: tfc.Tensor[], index: number)
tfjs-layers/src/layers/convolutional_recurrent.ts:452
Method
applyGradients
(variableGradients: NamedVariableMap)
tfjs-core/src/serialization_test.ts:29
Method
applyGradients
(variableGradients: NamedVariableMap)
tfjs-core/src/serialization_test.ts:45
Method
applyGradients
(variableGradients: NamedVariableMap)
tfjs-core/src/serialization_test.ts:62
Method
applyGradients
(variableGradients: NamedVariableMap)
tfjs-core/src/serialization_test.ts:78
Method
applyGradients
(variableGradients: NamedVariableMap|NamedTensor[])
tfjs-core/src/optimizers/adam_optimizer.ts:64
Method
applyGradients
(variableGradients: NamedTensorMap|NamedTensor[])
tfjs-core/src/optimizers/rmsprop_optimizer.ts:63
Method
applyGradients
(variableGradients: NamedVariableMap|NamedTensor[])
tfjs-core/src/optimizers/adamax_optimizer.ts:64
← previous
next →
3,001–3,100 of 6,022, ranked by callers