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
rnn
(args: RNNLayerArgs)
tfjs-layers/src/exports_layers.ts:1509
Function
rnnStepForTest
* A simplistic RNN step function for testing. * This step function simply * - calculates a reduced mean over all input elements, for each sample. *
tfjs-layers/src/layers/recurrent_test.ts:42
Function
rotateWithOffset
(args: { inputs: RotateWithOffsetInputs, backend: BackendWasm, attrs: RotateWithOffsetAttrs })
tfjs-backend-wasm/src/kernels/RotateWithOffset.ts:44
Function
rotateWithOffset_
* Rotates the input image tensor counter-clockwise with an optional offset * center of rotation. Currently available in the CPU, WebGL, and WASM back
tfjs-core/src/ops/image/rotate_with_offset.ts:45
Function
round_
* Computes round of input `tf.Tensor` element-wise: `round(x)`. * It implements banker's rounding. * * ```js * const x = tf.tensor1d([.6, 1.1, -3.
tfjs-core/src/ops/round.ts:40
Function
rowMajorBatch
* Groups elements into batches, represented as arrays of elements. * * We can think of the elements of this iterator as 'rows' (even if they are
tfjs-data/src/iterators/lazy_iterator.ts:351
Function
rsqrt_
* Computes reciprocal of square root of the input `tf.Tensor` element-wise: * `y = 1 / sqrt(x)` * * ```js * const x = tf.tensor1d([1, 2, 4, -1]);
tfjs-core/src/ops/rsqrt.ts:40
Function
runBrowserStackBenchmark
* Run benchmark for singular browser-device combination. * * This function utilizes a promise that is fulfilled once the corresponding * result is
e2e/benchmarks/browserstack-benchmark/app.js:278
Function
runInternalGraph
* Computes output tensors for new inputs. * * Note: * - Expects `inputs` to be a list (potentially with 1 element). * * @param inputs
tfjs-layers/src/engine/container.ts:871
Function
runProgram
( gpgpu: GPGPUContext, binary: GPGPUBinary, inputs: TensorData[], output: TensorData, customUniformVal
tfjs-backend-webgl/src/gpgpu_math.ts:239
Method
runSavedModel
( id: number, inputs: Tensor[], inputTensorInfos: ModelTensorInfo[], outputOpNames: string[])
tfjs-node/src/nodejs_kernel_backend.ts:484
Method
save
* Save model artifacts. This IO handler cannot support writing to the * packaged bundle at runtime and is exclusively for loading a model * that
tfjs-react-native/src/bundle_resource_io.ts:39
Method
save
(modelArtifacts: ModelArtifacts)
tfjs-core/src/io/indexed_db.ts:100
Method
save
(modelArtifacts: ModelArtifacts)
tfjs-core/src/io/browser_files.ts:68
Method
save
* Save model artifacts to browser local storage. * * See the documentation to `browserLocalStorage` for details on the saved * artifacts.
tfjs-core/src/io/local_storage.ts:166
Method
saveFunc
(tensors)
tfjs-core/src/engine.ts:678
Function
saveHandler
(artifacts: tf.io.ModelArtifacts)
tfjs-core/src/io/passthrough_test.ts:83
Function
saveIterations
()
tfjs-core/src/optimizers/optimizer.ts:133
Function
saveModelRunner
()
tfjs-react-native/integration_rn59/components/ml.ts:150
Function
save_keras_model
r"""Save a Keras model and its weights in TensorFlow.js format. Args: model: An instance of `keras.Model`. artifacts_dir: The directory in
tfjs-converter/python/tensorflowjs/converters/keras_h5_conversion.py:431
Function
scale_after_normalization
(node)
tfjs-converter/python/tensorflowjs/converters/graph_rewrite_util.py:67
Function
scatterND_
* Creates a new tensor by applying sparse updates to individual * values or slices within a zero tensor of the given shape tensor according to * ind
tfjs-core/src/ops/scatter_nd.ts:49
Function
scatterNd
(args: { inputs: ScatterNdInputs, backend: MathBackendCPU, attrs: ScatterNdAttrs })
tfjs-backend-cpu/src/kernels/ScatterNd.ts:23
Function
scatterNd
(args: { inputs: ScatterNdInputs, backend: WebGPUBackend, attrs: ScatterNdAttrs })
tfjs-backend-webgpu/src/kernels/ScatterNd.ts:26
Function
scatterNd
(args: { inputs: ScatterNdInputs, backend: MathBackendWebGL, attrs: ScatterNdAttrs })
tfjs-backend-webgl/src/kernels/ScatterNd.ts:25
Function
scatterNd
( args: {backend: BackendWasm, inputs: ScatterNdInputs, attrs: ScatterNdAttrs})
tfjs-backend-wasm/src/kernels/ScatterNd.ts:43
Function
searchSorted
(args: { inputs: SearchSortedInputs, backend: WebGPUBackend, attrs: SearchSortedAttrs })
tfjs-backend-webgpu/src/kernels/SearchSorted.ts:23
Function
searchSorted
(args: { inputs: SearchSortedInputs, backend: MathBackendWebGL, attrs: SearchSortedAttrs })
tfjs-backend-webgl/src/kernels/SearchSorted.ts:23
Function
searchSorted
(args: { inputs: SearchSortedInputs, backend: BackendWasm, attrs: SearchSortedAttrs, })
tfjs-backend-wasm/src/kernels/SearchSorted.ts:41
Function
searchSorted_
* Searches for where a value would go in a sorted sequence. * * This is not a method for checking containment (like javascript in). * * The typica
tfjs-core/src/ops/search_sorted.ts:80
Method
second
()
tfjs-data/src/util/deep_map_test.ts:122
Function
segOpComputeOptimalWindowSize
( inSize: number, numSegments: number)
tfjs-core/src/ops/segment_util.ts:30
Method
segment
(input: ImageData|HTMLImageElement|HTMLCanvasElement| HTMLVideoElement)
tfjs-tflite/src/tflite_task_library_client/image_segmenter.ts:117
Function
select
(args: {inputs: SelectInputs, backend: MathBackendCPU})
tfjs-backend-cpu/src/kernels/Select.ts:23
Function
select
(args: {inputs: SelectInputs, backend: WebGPUBackend})
tfjs-backend-webgpu/src/kernels/Select.ts:23
Function
select
(args: {inputs: SelectInputs, backend: MathBackendWebGL})
tfjs-backend-webgl/src/kernels/Select.ts:23
Function
select
(args: {inputs: SelectInputs, backend: BackendWasm})
tfjs-backend-wasm/src/kernels/Select.ts:36
Function
selu_
* Computes scaled exponential linear element-wise. * * `x < 0 ? scale * alpha * (exp(x) - 1) : scale * x` * * ```js * const x = tf.tensor1d([-1,
tfjs-core/src/ops/selu.ts:41
Function
separableConv2d_
* 2-D convolution with separable filters. * * Performs a depthwise convolution that acts separately on channels followed * by a pointwise convoluti
tfjs-core/src/ops/separable_conv2d.ts:72
Function
serial
* Force an iterator to execute serially: each next() call will await the * prior one, so that they cannot execute concurrently.
tfjs-data/src/iterators/lazy_iterator.ts:474
Function
serialForEach
* Apply a function to every element of the stream, forcing serial execution. * * @param f A function to apply to each stream element. Should re
tfjs-data/src/iterators/lazy_iterator.ts:329
Function
serialMapAsync
* Maps this stream through a 1-to-1 transform, forcing serial execution. * * @param transform A function mapping a stream element to a transform
tfjs-data/src/iterators/lazy_iterator.ts:297
Function
serialNext
()
tfjs-data/src/iterators/lazy_iterator.ts:865
Method
setActiveTab
(tab: string)
tfjs-vis/src/components/visor.tsx:200
Function
setBackend
(requestedBackend: 'cpu'|'webgl')
tfjs-layers/src/backend/tfjs_backend.ts:32
Function
setCallHook
* Set call hook. * This is currently used for testing only. * @param callHook
tfjs-layers/src/engine/topology.ts:866
Function
setFastWeightInitDuringBuild
* Set the fast-weight-initialization flag. * * In cases where the initialized weight values will be immediately * overwritten by loaded weigh
tfjs-layers/src/engine/topology.ts:1317
Function
setFastWeightInitDuringBuild
(value: boolean)
tfjs-layers/src/layers/wrappers.ts:129
Function
setMaskMetadata
( inputs: Tensor|Tensor[], outputs: Tensor|Tensor[], previousMask?: Tensor|Tensor[])
tfjs-layers/src/engine/topology.ts:1390
Function
setModel
(model: Container)
tfjs-layers/src/callbacks.ts:23
Function
setModel
(model: Container)
tfjs-layers/src/base_callbacks.ts:84
Method
setModel
(model: LayersModel)
tfjs-node/src/callbacks.ts:268
Method
setMomentum
* Sets the momentum of the optimizer. * * @param momentum
tfjs-core/src/optimizers/momentum_optimizer.ts:102
Method
setOutputPackedMatrixWriteRegion
( startRow: number, numRows: number, startColumn: number, numColumns: number)
tfjs-backend-webgl/src/gpgpu_context.ts:440
Function
setParams
(params: Params)
tfjs-layers/src/base_callbacks.ts:61
Function
setPlatform
(platformName: string, platform: Platform)
tfjs-core/src/globals.ts:382
Method
setStates
(states: Tensor[])
tfjs-layers/src/layers/recurrent.ts:416
Function
setSystemFetch
(fetchFn: FetchFn)
tfjs-core/src/platforms/platform_node.ts:33
Function
setThreadsCount
(numThreads: number)
tfjs-backend-wasm/src/backend_wasm.ts:525
Method
setTimeoutCustom
()
tfjs-react-native/src/platform_react_native.ts:172
Method
setTimeoutCustom
(functionRef: Function, delay: number)
tfjs-core/src/platforms/platform_browser.ts:65
Method
setUp
(self)
tfjs-converter/python/test_nightly_pip_package.py:34
Method
setUp
(self)
tfjs-converter/python/test_pip_package.py:671
Method
setUp
(self)
tfjs-converter/python/tensorflowjs/write_weights_test.py:29
Method
setUp
(self)
tfjs-converter/python/tensorflowjs/read_weights_test.py:34
Method
setUp
(self)
tfjs-converter/python/tensorflowjs/converters/wizard_test.py:40
Method
setUp
(self)
tfjs-converter/python/tensorflowjs/converters/normalize_bias_add_test.py:30
Method
setUp
(self)
tfjs-converter/python/tensorflowjs/converters/tf_module_mapper_test.py:24
Method
setUp
(self)
tfjs-converter/python/tensorflowjs/converters/keras_h5_conversion_test.py:40
Method
setUp
(self)
tfjs-converter/python/tensorflowjs/converters/converter_test.py:330
Method
setUp
(self)
tfjs-converter/python/tensorflowjs/converters/converter_test.py:423
Method
setUp
(self)
tfjs-converter/python/tensorflowjs/converters/keras_tfjs_loader_test.py:38
Method
setUp
(self)
tfjs-converter/python/tensorflowjs/converters/fold_batch_norms_test.py:34
Method
setUp
(self)
tfjs-converter/python/tensorflowjs/converters/fuse_prelu_test.py:36
Method
setUp
(self)
tfjs-converter/python/tensorflowjs/converters/fuse_depthwise_conv2d_test.py:30
Method
setUp
(self)
tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2_test.py:48
Method
setUpClass
(cls)
tfjs-converter/python/test_pip_package.py:191
Function
setWeights
* Sets the weights of the layer, from Tensors. * * @param weights a list of Tensors. The number of arrays and their shape * must match numb
tfjs-layers/src/engine/topology.ts:1220
Function
setWeights
(weights: Tensor[])
tfjs-layers/src/layers/wrappers.ts:113
Function
setWeights
(weightValues: NamedTensor[])
tfjs-core/src/optimizers/optimizer.ts:148
Method
setWeights
* Set the weights of the model. * * @param weights An `Array` of `tf.Tensor`s with shapes and types matching * the output of `getWeights(
tfjs-layers/src/layers/recurrent.ts:2089
Method
setWeights
(weightValues: NamedTensor[])
tfjs-core/src/optimizers/adam_optimizer.ts:145
Method
setWeights
(weightValues: NamedTensor[])
tfjs-core/src/optimizers/rmsprop_optimizer.ts:170
Method
setWeights
(weightValues: NamedTensor[])
tfjs-core/src/optimizers/adamax_optimizer.ts:141
Method
setWeights
(weightValues: NamedTensor[])
tfjs-core/src/optimizers/adadelta_optimizer.ts:123
Method
setWeights
(weightValues: NamedTensor[])
tfjs-core/src/optimizers/momentum_optimizer.ts:112
Method
setWeights
(weightValues: NamedTensor[])
tfjs-core/src/optimizers/adagrad_optimizer.ts:101
Function
setdiff1dAsync_
* Computes the difference between two lists of numbers. * * Given a Tensor `x` and a Tensor `y`, this operation returns a Tensor `out` * that repre
tfjs-core/src/ops/setdiff1d_async.ts:51
Function
setup
()
tfjs-data/demo/boston-housing/ui.ts:34
Function
setup
(backend: BackendWasm)
tfjs-backend-wasm/src/kernels/FusedConv2D.ts:34
Function
setup
(backend: BackendWasm)
tfjs-backend-wasm/src/kernels/Select.ts:26
Function
setup
(backend: BackendWasm)
tfjs-backend-wasm/src/kernels/Transform.ts:30
Function
setup
(backend: BackendWasm)
tfjs-backend-wasm/src/kernels/SparseSegmentReduction.ts:29
Function
setup
(backend: BackendWasm)
tfjs-backend-wasm/src/kernels/All.ts:26
Function
setup
(backend: BackendWasm)
tfjs-backend-wasm/src/kernels/Any.ts:26
Function
setup
(backend: BackendWasm)
tfjs-backend-wasm/src/kernels/ResizeBilinear.ts:29
Function
setup
(backend: BackendWasm)
tfjs-backend-wasm/src/kernels/CropAndResize.ts:35
Function
setup
(backend: BackendWasm)
tfjs-backend-wasm/src/kernels/StridedSlice.ts:30
Function
setup
(backend: BackendWasm)
tfjs-backend-wasm/src/kernels/ResizeNearestNeighbor.ts:45
Function
setup
(backend: BackendWasm)
tfjs-backend-wasm/src/kernels/MaxPool3D.ts:31
← previous
next →
5,301–5,400 of 6,022, ranked by callers