MCPcopy Create free account

hub / github.com/tensorflow/tfjs / functions

Functions6,022 in github.com/tensorflow/tfjs

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
Functiona
()
tfjs-backend-webgpu/src/backend_webgpu_test.ts:662
Functiona
()
tfjs-core/src/tensor_util_test.ts:84
Functiona
()
tfjs-core/src/debug_mode_test.ts:37
Functiona
()
tfjs-backend-webgl/src/backend_webgl_test.ts:62
Functionabs
(args: {inputs: AbsInputs, backend: MathBackendWebGL})
tfjs-backend-webgl/src/kernels/Abs.ts:27
FunctionabsDiffLoss
(x: Tensor, y: Tensor)
tfjs-layers/src/engine/training_test.ts:716
Functionabs_
* 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
FunctionabsoluteDifference_
* 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
Functionacos_
* 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
Functionacosh_
* 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
MethodacquireTexture
( shapeRC: [number, number], usage: TextureUsage, isPacked: boolean)
tfjs-backend-webgl/src/texture_manager.ts:36
Functionactivation
(args: ActivationLayerArgs)
tfjs-layers/src/exports_layers.ts:482
Functionadd
(config?: SymbolicTensor[]|Tensor[]|LayerArgs)
tfjs-layers/src/layers/merge.ts:327
MethodaddFeed
* Add a Feed to the FeedDict. * @param feed The new `Feed` to add. * @returns This `FeedDict`.
tfjs-layers/src/engine/executor.ts:113
FunctionaddInboundNode
* 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
FunctionaddLoss
* 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
FunctionaddN
(args: {inputs: AddNInputs, backend: MathBackendCPU})
tfjs-backend-cpu/src/kernels/AddN.ts:23
FunctionaddN
(args: {inputs: AddNInputs, backend: WebGPUBackend})
tfjs-backend-webgpu/src/kernels/AddN.ts:25
FunctionaddN
(args: {inputs: AddNInputs, backend: MathBackendWebGL})
tfjs-backend-webgl/src/kernels/AddN.ts:25
FunctionaddN_
* 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
MethodaddV2_conv2d
(x)
tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2_test.py:204
FunctionaddWeight
* 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
Functionadd_
* 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
Functionaddn
(args: {inputs: TensorInfo[], backend: BackendWasm})
tfjs-backend-wasm/src/kernels/AddN.ts:37
Functionall
( args: {inputs: AllInputs, attrs: AllAttrs, backend: WebGPUBackend})
tfjs-backend-webgpu/src/kernels/All.ts:23
Functionall
( args: {inputs: AllInputs, backend: MathBackendWebGL, attrs: AllAttrs})
tfjs-backend-webgl/src/kernels/All.ts:26
Functionall
(args: {backend: BackendWasm, inputs: AllInputs, attrs: AllAttrs})
tfjs-backend-wasm/src/kernels/All.ts:30
Functionall_
* 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
FunctionalphaDropout
(args: AlphaDropoutArgs)
tfjs-layers/src/exports_layers.ts:1675
Methodanswer
(context: string, question: string)
tfjs-tflite/src/tflite_task_library_client/bert_qa.ts:59
Functionany
( args: {inputs: AnyInputs, attrs: AnyAttrs, backend: WebGPUBackend})
tfjs-backend-webgpu/src/kernels/Any.ts:23
Functionany
( args: {inputs: AnyInputs, backend: MathBackendWebGL, attrs: AnyAttrs})
tfjs-backend-webgl/src/kernels/Any.ts:25
Functionany
(args: {backend: BackendWasm, inputs: AnyInputs, attrs: AnyAttrs})
tfjs-backend-wasm/src/kernels/Any.ts:30
Functionany_
* 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
MethodappendMetric
(outputIndex: number, metricName: string, metricTensor: LossOrMetricFn)
tfjs-layers/src/engine/training.ts:694
MethodappendToNGram
(str: Uint8Array)
tfjs-backend-cpu/src/kernels/StringNGrams_impl.ts:90
Functionapply
( inputs: Tensor|Tensor[]|SymbolicTensor|SymbolicTensor[], kwargs?: Kwargs)
tfjs-layers/src/engine/topology.ts:947
Methodapply
(w: Tensor)
tfjs-layers/src/constraints.ts:79
Methodapply
(w: Tensor)
tfjs-layers/src/constraints.ts:121
Methodapply
(w: Tensor)
tfjs-layers/src/constraints.ts:136
Methodapply
(w: Tensor)
tfjs-layers/src/constraints.ts:198
Methodapply
(shape: Shape, dtype?: DataType)
tfjs-layers/src/initializers.ts:56
Methodapply
(shape: Shape, dtype?: DataType)
tfjs-layers/src/initializers.ts:66
Methodapply
(shape: Shape, dtype?: DataType)
tfjs-layers/src/initializers.ts:93
Methodapply
(shape: Shape, dtype?: DataType)
tfjs-layers/src/initializers.ts:130
Methodapply
(shape: Shape, dtype?: DataType)
tfjs-layers/src/initializers.ts:165
Methodapply
(shape: Shape, dtype?: DataType)
tfjs-layers/src/initializers.ts:207
Methodapply
(shape: Shape, dtype?: DataType)
tfjs-layers/src/initializers.ts:238
Methodapply
(shape: Shape, dtype?: DataType)
tfjs-layers/src/initializers.ts:332
Methodapply
(shape: Shape, dtype?: DataType)
tfjs-layers/src/initializers.ts:542
Methodapply
* Porting note: Renamed from __call__. * @param x Variable of which to calculate the regularization score.
tfjs-layers/src/regularizers.ts:73
Methodapply
(x: Tensor)
tfjs-layers/src/activations.ts:62
Methodapply
(x: Tensor)
tfjs-layers/src/activations.ts:74
Methodapply
(x: Tensor)
tfjs-layers/src/activations.ts:86
Methodapply
(x: Tensor)
tfjs-layers/src/activations.ts:96
Methodapply
(x: Tensor)
tfjs-layers/src/activations.ts:108
Methodapply
(x: Tensor)
tfjs-layers/src/activations.ts:120
Methodapply
(x: Tensor)
tfjs-layers/src/activations.ts:132
Methodapply
(x: Tensor)
tfjs-layers/src/activations.ts:144
Methodapply
(x: Tensor)
tfjs-layers/src/activations.ts:156
Methodapply
* Calculate the activation function. * * @param x Tensor. * @param axis Integer, axis along which the softmax normalization is applied. *
tfjs-layers/src/activations.ts:180
Methodapply
* 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
Methodapply
* Calculate the activation function. * * @param x Tensor. * @returns a Tensor of the same shape as x
tfjs-layers/src/activations.ts:223
Methodapply
* Calculate the activation function. * * @param x Tensor. * @returns a Tensor of the same shape as x
tfjs-layers/src/activations.ts:249
Methodapply
* Calculate the activation function. * * @param x Tensor. * @returns a Tensor of the same shape as x
tfjs-layers/src/activations.ts:283
Methodapply
* 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
Methodapply
( inputs: Tensor|Tensor[]|SymbolicTensor|SymbolicTensor[], kwargs?: Kwargs)
tfjs-layers/src/layers/wrappers.ts:354
Methodapply
(shape: Shape, dtype?: DataType)
tfjs-layers/src/layers/recurrent.ts:1719
MethodapplyDropout
(x: tfc.Tensor, mask: tfc.Tensor[], index: number)
tfjs-layers/src/layers/convolutional_recurrent.ts:452
MethodapplyGradients
(variableGradients: NamedVariableMap)
tfjs-core/src/serialization_test.ts:29
MethodapplyGradients
(variableGradients: NamedVariableMap)
tfjs-core/src/serialization_test.ts:45
MethodapplyGradients
(variableGradients: NamedVariableMap)
tfjs-core/src/serialization_test.ts:62
MethodapplyGradients
(variableGradients: NamedVariableMap)
tfjs-core/src/serialization_test.ts:78
MethodapplyGradients
(variableGradients: NamedVariableMap|NamedTensor[])
tfjs-core/src/optimizers/adam_optimizer.ts:64
MethodapplyGradients
(variableGradients: NamedTensorMap|NamedTensor[])
tfjs-core/src/optimizers/rmsprop_optimizer.ts:63
MethodapplyGradients
(variableGradients: NamedVariableMap|NamedTensor[])
tfjs-core/src/optimizers/adamax_optimizer.ts:64
← previousnext →3,001–3,100 of 6,022, ranked by callers