MCPcopy Index your code

hub / github.com/tensorflow/tfjs / functions

Functions6,022 in github.com/tensorflow/tfjs

↓ 3,971 callersMethoddata
()
tfjs-core/src/tensor.ts:541
↓ 3,222 callersFunctionexpectArraysClose
( actual: TypedArray|number|RecursiveArray<number>, expected: TypedArray|number|RecursiveArray<number>
tfjs-core/src/test_util.ts:26
↓ 1,193 callersMethodpush
* Adds an item to the end of the buffer.
tfjs-data/src/util/ring_buffer.ts:104
↓ 903 callersMethodjoin
* Concatenate a number of ArrayBuffers into one. * * @param buffers An array of ArrayBuffers to concatenate, or a single * ArrayBuffer.
tfjs-core/src/io/composite_array_buffer.ts:50
↓ 875 callersFunctionexpectTensorsClose
( actual: Tensor|number[], expected: Tensor|number[], epsilon?: number)
tfjs-layers/src/utils/test_utils.ts:26
↓ 871 callersMethodscalar
* Write a scalar summary. * * @param name A name of the summary. The summary tag for TensorBoard will be * this name. * @param value A r
tfjs-node/src/tensorboard.ts:42
↓ 726 callersMethodget
* Returns the value in the buffer at the provided location. * * @param locs The location indices. * * @doc {heading: 'Tensors', subheading
tfjs-core/src/tensor.ts:100
↓ 635 callersMethodadd
(b: Tensor|TensorLike)
tfjs-core/src/public/chained_ops/add.ts:24
↓ 622 callersFunctiontensor2d
( values: TensorLike2D, shape?: [number, number], dtype?: DataType)
tfjs-core/src/ops/tensor2d.ts:48
↓ 593 callersMethodmemory
()
tfjs-core/src/backends/backend.ts:40
↓ 592 callersMethodslice
( this: T, begin: number|number[], size?: number|number[])
tfjs-core/src/public/chained_ops/slice.ts:25
↓ 537 callersFunctiongetParamValue
( paramName: string, node: Node, tensorMap: NamedTensorsMap, context: ExecutionContext, resourceManage
tfjs-converter/src/operations/executors/utils.ts:25
↓ 471 callersFunctiondescribeWithFlags
( name: string, constraints: Constraints, tests: (env: TestEnv) => void)
tfjs-core/src/jasmine_util.ts:209
↓ 467 callersFunctionexpectArraysEqual
(actual: TensorLike, expected: TensorLike)
tfjs-core/src/test_util.ts:113
↓ 441 callersMethodarray
* Returns the tensor data as a nested array. The transfer of data is done * asynchronously. * * @doc {heading: 'Tensors', subheading: 'Classe
tfjs-core/src/tensor.ts:326
↓ 441 callersMethodgrad
(dy: Tensor)
tfjs-core/src/engine.ts:509
↓ 392 callersMethodget
(dataId: DataId)
tfjs-core/src/backends/backend.ts:53
↓ 376 callersFunctionconvertToTensor
( x: T|TensorLike, argName: string, functionName: string, parseAsDtype: DataType|'numeric'|'string_or_
tfjs-core/src/tensor_util_env.ts:98
↓ 375 callersFunctiontensor1d
(values: TensorLike1D, dtype?: DataType)
tfjs-core/src/ops/tensor1d.ts:41
↓ 365 callersMethodlog
(this: T)
tfjs-core/src/public/chained_ops/log.ts:25
↓ 356 callersMethodset
* Sets a value in the buffer at a given location. * * @param value The value to set. * @param locs The location indices. * * @doc {hea
tfjs-core/src/tensor.ts:80
↓ 355 callersFunctioncreateTensorAttr
(index: number)
tfjs-converter/src/operations/executors/test_helper.ts:63
↓ 334 callersMethodapply
( inputs: Tensor|Tensor[]|SymbolicTensor|SymbolicTensor[], kwargs?: Kwargs)
tfjs-layers/src/layers/recurrent.ts:641
↓ 268 callersMethodclone
(x: T)
tfjs-core/src/tensor.ts:200
↓ 266 callersMethoddataSync
()
tfjs-core/src/tensor.ts:540
↓ 264 callersFunctionscalar
( value: number|boolean|string|Uint8Array, dtype?: DataType)
tfjs-core/src/ops/scalar.ts:38
↓ 261 callersFunctionones
(shape: number[])
tfjs-core/src/engine.ts:1284
↓ 256 callersMethodconcat
(tensors: T|Array<T|TensorLike>, axis?: number)
tfjs-core/src/public/chained_ops/concat.ts:23
↓ 246 callersFunctionenv
()
tfjs-core/src/environment.ts:212
↓ 242 callersFunctionmemory
()
tfjs-core/src/globals.ts:110
↓ 242 callersFunctionop
(f: {[name: string]: T})
tfjs-core/src/ops/operation.ts:27
↓ 221 callersMethodmul
(b: Tensor|TensorLike)
tfjs-core/src/public/chained_ops/mul.ts:23
↓ 214 callersFunctiontidy
( nameOrFn: string|ScopeFn<T>, fn?: ScopeFn<T>)
tfjs-core/src/globals.ts:190
↓ 212 callersMethodgetBool
(flagName: string)
tfjs-core/src/environment.ts:116
↓ 212 callersMethodpredict
* Execute the inference for the input tensors. * * @param input The input tensors, when there is single input for the model, * inputs param s
tfjs-core/src/model_types.ts:81
↓ 211 callersFunctiontensor3d
( values: TensorLike3D, shape?: [number, number, number], dtype?: DataType)
tfjs-core/src/ops/tensor3d.ts:48
↓ 209 callersFunctionzeros
( shape: ShapeMap[R], dtype: DataType = 'float32')
tfjs-core/src/ops/zeros.ts:38
↓ 207 callersMethodmatMul
( b: Tensor|TensorLike, transposeA?: boolean, transposeB?: boolean)
tfjs-core/src/public/chained_ops/mat_mul.ts:23
↓ 205 callersMethodapply
( inputs: Tensor|Tensor[]|SymbolicTensor|SymbolicTensor[], kwargs?: Kwargs)
tfjs-layers/src/engine/input_layer.ts:118
↓ 204 callersMethodnumDataIds
Returns number of data ids currently in the storage.
tfjs-core/src/backends/backend.ts:42
↓ 203 callersMethodreshape
(shape: number[])
tfjs-core/src/public/chained_ops/reshape.ts:23
↓ 201 callersMethodrunKernel
* Execute a kernel with the given name and return the output tensor. * * @param kernelName The name of the kernel to execute. * @param inputs
tfjs-core/src/engine.ts:539
↓ 201 callersMethodsqrt
(this: T)
tfjs-core/src/public/chained_ops/sqrt.ts:25
↓ 186 callersMethodread
(dataId: DataId)
tfjs-core/src/tensor.ts:186
↓ 185 callersMethodgetWeights
()
tfjs-layers/src/layers/wrappers.ts:306
↓ 184 callersFunctionreshape
(args: ReshapeLayerArgs)
tfjs-layers/src/exports_layers.ts:637
↓ 184 callersFunctiontensor
( values: TensorLike|WebGLData|WebGPUData, shape?: ShapeMap[R], dtype?: DataType)
tfjs-core/src/ops/tensor.ts:204
↓ 164 callersMethodconv2d
( filter: Tensor4D|TensorLike4D, strides: [number, number]|number, pad: 'valid'|'same'|number,
tfjs-core/src/public/chained_ops/conv2d.ts:23
↓ 163 callersMethodpow
(exp: Tensor|TensorLike)
tfjs-core/src/public/chained_ops/pow.ts:23
↓ 161 callersMethodfloor
(this: T)
tfjs-core/src/public/chained_ops/floor.ts:25
↓ 157 callersFunctiondescribeMathCPUAndGPU
(testName: string, tests: () => void)
tfjs-layers/src/utils/test_utils.ts:88
↓ 153 callersMethodsplit
(numOrSizeSplits: number[]|number, axis?: number)
tfjs-core/src/public/chained_ops/split.ts:23
↓ 151 callersMethodmax
(axis?: number|number[], keepDims?: boolean)
tfjs-core/src/public/chained_ops/max.ts:24
↓ 149 callersFunctioncreateTensorsTypeOpAttr
( attrName: string, tensorsOrDtype: tf.Tensor|tf.Tensor[]|tf.DataType)
tfjs-node/src/nodejs_kernel_backend.ts:740
↓ 149 callersMethodexp
(this: T)
tfjs-core/src/public/chained_ops/exp.ts:25
↓ 147 callersMethodapply
* Calculate the activation function. * * @param x: Input. * @param alpha: Scaling factor the negative section. * @return Output of the ELU
tfjs-layers/src/activations.ts:46
↓ 144 callersMethodthrowIfDisposed
()
tfjs-core/src/tensor.ts:459
↓ 139 callersFunctiongetGlobalTensorClass
()
tfjs-core/src/tensor.ts:519
↓ 130 callersFunctionMatch
Matches the string with the pattern, caching the compiled regexp.
tfjs-backend-wasm/tools/cpplint.py:657
↓ 129 callersFunctionSearch
Searches the string for the pattern, caching the compiled regexp.
tfjs-backend-wasm/tools/cpplint.py:685
↓ 126 callersMethoddisposeData
(dataId: DataId, force?: boolean)
tfjs-core/src/backends/backend.ts:35
↓ 126 callersMethodhas
(dataId: DataId)
tfjs-core/src/backends/backend.ts:65
↓ 124 callersMethodcompile
* Configures and prepares the model for training and evaluation. Compiling * outfits the model with an optimizer, loss, and/or metrics. Calling `
tfjs-layers/src/engine/training.ts:583
↓ 124 callersMethodmin
(axis?: number|number[], keepDims?: boolean)
tfjs-core/src/public/chained_ops/min.ts:24
↓ 123 callersMethodceil
(this: T)
tfjs-core/src/public/chained_ops/ceil.ts:25
↓ 123 callersMethodmakeTensorInfo
* Create a data bucket in cpu backend. * @param shape Shape of the `TensorInfo`. * @param dtype DType of the `TensorInfo`. * @param values Th
tfjs-backend-cpu/src/backend_cpu.ts:79
↓ 122 callersMethodtidy
(nameOrFn: string|ScopeFn<T>, fn?: ScopeFn<T>)
tfjs-core/src/engine.ts:441
↓ 120 callersMethodassign
* Assign a new `tf.Tensor` to this variable. The new `tf.Tensor` must have * the same shape and dtype as the old `tf.Tensor`. * * @param newV
tfjs-core/src/tensor.ts:583
↓ 117 callersFunctiongetInitializer
(identifier: InitializerIdentifier|Initializer| serialization.ConfigDict)
tfjs-layers/src/initializers.ts:636
↓ 116 callersFunctiongrads
* Provided `f(x1, x2,...)`, returns another function `g([x1, x2,...], dy?)`, * which gives an array of gradients of `f()` with respect to each input
tfjs-core/src/gradients.ts:111
↓ 115 callersMethodexecuteSingleOutput
* Executes a TensorFlow Eager Op that provides one output Tensor. * @param name The name of the Op to execute. * @param opAttrs The list of Op a
tfjs-node/src/nodejs_kernel_backend.ts:189
↓ 112 callersMethodbackend
()
tfjs-core/src/engine.ts:211
↓ 112 callersFunctionvalidateParam
( node: Node, opMappers: OpMapper[], tfOpName?: string)
tfjs-converter/src/operations/executors/test_helper.ts:76
↓ 110 callersMethoddisposeIntermediateTensorInfo
(tensorInfo: TensorInfo)
tfjs-backend-webgl/src/backend_webgl.ts:267
↓ 109 callersMethoddispose
* Disposes `tf.Tensor` from memory. * * @doc {heading: 'Tensors', subheading: 'Classes'}
tfjs-core/src/tensor.ts:443
↓ 109 callersFunctiontensor4d
( values: TensorLike4D, shape?: [number, number, number, number], dtype?: DataType)
tfjs-core/src/ops/tensor4d.ts:48
↓ 108 callersMethodmakeOutput
* Make a tensor info for the output of an op. If `memoryOffset` is not * present, this method allocates memory on the WASM heap. If `memoryOffset`
tfjs-backend-wasm/src/backend_wasm.ts:204
↓ 106 callersMethodgetNumber
(flagName: string)
tfjs-core/src/environment.ts:112
↓ 105 callersMethoddispose
()
tfjs-core/src/optimizers/sgd_optimizer.ts:76
↓ 105 callersMethoddispose
* Delete the SavedModel from nodeBackend and delete corresponding session in * the C++ backend if the session is only used by this TFSavedModel.
tfjs-node/src/saved_model.ts:238
↓ 105 callersMethoddisposeIntermediateTensorInfo
(tensorInfo: TensorInfo)
tfjs-backend-cpu/src/backend_cpu.ts:195
↓ 104 callersMethodsetBackend
(backendName: string)
tfjs-core/src/engine.ts:274
↓ 102 callersMethodround
(this: T)
tfjs-core/src/public/chained_ops/round.ts:25
↓ 102 callersMethodrunWebGLProgram
( program: GPGPUProgram, inputs: TensorInfo[], outputDtype: DataType, customUniformValues?: number
tfjs-backend-webgl/src/backend_webgl.ts:874
↓ 101 callersMethodabs
(this: T)
tfjs-core/src/public/chained_ops/abs.ts:25
↓ 101 callersMethodtranspose
(perm?: number[])
tfjs-core/src/public/chained_ops/transpose.ts:24
↓ 98 callersFunctioncreateNumberAttr
(value: number)
tfjs-converter/src/operations/executors/test_helper.ts:21
↓ 98 callersMethodrunWebGPUProgram
( program: webgpu_program.WebGPUProgram, inputs: TensorInfo[], outputDtype: DataType, programDefin
tfjs-backend-webgpu/src/backend_webgpu.ts:850
↓ 98 callersMethodsquare
(this: T)
tfjs-core/src/public/chained_ops/square.ts:25
↓ 98 callersMethodtoString
* Returns a human-readable description of the tensor. Useful for logging. * * @doc {heading: 'Tensors', subheading: 'Classes'}
tfjs-core/src/tensor.ts:491
↓ 93 callersFunctioncreateNumberAttrFromIndex
(inputIndex: number)
tfjs-converter/src/operations/executors/test_helper.ts:25
↓ 93 callersMethodgetConfig
()
tfjs-layers/src/layers/merge.ts:1067
↓ 91 callersMethodvariable
(trainable = true, name?: string, dtype?: DataType)
tfjs-core/src/tensor.ts:500
↓ 90 callersFunctiondescribeMathCPU
(testName: string, tests: () => void)
tfjs-layers/src/utils/test_utils.ts:115
↓ 90 callersMethodreadSync
(dataId: DataId)
tfjs-core/src/tensor.ts:187
↓ 88 callersMethodappend
(callback: BaseCallback)
tfjs-layers/src/base_callbacks.ts:118
↓ 88 callersMethodwrite
(values: BackendValues, shape: number[], dtype: DataType)
tfjs-core/src/backends/backend.ts:36
↓ 87 callersFunctionmain
()
tfjs-backend-wasm/tools/cpplint.py:6223
↓ 86 callersFunctioncomputeDispatch
( layout: {x: number[], y?: number[], z?: number[]}, outputShape: number[], workgroupSize: [number, nu
tfjs-backend-webgpu/src/webgpu_util.ts:41
↓ 85 callersFunctionadd
(a: Tensor, b: Tensor)
tfjs-core/src/engine.ts:1311
next →1–100 of 6,022, ranked by callers