MCPcopy Create free account

hub / github.com/tensorflow/tfjs / functions

Functions6,022 in github.com/tensorflow/tfjs

⨍ Functions 6,022◇ Types & classes 1,932↳ Endpoints 2
↓ 7 callersFunctionpredict
()
e2e/benchmarks/benchmark_util_test.js:109
↓ 7 callersFunctionreduce
( x: TensorInfo, axis: number|number[], keepDims: boolean, reduceType: ReduceTypes, backend: WebGPUBac
tfjs-backend-webgpu/src/kernel_utils/reduce.ts:35
↓ 7 callersFunctionreduce
( x: TensorInfo, dtype: DataType, reductionType: ReduceTypes, backend: MathBackendWebGL)
tfjs-backend-webgl/src/kernel_utils/reduce.ts:46
↓ 7 callersFunctionregisterKernel
(config: KernelConfig)
tfjs-core/src/kernel_registry.ts:131
↓ 7 callersMethodrelu6
()
tfjs-core/src/public/chained_ops/relu6.ts:23
↓ 7 callersFunctionresetBackend
* Reset the target backend. * * @param backendName The name of the backend to be reset.
e2e/benchmarks/benchmark_util.js:661
↓ 7 callersFunctionround
* Rounds a value depending on the rounding mode * @param value * @param roundingMode A string from: 'ceil', 'round', 'floor'. If none is * prov
tfjs-core/src/ops/conv_util.ts:548
↓ 7 callersMethodscatter
* Scatter the values of a Tensor in specific indices of a TensorArray. * @param indices number[] values in [0, max_value). If the * TensorArr
tfjs-converter/src/executor/tensor_array.ts:247
↓ 7 callersMethodsize
()
tfjs-converter/src/executor/tensor_array.ts:66
↓ 7 callersFunctionsliceArraysByIndices
( arrays: Tensor|Tensor[], indices: Tensor1D)
tfjs-layers/src/engine/training_tensors.ts:202
↓ 7 callersFunctionsliceUpdate
( inputs: Tensor, startIndices: number[], updates: Tensor)
tfjs-layers/src/layers/nlp/utils.ts:44
↓ 7 callersFunctionsparseTensorValue2x3x4
()
tfjs-core/src/ops/sparse/sparse_reshape_test.ts:30
↓ 7 callersFunctionstringByteLength
(str: string)
tfjs-core/src/io/io_utils.ts:411
↓ 7 callersFunctionsubSurface
(parent: Drawable, name: string, opts: Options = {})
tfjs-vis/src/util/dom.ts:30
↓ 7 callersFunctiontanh
(x: number)
tfjs-core/src/util_base.ts:241
↓ 7 callersFunctiontensorArrTo2DArr
(inputs: Tensor[])
tfjs-layers/src/layers/nlp/utils.ts:30
↓ 7 callersFunctiontensorToArr
(input: Tensor)
tfjs-layers/src/layers/nlp/utils.ts:26
↓ 7 callersFunctiontranspose
( args: {inputs: TransposeInputs, backend: BackendWasm, attrs: TransposeAttrs})
tfjs-backend-wasm/src/kernels/Transpose.ts:41
↓ 7 callersFunctionuncapitalize
(name: Name)
tfjs-converter/src/operations/executors/test_helper.ts:99
↓ 6 callersFunction_createKerasModel
Create a Keras model for testing. Args: layer_name_prefix: A prefix string for layer names. This helps avoid clashes in layer names betwe
tfjs-converter/python/test_pip_package.py:44
↓ 6 callersMethod_create_saved_model
Test a basic model with functions to make sure functions are inlined.
tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2_test.py:248
↓ 6 callersMethodadadelta
* Constructs a `tf.AdadeltaOptimizer` that uses the Adadelta algorithm. * See [https://arxiv.org/abs/1212.5701](https://arxiv.org/abs/1212.5701)
tfjs-core/src/optimizers/optimizer_constructors.ts:145
↓ 6 callersMethodadd
* Add a key-value pair to the FeedDict. * * @param key The key of the feed. * @param value The value of the tensor feed. * @param mask The
tfjs-layers/src/engine/executor.ts:95
↓ 6 callersFunctionassertNonNull
(a: TensorLike)
tfjs-core/src/util_base.ts:164
↓ 6 callersMethodassertNotDisposed
()
tfjs-layers/src/variables.ts:117
↓ 6 callersFunctionbackpropagateGradients
( tensorAccumulatedGradientMap: {[tensorId: number]: Tensor}, filteredTape: TapeNode[], tidy: (f: Func
tfjs-core/src/tape.ts:131
↓ 6 callersMethodcapture
()
tfjs-data/src/iterators/webcam_iterator.ts:201
↓ 6 callersFunctioncheckStringTypeUnionValue
( values: string[], label: string, value: string)
tfjs-layers/src/utils/generic_utils.ts:397
↓ 6 callersFunctioncomplex
(args: {inputs: ComplexInputs, backend: WebGPUBackend})
tfjs-backend-webgpu/src/kernels/Complex.ts:31
↓ 6 callersMethodcreateFloat32MatrixTexture
(rows: number, columns: number)
tfjs-backend-webgl/src/gpgpu_context.ts:196
↓ 6 callersMethodcreatePackedMatrixTexture
(rows: number, columns: number)
tfjs-backend-webgl/src/gpgpu_context.ts:237
↓ 6 callersMethodcreateProgram
(fragmentShader: WebGLShader)
tfjs-backend-webgl/src/gpgpu_context.ts:331
↓ 6 callersFunctioncreateSimpleTwoLayerContainer
()
tfjs-layers/src/engine/container_test.ts:366
↓ 6 callersFunctioncreateStaticHashtable
( keys: K[], values: V[], defaultVal: V)
tfjs-layers/src/layers/nlp/tokenizers_utils.ts:98
↓ 6 callersFunctioncreateTensorShapeAttr
(value: number[])
tfjs-converter/src/operations/executors/test_helper.ts:41
↓ 6 callersMethodcustomGrad
(f: CustomGradientFunc<T>)
tfjs-core/src/engine.ts:1157
↓ 6 callersMethoddecodeJpeg
( contents: Uint8Array, channels: number, ratio: number, fancyUpscaling: boolean, tryRecoverTrunca
tfjs-node/src/nodejs_kernel_backend.ts:338
↓ 6 callersMethoddeleteTensor
(tensorId: number)
tfjs-node/src/tfjs_binding.ts:42
↓ 6 callersFunctionderegisterOp
(name: string)
tfjs-converter/src/operations/custom_op/register.ts:78
↓ 6 callersFunctiondeserializeKerasObject
( identifier: string|serialization.ConfigDict, moduleObjects = {} as {[objName: string]: any}, cus
tfjs-layers/src/utils/generic_utils.ts:195
↓ 6 callersFunctiondisposeIntermediateTensorInfoOrNull
( backend: WebGPUBackend, tensorInfo: TensorInfo)
tfjs-backend-webgpu/src/kernels/TopK.ts:28
↓ 6 callersMethodendScope
* End a scope. Use this with startScope() to achieve the same functionality * as scope() without the need for a function closure.
tfjs-core/src/engine.ts:1071
↓ 6 callersMethodensureCommandEncoderReady
()
tfjs-backend-webgpu/src/backend_webgpu.ts:311
↓ 6 callersMethodenterFrame
* Enter a new frame, a new context is pushed on the current context list. * @param frameId new frame id
tfjs-converter/src/executor/execution_context.ts:116
↓ 6 callersFunctionexpectTensorsNotClose
( t1: Tensor|number[], t2: Tensor|number[], epsilon?: number)
tfjs-layers/src/utils/test_utils.ts:59
↓ 6 callersFunctiongetDefaultHeight
(element: HTMLElement)
tfjs-vis/src/util/dom.ts:85
↓ 6 callersFunctiongetDefaultWidth
(element: HTMLElement)
tfjs-vis/src/util/dom.ts:74
↓ 6 callersFunctiongetEffectiveFilterSize
(filterSize: number, dilation: number)
tfjs-core/src/ops/conv_util.ts:421
↓ 6 callersFunctiongetNoiseShape
(x: Tensor, noiseShape?: number[])
tfjs-core/src/ops/dropout_util.ts:29
↓ 6 callersFunctiongetSamplerFromInInfo
( inInfo: InputInfo, enableShapeUniforms = false)
tfjs-backend-webgl/src/shader_compiler.ts:154
↓ 6 callersFunctiongetSqueezedParams
(params: string[], keptDims: number[])
tfjs-backend-webgl/src/shader_compiler.ts:1913
↓ 6 callersMethodgetTexture
(dataId: DataId)
tfjs-backend-webgl/src/backend_webgl.ts:716
↓ 6 callersFunctiongetTunableRange
* Heuristically determine flag's value range based on flag's default value. * * Assume that the flag's default value has already chosen the best opt
e2e/benchmarks/local-benchmark/index.js:212
↓ 6 callersFunctiongetUniformSampler
(inputInfo: InputInfo)
tfjs-backend-webgl/src/shader_compiler.ts:1723
↓ 6 callersFunctionhistory
( container: Drawable, history: HistoryLike, metrics: string[], opts: HistoryOptions = {})
tfjs-vis/src/show/history.ts:105
↓ 6 callersMethodinsert
* Insert token <=> encoded outputs pairs.
tfjs-layers/src/layers/nlp/tokenizers_utils.ts:136
↓ 6 callersFunctionisString
(value: {})
tfjs-core/src/util_base.ts:520
↓ 6 callersFunctioniteratorFromFunction
( func: () => IteratorResult<T>| Promise<IteratorResult<T>>)
tfjs-data/src/iterators/lazy_iterator.ts:65
↓ 6 callersFunctionloadDictionary
(modelUrl: string)
tfjs-automl/src/util.ts:26
↓ 6 callersFunctionlog
(...msg: Array<{}>)
tfjs-core/src/log.ts:26
↓ 6 callersFunctionmaximum
(config?: SymbolicTensor[]|Tensor[]|LayerArgs)
tfjs-layers/src/layers/merge.ts:553
↓ 6 callersMethodmomentum
* Constructs a `tf.MomentumOptimizer` that uses momentum gradient * descent. * * See * [http://proceedings.mlr.press/v28/sutskever13.pdf](
tfjs-core/src/optimizers/optimizer_constructors.ts:83
↓ 6 callersMethodnext
()
tfjs-data/src/dataset_test.ts:34
↓ 6 callersMethodnext
()
tfjs-layers/src/backend/random_seed.ts:23
↓ 6 callersFunctionnonMaxSuppSanityCheck
( boxes: Tensor2D, scores: Tensor1D, maxOutputSize: number, iouThreshold: number, scoreThreshold: numb
tfjs-core/src/ops/nonmax_util.ts:21
↓ 6 callersFunctionnormalizeArray
( value: number|number[], n: number, name: string)
tfjs-layers/src/utils/conv_utils.ts:24
↓ 6 callersFunctionpad
( args: {inputs: PadV2Inputs, backend: BackendWasm, attrs: PadV2Attrs})
tfjs-backend-wasm/src/kernels/PadV2.ts:44
↓ 6 callersFunctionprofileInference
* Executes `predict()` and returns a promise that resolves with information * about the memory usage: * - `newBytes`: the number of new bytes alloca
e2e/benchmarks/benchmark_util.js:465
↓ 6 callersFunctionrange
(begin: number, end: number)
tfjs-layers/src/utils/math_utils.ts:157
↓ 6 callersFunctionrange
(args: {backend: BackendWasm, attrs: RangeAttrs})
tfjs-backend-wasm/src/kernels/Range.ts:24
↓ 6 callersFunctionreplaceHTMLVideoElementSource
( videoElement: HTMLVideoElement)
tfjs-data/src/util/test_utils.ts:37
↓ 6 callersFunctionrnn
( stepFunction: RnnStepFunction, inputs: Tensor, initialStates: Tensor[], goBackwards = false, mask?:
tfjs-layers/src/layers/recurrent.ts:142
↓ 6 callersMethodsave
(modelArtifacts: tf.io.ModelArtifacts)
tfjs-node/src/io/file_system.ts:80
↓ 6 callersFunctionscatter
( tensor: Tensor, indices: number[], elementShape: number[], numElements?: number)
tfjs-converter/src/executor/tensor_list.ts:358
↓ 6 callersFunctionscatterImpl
( indices: TensorBuffer<R, 'int32'>, updates: TensorBuffer<R, D>, shape: number[], outputSize: number,
tfjs-backend-cpu/src/kernels/Scatter_impl.ts:26
↓ 6 callersFunctionscheduleModels
* Get the models to benchmark for the day running the script. (All models are * spilted to n buckets and n === period, associated with the date of th
e2e/benchmarks/browserstack-benchmark/app.js:444
↓ 6 callersFunctionsetupTestFilters
( testFilters: TestFilter[], customInclude: (name: string) => boolean)
tfjs-core/src/jasmine_util.ts:115
↓ 6 callersFunctionslice
( args: {inputs: SliceInputs, attrs: SliceAttrs, backend: BackendWasm})
tfjs-backend-wasm/src/kernels/Slice.ts:23
↓ 6 callersFunctionsoftmax
(args?: SoftmaxLayerArgs)
tfjs-layers/src/exports_layers.ts:192
↓ 6 callersFunctionsqArr
(arr: number[])
tfjs-core/src/ops/local_response_normalization_test.ts:23
↓ 6 callersFunctionsqueezeInputInfo
( inInfo: InputInfo, squeezedShape: number[])
tfjs-backend-webgl/src/shader_compiler.ts:1905
↓ 6 callersMethodstart
()
tfjs-data/src/iterators/webcam_iterator.ts:96
↓ 6 callersMethodsubmitQueue
()
tfjs-backend-webgpu/src/backend_webgpu.ts:289
↓ 6 callersFunctionsumArr
(arr: number[])
tfjs-core/src/ops/local_response_normalization_test.ts:24
↓ 6 callersFunctiontensorStats
(input: Tensor)
tfjs-vis/src/util/math.ts:97
↓ 6 callersFunctiontile
( args: {inputs: TileInputs, backend: MathBackendCPU, attrs: TileAttrs})
tfjs-backend-cpu/src/kernels/Tile.ts:24
↓ 6 callersFunctiontimeInference
* Executes `predict()` and times the inference process for `numRuns` rounds. * Then returns a promise that resolves with information about the infere
e2e/benchmarks/benchmark_util.js:246
↓ 6 callersFunctiontoSnakeCase
(name: string)
tfjs-layers/src/utils/generic_utils.ts:109
↓ 6 callersMethodwrap
* Map any index into the range 0 <= index < 2*capacity.
tfjs-data/src/util/ring_buffer.ts:50
↓ 5 callersFunctionGetPreviousNonBlankLine
Return the most recent non-blank line and its line number. Args: clean_lines: A CleansedLines instance containing the file contents. linenu
tfjs-backend-wasm/tools/cpplint.py:3713
↓ 5 callersFunctionParseNolintSuppressions
Updates the global list of line error-suppressions. Parses any NOLINT comments on the current line, updating the global error_suppressions store.
tfjs-backend-wasm/tools/cpplint.py:583
↓ 5 callersMethod_createNestedSequentialModel
(self)
tfjs-converter/python/test_pip_package.py:689
↓ 5 callersFunctionaccuracy
( labels: Tensor, predictions: Tensor)
tfjs-vis/src/util/math.ts:242
↓ 5 callersMethodadagrad
* Constructs a `tf.AdagradOptimizer` that uses the Adagrad algorithm. * See * [http://www.jmlr.org/papers/volume12/duchi11a/duchi11a.pdf]( *
tfjs-core/src/optimizers/optimizer_constructors.ts:185
↓ 5 callersMethodapply
(shape: Shape, dtype?: DataType)
tfjs-layers/src/layers/convolutional_recurrent.ts:402
↓ 5 callersMethodasType
(this: T, dtype: DataType)
tfjs-core/src/public/chained_ops/as_type.ts:25
↓ 5 callersFunctionassert
(val: boolean, message?: string)
tfjs-layers/src/utils/generic_utils.ts:39
↓ 5 callersFunctionassertNotComplex
( tensor: TensorInfo|TensorInfo[], opName: string)
tfjs-backend-webgl/src/webgl_util.ts:710
↓ 5 callersFunctionbrowserIndexedDB
(modelPath: string)
tfjs-core/src/io/indexed_db.ts:261
↓ 5 callersFunctionbrowserLocalStorage
(modelPath: string)
tfjs-core/src/io/local_storage.ts:348
← previousnext →701–800 of 6,022, ranked by callers