MCPcopy Create free account

hub / github.com/tensorflow/tfjs / functions

Functions6,022 in github.com/tensorflow/tfjs

Functionpack
( args: {inputs: PackInputs, backend: BackendWasm, attrs: PackAttrs})
tfjs-backend-wasm/src/kernels/Pack.ts:24
FunctionpackXYSampleWeight
(x: Tensor, y?: Tensor, sampleWeight?: Tensor)
tfjs-layers/src/layers/nlp/utils.ts:72
FunctionpackageSelected
(pkg: string)
scripts/publish-npm.ts:280
Functionpad1d_
* Pads a `tf.Tensor1D` with a given value and paddings. See `pad` for details.
tfjs-core/src/ops/pad1d.ts:26
Functionpad2d_
* Pads a `tf.Tensor2D` with a given value and paddings. See `pad` for details.
tfjs-core/src/ops/pad2d.ts:26
Functionpad3d_
* Pads a `tf.Tensor3D` with a given value and paddings. See `pad` for details.
tfjs-core/src/ops/pad3d.ts:26
Functionpad4d_
* Pads a `tf.Tensor4D` with a given value and paddings. See `pad` for details.
tfjs-core/src/ops/pad4d.ts:26
MethodpadTokenId
()
tfjs-layers/src/layers/nlp/models/gpt2/gpt2_tokenizer.ts:104
FunctionpadV2
( args: {inputs: PadV2Inputs, backend: MathBackendCPU, attrs: PadV2Attrs})
tfjs-backend-cpu/src/kernels/PadV2.ts:23
FunctionpadV2
(args: {inputs: PadV2Inputs, backend: WebGPUBackend, attrs: PadV2Attrs})
tfjs-backend-webgpu/src/kernels/PadV2.ts:26
Functionpad_
* Pads a `tf.Tensor` with a given value and paddings. * * This operation implements `CONSTANT` mode. For `REFLECT` and `SYMMETRIC`, * refer to `tf.
tfjs-core/src/ops/pad.ts:53
FunctionparseWeights
(weights: NamedTensorMap)
tfjs-layers/src/engine/container.ts:666
Functionpermute
(args: PermuteLayerArgs)
tfjs-layers/src/exports_layers.ts:669
Functionpip_main
Entry point for pip-packaged binary. Note that pip-packaged binary calls the entry method without any arguments, which is why this method is need
tfjs-converter/python/tensorflowjs/converters/wizard.py:640
Functionpip_main
Entry point for pip-packaged binary. Note that pip-packaged binary calls the entry method without any arguments, which is why this method is need
tfjs-converter/python/tensorflowjs/converters/converter.py:952
FunctionplotData
( epoch: number, trainLoss: number, valLoss: number, result?: Result)
tfjs-data/demo/boston-housing/ui.ts:52
Functionpoisson
(yTrue: Tensor, yPred: Tensor)
tfjs-layers/src/losses.ts:212
Functionpool_
* Performs an N-D pooling operation * * @param input The input tensor, of rank 4 or rank 3 of shape * `[batch, height, width, inChannels]`. If
tfjs-core/src/ops/pool.ts:59
MethodpoolingFunction
( inputs: Tensor, poolSize: [number, number], strides: [number, number], padding: PaddingMode, dat
tfjs-layers/src/layers/pooling.ts:264
MethodpoolingFunction
( inputs: Tensor, poolSize: [number, number], strides: [number, number], padding: PaddingMode, dat
tfjs-layers/src/layers/pooling.ts:392
MethodpoolingFunction
( inputs: Tensor, poolSize: [number, number], strides: [number, number], padding: PaddingMode, dat
tfjs-layers/src/layers/pooling.ts:409
MethodpoolingFunction
( inputs: Tensor, poolSize: [number, number, number], strides: [number, number, number], padding:
tfjs-layers/src/layers/pooling.ts:542
MethodpoolingFunction
( inputs: Tensor, poolSize: [number, number, number], strides: [number, number, number], padding:
tfjs-layers/src/layers/pooling.ts:561
Functionpow_
* Computes the power of one `tf.Tensor` to another. Supports broadcasting. * * Given a `tf.Tensor` x and a `tf.Tensor` y, this operation computes x^
tfjs-core/src/ops/pow.ts:55
Functionprecision
(yTrue: Tensor, yPred: Tensor)
tfjs-layers/src/metrics.ts:62
Functionprecision
(yTrue: Tensor, yPred: Tensor)
tfjs-layers/src/exports_metrics.ts:165
FunctionprecisionTestRunner
()
tfjs-react-native/integration_rn59/components/ml.ts:43
Functionpredict
Use tfjs binary to make inference and store output in file. Args: binary_path: Path to the nodejs binary. The path can be an absolute path
tfjs-inference/python/inference.py:24
Methodpredict
* Generates output predictions for the input samples. * * Computation is done in batches. * * Note: the "step" mode of predict() is curren
tfjs-layers/src/engine/training.ts:1105
Methodpredict
( x: Tensor | Tensor[], args?: ModelPredictConfig )
tfjs-layers/src/layers/nlp/utils.ts:162
Methodpredict
* Execute the inference for the input tensors. * * @param inputs The input tensors, when there is single input for the model, * inputs pa
tfjs-tfdf/src/tfdf_model.ts:93
Methodpredict
* Execute the inference for the input tensors. * * @param input The input tensors, when there is single input for the model, * inputs param s
tfjs-node/src/saved_model.ts:292
Methodpredict
* Execute the inference for the input tensors. * * @param input The input tensors, when there is single input for the model, * inputs param s
tfjs-converter/src/executor/graph_model.ts:357
Methodpredict
* Infer through MobileNet. This does standard ImageNet pre-processing before * inferring through the model. This method returns named activations a
tfjs-converter/demo/mobilenet/mobilenet.js:50
Methodpredict
(init, loop, loop2, inc)
tfjs-converter/demo/control_flow/loop_model.js:38
MethodpredictOnBatch
* Returns predictions for a single batch of samples. * * ```js * const model = tf.sequential({ * layers: [tf.layers.dense({units: 1, inp
tfjs-layers/src/engine/training.ts:1137
MethodpredictOnBatch
(x: Tensor|Tensor[])
tfjs-layers/src/layers/nlp/utils.ts:177
Functionprefetch
* Creates a `Dataset` that prefetches elements from this dataset. * * @param bufferSize: An integer specifying the number of elements to be
tfjs-data/src/dataset.ts:312
Functionprefetch
* Prefetch the first `bufferSize` items in this stream. * * Note this prefetches Promises, but makes no guarantees about when those * Promise
tfjs-data/src/iterators/lazy_iterator.ts:452
FunctionprefixWeightUrlConverter
(weightFile: string)
tfjs-core/src/io/http_test.ts:835
Functionprelu
(args: {inputs: PreluInputs, backend: WebGPUBackend})
tfjs-backend-webgpu/src/kernels/Prelu.ts:25
Functionprelu
(args: {inputs: PreluInputs, backend: MathBackendWebGL})
tfjs-backend-webgl/src/kernels/Prelu.ts:30
Functionprelu
(args: {inputs: PreluInputs, backend: BackendWasm})
tfjs-backend-wasm/src/kernels/Prelu.ts:34
Functionprelu_
* Computes leaky rectified linear element-wise with parametric alphas. * * `x < 0 ? alpha * x : f(x) = x` * * ```js * const x = tf.tensor1d([-1,
tfjs-core/src/ops/prelu.ts:43
FunctionprepareAndValidate
(tensor: TensorInfo, indices: TensorInfo)
tfjs-core/src/ops/gather_nd_util.ts:28
FunctionprepareSplitSize
( x: Tensor|TensorInfo, numOrSizeSplits: number[]|number, axis = 0)
tfjs-core/src/ops/split_util.ts:26
MethodpreprocessFeatures
* An overridable function which preprocesses features.
tfjs-layers/src/layers/nlp/utils.ts:120
MethodpreprocessLabels
* An overridable function which preprocesses labels.
tfjs-layers/src/layers/nlp/utils.ts:127
MethodpreprocessSamples
* An overridable function which preprocesses entire samples.
tfjs-layers/src/layers/nlp/utils.ts:134
MethodpreprocessSamples
(x: Tensor, y?: Tensor, sampleWeight?: Tensor)
tfjs-layers/src/layers/nlp/models/task.ts:56
Methodpreprocessor
* A `LayersModel` instance used to preprocess inputs.
tfjs-layers/src/layers/nlp/models/task.ts:75
MethodpreprocessorCls
( cls: serialization.SerializableConstructor<T> )
tfjs-layers/src/layers/nlp/models/task.ts:103
Methodpresets
( cls: serialization.SerializableConstructor<T> )
tfjs-layers/src/layers/nlp/models/task.ts:109
Methodpresets
( cls: serialization.SerializableConstructor<T> )
tfjs-layers/src/layers/nlp/models/gpt2/gpt2_causal_lm.ts:168
Methodprint
* Prints the `tf.Tensor`. See `tf.print` for details. * * @param verbose Whether to print verbose information about the tensor, * includin
tfjs-core/src/tensor.ts:473
FunctionprintMemory
(bytes)
e2e/benchmarks/local-benchmark/util.js:57
FunctionprintReleaseUnit
(releaseUnit: ReleaseUnit, id: number)
scripts/release-util.ts:267
FunctionprintTime
(elapsed)
e2e/benchmarks/local-benchmark/util.js:53
MethodprocessUnseenTokens
()
tfjs-layers/src/layers/nlp/tokenizers.ts:516
Functionprod
( args: {inputs: ProdInputs, backend: WebGPUBackend, attrs: ProdAttrs})
tfjs-backend-webgpu/src/kernels/Prod.ts:23
Functionprod
( args: {inputs: ProdInputs, backend: MathBackendWebGL, attrs: ProdAttrs})
tfjs-backend-webgl/src/kernels/Prod.ts:27
Functionprod
(args: { backend: BackendWasm, inputs: ProdInputs, attrs: ProdAttrs })
tfjs-backend-wasm/src/kernels/Prod.ts:39
Functionprod_
* Computes the product of elements across dimensions of a `tf.Tensor`. * * Reduces the input along the dimensions given in `axes`. Unless `keepDims`
tfjs-core/src/ops/prod.ts:59
Functionprofile
(f: () => (TensorContainer | Promise<TensorContainer>))
tfjs-core/src/globals.ts:145
Methodpump
()
tfjs-data/src/iterators/byte_chunk_iterator.ts:108
Methodpump
()
tfjs-data/src/iterators/string_iterator.ts:86
Methodpush
(value: T)
tfjs-data/src/util/growing_ring_buffer.ts:35
Functionqr_
* Compute QR decomposition of m-by-n matrix using Householder transformation. * * Implementation based on * [http://www.cs.cornell.edu/~bindel/cl
tfjs-core/src/ops/linalg/qr.ts:84
Functionquantize_weights
Quantizes the weights by linearly re-scaling across available bits. The weights are quantized by linearly re-scaling the values between the minim
tfjs-converter/python/tensorflowjs/quantization.py:93
FunctionqueryTimerIsEnabled
()
e2e/benchmarks/local-benchmark/util.js:71
Functionquestion
(questionStr: string)
scripts/release_notes/util.ts:61
Functionr2Score
(yTrue: Tensor, yPred: Tensor)
tfjs-layers/src/exports_metrics.ts:333
FunctionraggedGather
(args: { inputs: RaggedGatherInputs, backend: MathBackendCPU, attrs: RaggedGatherAttrs })
tfjs-backend-cpu/src/kernels/RaggedGather.ts:24
FunctionraggedGather
(args: { inputs: RaggedGatherInputs, backend: MathBackendWebGL, attrs: RaggedGatherAttrs })
tfjs-backend-webgl/src/kernels/RaggedGather.ts:23
FunctionraggedGather_
( paramsNestedSplits: Tensor[], paramsDenseValues: Tensor|TensorLike, indices: Tensor|TensorLike, outp
tfjs-core/src/ops/ragged_gather.ts:51
FunctionraggedRange
( args: {inputs: RaggedRangeInputs, backend: MathBackendCPU})
tfjs-backend-cpu/src/kernels/RaggedRange.ts:24
FunctionraggedRange
( args: {inputs: RaggedRangeInputs, backend: MathBackendWebGL})
tfjs-backend-webgl/src/kernels/RaggedRange.ts:23
FunctionraggedRange_
* Returns a RaggedTensor result composed from rtDenseValues and rtNestedSplits, * such that result[i] = [starts[i], starts[i] + deltas[i], ..., limit
tfjs-core/src/ops/ragged_range.ts:41
FunctionraggedTensorToTensor
(args: { inputs: RaggedTensorToTensorInputs, backend: MathBackendCPU, attrs: RaggedTensorToTensorAttrs }
tfjs-backend-cpu/src/kernels/RaggedTensorToTensor.ts:24
FunctionraggedTensorToTensor
(args: { inputs: RaggedTensorToTensorInputs, backend: MathBackendWebGL, attrs: RaggedTensorToTensorAttrs
tfjs-backend-webgl/src/kernels/RaggedTensorToTensor.ts:23
FunctionraggedTensorToTensor_
* Create a dense tensor from a ragged tensor, possibly altering its shape. * * The raggedTensorToTensor op creates a dense tensor from am array of r
tfjs-core/src/ops/ragged_tensor_to_tensor.ts:78
FunctionrandUniform
(a: number, b: number)
tfjs-core/src/util_base.ts:121
Functionrand_
* Creates a `tf.Tensor` with values sampled from a random number generator * function defined by the user. * * @param shape An array of integers de
tfjs-core/src/ops/rand.ts:37
FunctionrandomGamma_
* Creates a `tf.Tensor` with values sampled from a gamma distribution. * * ```js * tf.randomGamma([2, 2], 1).print(); * ``` * * @param shape An
tfjs-core/src/ops/random_gamma.ts:42
FunctionrandomNormalVariable
( shape: Shape, mean = 0.0, stddev = 1.0, dtype?: DataType, seed?: number, name = 'randomNormal')
tfjs-layers/src/variables.ts:275
FunctionrandomNormal_
* Creates a `tf.Tensor` with values sampled from a normal distribution. * * ```js * tf.randomNormal([2, 2]).print(); * ``` * * @param shape An a
tfjs-core/src/ops/random_normal.ts:41
FunctionrandomStandardNormal_
* Creates a `tf.Tensor` with values sampled from a normal distribution. * * The generated values will have mean 0 and standard deviation 1. * * ``
tfjs-core/src/ops/random_standard_normal.ts:39
FunctionrandomUniformInt_
* Creates a `tf.Tensor` with integers sampled from a uniform distribution. * * The generated values are uniform integers in the range [minval, maxva
tfjs-core/src/ops/random_uniform_int.ts:43
FunctionrandomUniformVariable
( shape: Shape, minval: number, maxval: number, dtype?: DataType, seed?: number, name = 'randomUniform
tfjs-layers/src/variables.ts:235
FunctionrandomUniform_
* Creates a `tf.Tensor` with values sampled from a uniform distribution. * * The generated values follow a uniform distribution in the range [minval
tfjs-core/src/ops/random_uniform.ts:49
FunctionrandomWidth
(args: RandomWidthArgs)
tfjs-layers/src/exports_layers.ts:1887
Functionrange
(args: {backend: MathBackendCPU, attrs: RangeAttrs})
tfjs-backend-cpu/src/kernels/Range.ts:23
Functionrange
(args: {backend: WebGPUBackend, attrs: RangeAttrs})
tfjs-backend-webgpu/src/kernels/Range.ts:24
Methodrank
()
tfjs-core/src/tensor.ts:148
Methodrank
()
tfjs-core/src/tensor.ts:298
FunctionreLU
(args?: ReLULayerArgs)
tfjs-layers/src/exports_layers.ts:123
Methodread
(dataId: DataId)
tfjs-backend-cpu/src/backend_cpu.ts:129
Methodread
(dataId: object)
tfjs-backend-webgpu/src/backend_webgpu.ts:508
Methodread
(dataId: DataId)
tfjs-core/src/engine.ts:1223
Methodread
(dataId: object)
tfjs-core/src/backends/backend.ts:113
← previousnext →5,101–5,200 of 6,022, ranked by callers