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
↓ 2 callersFunctionaddResultToFirestore
* After being returned from Browserstack, benchmark results are stored as * a list of fulfilled promises. * * As results are being iterated through
e2e/benchmarks/browserstack-benchmark/firestore.js:62
↓ 2 callersMethodaddStructuredOutputNames
(outputTensors: Tensor|Tensor[])
tfjs-converter/src/executor/graph_model.ts:303
↓ 2 callersMethodaddTapeNode
( kernelName: string, inputs: NamedTensorMap, outputs: Tensor[], gradientsFunc: GradFunc, saved: T
tfjs-core/src/engine.ts:1004
↓ 2 callersFunctionaggregateKernelTime
* Aggregate kernels by name and sort the array in non-ascending order of time. * Return an array of objects with `name` and `timeMs` fields. * * @p
e2e/benchmarks/benchmark_util.js:519
↓ 2 callersFunctionalignInputWithDataFormat
(input: TensorInfo, dataFormat: 'NHWC'|'NCHW')
tfjs-backend-webgl/src/kernels/FusedConv2D.ts:68
↓ 2 callersFunctionanalyzeEinsumString
( equation: string, biasAxes: string, inputShape: Shape, outputShape: Shape )
tfjs-layers/src/layers/nlp/einsum_dense.ts:39
↓ 2 callersMethodapplyGradients
(variableGradients: NamedTensorMap|NamedTensor[])
tfjs-core/src/optimizers/sgd_optimizer.ts:45
↓ 2 callersFunctionareClose
( a, e, epsilon, epsilonOfBigNumber = 0.1, relativeEpsilon = 0.01)
e2e/benchmarks/local-benchmark/util.js:76
↓ 2 callersFunctionargMax
( args: {inputs: ArgMaxInputs, backend: MathBackendCPU, attrs: ArgMaxAttrs})
tfjs-backend-cpu/src/kernels/ArgMax.ts:24
↓ 2 callersFunctionargMin
( args: {inputs: ArgMinInputs, backend: MathBackendCPU, attrs: ArgMinAttrs})
tfjs-backend-cpu/src/kernels/ArgMin.ts:24
↓ 2 callersFunctionargMinMaxReduce
( backend: MathBackendWebGL, x: TensorInfo, axis: number, reduceType: 'min'|'max')
tfjs-backend-webgl/src/kernel_utils/arg_min_max.ts:71
↓ 2 callersFunctionarray
(items: T[])
tfjs-data/src/dataset.ts:581
↓ 2 callersFunctionarrayConcat
(arrays: number[][])
tfjs-core/src/gradients/GatherV2_grad.ts:93
↓ 2 callersFunctionarrayRange
(start: number, stop: number)
tfjs-core/src/gradients/GatherV2_grad.ts:85
↓ 2 callersFunctionassertDtype
( expectedDtype: DataType|'numeric'|'string_or_numeric', actualDType: DataType, argName: string, funct
tfjs-core/src/tensor_util_env.ts:81
↓ 2 callersFunctionassertLabelsMatchShape
( inputValues: number[][]|tf.Tensor2D, labels: string[], dimension: 0|1)
tfjs-vis/src/render/heatmap.ts:285
↓ 2 callersFunctionassertNotComplex
( tensor: TensorInfo|TensorInfo[], opName: string)
tfjs-backend-webgpu/src/webgpu_util.ts:162
↓ 2 callersFunctionaveragePooling1d
(args: Pooling1DLayerArgs)
tfjs-layers/src/exports_layers.ts:945
↓ 2 callersFunctionaveragePooling2d
(args: Pooling2DLayerArgs)
tfjs-layers/src/exports_layers.ts:980
↓ 2 callersFunctionaveragePooling3d
(args: Pooling3DLayerArgs)
tfjs-layers/src/exports_layers.ts:1013
↓ 2 callersFunctionavgPool
( args: {inputs: AvgPoolInputs, backend: MathBackendCPU, attrs: AvgPoolAttrs})
tfjs-backend-cpu/src/kernels/AvgPool.ts:24
↓ 2 callersFunctionaxesAreInnerMostDims
(axes: number[], rank: number)
tfjs-core/src/ops/axis_util.ts:24
↓ 2 callersMethodbackendNames
()
tfjs-core/src/engine.ts:231
↓ 2 callersFunctionbase64ImageToTensor
(base64: string)
tfjs-react-native/integration_rn59/components/webcam/image_utils.ts:43
↓ 2 callersMethodbeginQuery
()
tfjs-backend-webgl/src/gpgpu_context.ts:498
↓ 2 callersFunctionbincountImpl
( xVals: TypedArray, weightsVals: TypedArray, weightsDtype: DataType, weightsShape: number[], size: nu
tfjs-backend-cpu/src/kernels/Bincount_impl.ts:20
↓ 2 callersMethodbindTextureToFrameBuffer
(texture: WebGLTexture)
tfjs-backend-webgl/src/gpgpu_context.ts:624
↓ 2 callersFunctionboolNumToString
(v: number)
tfjs-core/src/tensor_format.ts:86
↓ 2 callersMethodbufferSync
* Returns a `tf.TensorBuffer` that holds the underlying data. * @doc {heading: 'Tensors', subheading: 'Classes'}
tfjs-core/src/tensor.ts:316
↓ 2 callersMethodbuild
(inputShape: Shape|Shape[])
tfjs-layers/src/layers/wrappers.ts:497
↓ 2 callersMethodbuild
(inputShape: Shape)
tfjs-layers/src/layers/nlp/modeling/position_embedding.ts:113
↓ 2 callersMethodbuildVao
(program: GPGPUContextProgram)
tfjs-backend-webgl/src/gpgpu_context.ts:350
↓ 2 callersFunctionc
()
tfjs-core/src/debug_mode_test.ts:60
↓ 2 callersMethodcallAndReturnPaddingMask
( inputs: Tensor|Tensor[], kwargs: GPT2PreprocessorOptions )
tfjs-layers/src/layers/nlp/models/gpt2/gpt2_preprocessor.ts:167
↓ 2 callersFunctioncanonical
( x: number, c: number, strideI: number, dimI: number, masks: number[], validRange: number[])
tfjs-core/src/ops/slice_util.ts:710
↓ 2 callersMethodcast
(dtype: DataType)
tfjs-core/src/public/chained_ops/cast.ts:25
↓ 2 callersFunctioncheckInputData
* Check inputs provided by the user. * * Porting Note: This corresponds to _standardize_input_data() in Python * Keras. Because of the strong typ
tfjs-layers/src/engine/training.ts:297
↓ 2 callersMethodcheckInputShapeAndType
(inputs: NamedTensorMap)
tfjs-converter/src/executor/graph_executor.ts:704
↓ 2 callersMethodcheckInputs
(inputs: NamedTensorMap)
tfjs-converter/src/executor/graph_executor.ts:743
↓ 2 callersMethodcheckKernelForMemLeak
( kernelName: string, numDataIdsBefore: number, outInfos: TensorInfo[])
tfjs-core/src/engine.ts:561
↓ 2 callersMethodcheckKeyAndValueTensor
(key: Tensor, value: Tensor)
tfjs-converter/src/executor/hash_table.ts:152
↓ 2 callersMethodcheckNumericalProblems
(values: BackendValues)
tfjs-backend-webgl/src/backend_webgl.ts:490
↓ 2 callersMethodcheckOutputs
(outputs: string[])
tfjs-converter/src/executor/graph_executor.ts:765
↓ 2 callersMethodcheckShape
(layer: Layer)
tfjs-layers/src/models.ts:421
↓ 2 callersMethodcheckTensorForDisposal
( nodeName: string, node: Node, tensorMap: NamedTensorsMap, context: ExecutionContext, tensorsToKe
tfjs-converter/src/executor/graph_executor.ts:329
↓ 2 callersMethodcheckTrainableWeightsConsistency
* Check trainable weights count consistency. * * This will raise a warning if `this.trainableWeights` and * `this.collectedTrainableWeights`
tfjs-layers/src/engine/training.ts:796
↓ 2 callersFunctioncheckUserDefinedMetadata
( userDefinedMetadata: {}, modelName: string, checkSize = false)
tfjs-layers/src/user_defined_metadata.ts:30
↓ 2 callersFunctionclipByValue
(args: { inputs: ClipByValueInputs, backend: MathBackendWebGL, attrs: ClipByValueAttrs })
tfjs-backend-webgl/src/kernels/ClipByValue.ts:24
↓ 2 callersMethodclone
* This method is called instead of the public-facing tensor.clone() when * saving a tensor for backwards pass. It makes sure to add the clone *
tfjs-core/src/engine.ts:505
↓ 2 callersFunctioncloneModelInternal
( sourceURL: string, destURL: string, deleteSource = false)
tfjs-core/src/io/model_management.ts:107
↓ 2 callersMethodcloneTensorMap
(tensorsMap: NamedTensorsMap)
tfjs-converter/src/executor/graph_executor.ts:215
↓ 2 callersFunctioncompareData
(data1, data2, level = DUMP_LEVEL.BIGDIFF)
e2e/benchmarks/local-benchmark/dump.js:27
↓ 2 callersMethodcomputeBytes
(shape: [number, number], dtype: DataType)
tfjs-backend-webgl/src/backend_webgl.ts:1218
↓ 2 callersFunctioncomputeConv2DInfo
( inShape: [number, number, number, number], filterShape: [number, number, number, number], stride
tfjs-core/src/ops/conv_util.ts:177
↓ 2 callersFunctioncomputeDatasetStatistics
( dataset: Dataset<TabularRecord>)
tfjs-data/demo/boston-housing/stats.ts:54
↓ 2 callersFunctioncomputeDefaultPad
( inputShape: [number, number]|[number, number, number, number], fieldSize: number, stride: number, di
tfjs-core/src/ops/conv_util.ts:387
↓ 2 callersFunctioncomputeFlatOuterDims
(orig: number[], numOutDims: number)
tfjs-backend-cpu/src/kernels/RaggedGather_impl.ts:140
↓ 2 callersMethodcomputeOutputShape
* * @param inputShapes A list of [queryShape, valueShape] or * [queryShape, valueShape, keyShape]. If no keyShape provided, valueShape *
tfjs-layers/src/layers/nlp/multihead_attention.ts:871
↓ 2 callersFunctioncomputeWorkPerThreadForConv2d
( layout: {x: number[], y?: number[], z?: number[]}, outputShape: number[], isVec4 = false)
tfjs-backend-webgpu/src/webgpu_util.ts:120
↓ 2 callersFunctioncomputeWorkgroupSizeForConv2d
( layout: {x: number[], y?: number[], z?: number[]}, outputShape: number[], isVec4 = false)
tfjs-backend-webgpu/src/webgpu_util.ts:93
↓ 2 callersFunctionconfigureCallbacks
( callbacks: BaseCallback[], verbose: ModelLoggingVerbosity, epochs: number, initialEpoch: number, num
tfjs-layers/src/base_callbacks.ts:579
↓ 2 callersFunctionconv2DImpl
({ x, filter, convInfo, backend, bias = null, preluActivationWeights = null, leakyreluAlpha = 0,
tfjs-backend-webgpu/src/kernels/Conv2D_impl.ts:275
↓ 2 callersFunctionconv2dByMatMul
({ x, filter, convInfo, backend, bias = null, preluActivationWeights = null, leakyreluAlpha = 0,
tfjs-backend-webgl/src/kernels/Conv2D_impl.ts:79
↓ 2 callersFunctionconv2dImpl
( x: TensorInfo, filter: TensorInfo, convInfo: backend_util.Conv2DInfo, backend: NodeJSKernelBackend)
tfjs-node/src/kernels/Conv2D.ts:41
↓ 2 callersFunctionconv2dWithIm2Row
({ x, filter, convInfo, backend, bias = null, preluActivationWeights = null, leakyreluAlpha = 0,
tfjs-backend-webgl/src/kernels/Conv2D_impl.ts:243
↓ 2 callersFunctionconv3DBackpropInput
(args: { inputs: Conv3DBackpropInputV2Inputs, attrs: Conv3DBackpropInputV2Attrs, backend: MathBackendWeb
tfjs-backend-webgl/src/kernels/Conv3DBackpropInputV2.ts:23
↓ 2 callersFunctionconvertProtoClassesToClasses
(protoClasses: ProtoClass[])
tfjs-tflite/src/tflite_task_library_client/common.ts:45
↓ 2 callersMethodconvertTFLiteTensorInfos
(infos: TFLiteWebModelRunnerTensorInfo[])
tfjs-tflite/src/tflite_model.ts:267
↓ 2 callersMethodconvertValue
Handles proper rounding for non-floating-point numbers.
tfjs-core/src/ops/rand_util.ts:102
↓ 2 callersFunctioncopyArray
(dst: TypedArray, src: TypedArray, size: number)
tfjs-backend-cpu/src/kernels/RaggedTensorToTensor_impl.ts:444
↓ 2 callersFunctioncosineWindow
( windowLength: number, a: number, b: number)
tfjs-core/src/ops/signal_ops_util.ts:26
↓ 2 callersFunctioncount
(array: T[], refernce: T)
tfjs-layers/src/utils/generic_utils.ts:48
↓ 2 callersFunctioncreateArgMinMaxKernelConfig
(kernelName: 'ArgMin'| 'ArgMax')
tfjs-backend-wasm/src/kernels/argminmax_kernel.ts:24
↓ 2 callersFunctioncreateBooleanArrayAttrFromIndex
(inputIndex: number)
tfjs-converter/src/operations/executors/test_helper.ts:58
↓ 2 callersFunctioncreateDenseModelWithTwoOutputsAndData
()
tfjs-layers/src/engine/training_test.ts:602
↓ 2 callersFunctioncreateFrameBuffer
(gl: WebGL2RenderingContext)
tfjs-react-native/src/camera/camera_webgl_util.ts:300
↓ 2 callersFunctioncreateLayerAndData
( mergeMode: BidirectionalMergeMode, returnState: boolean)
tfjs-layers/src/layers/wrappers_test.ts:392
↓ 2 callersMethodcreateNGrams
( data: Uint8Array[], splitIndex: number, output: Uint8Array[], outputStartIndex: number, numNGram
tfjs-backend-cpu/src/kernels/StringNGrams_impl.ts:58
↓ 2 callersFunctioncreateProgramObjects
( gl: WebGL2RenderingContext, vertexShaderSource: string, fragmentShaderSource: string, vertices: Floa
tfjs-react-native/src/camera/camera_webgl_util.ts:367
↓ 2 callersFunctioncreateSimpleOneLayerContainer
(useRegularizers: boolean)
tfjs-layers/src/engine/container_test.ts:421
↓ 2 callersFunctioncreateStagingGPUBufferFromData
( device: GPUDevice, data: number[], dtype: tf.DataType)
tfjs-backend-webgpu/src/backend_webgpu_test.ts:486
↓ 2 callersFunctioncreateTensorFromGPUTest
(zeroCopy = false)
tfjs-backend-webgpu/src/backend_webgpu_test.ts:562
↓ 2 callersMethodcreate_frozen_model
(self)
tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2_test.py:374
↓ 2 callersFunctioncumImpl
( op: CumOpType, x: TensorInfo, backend: WebGPUBackend, axis: number, exclusive: boolean, reverse: boo
tfjs-backend-webgpu/src/kernels/Cum_impl.ts:26
↓ 2 callersFunctioncumImpl
( op: CumOpType, x: TensorInfo, backend: MathBackendWebGL, axis: number, exclusive: boolean, reverse:
tfjs-backend-webgl/src/kernels/Cum_impl.ts:26
↓ 2 callersFunctioncumsum
( args: {inputs: CumsumInputs, backend: MathBackendCPU, attrs: CumsumAttrs})
tfjs-backend-cpu/src/kernels/Cumsum.ts:24
↓ 2 callersMethoddatabaseAction
* Perform database action to put model artifacts into or read model artifacts * from IndexedDB object store. * * Whether the action is put or
tfjs-core/src/io/indexed_db.ts:130
↓ 2 callersMethoddecodeBmp
(contents: Uint8Array, channels: number)
tfjs-node/src/nodejs_kernel_backend.ts:375
↓ 2 callersFunctiondecodeString
(bytes: Uint8Array, encoding = 'utf-8')
tfjs-core/src/util.ts:131
↓ 2 callersFunctiondecodeWeight
( spec: WeightsManifestEntry, byteBuffer: ArrayBuffer)
tfjs-core/src/io/io_utils.ts:185
↓ 2 callersFunctiondefer
(f: () => T)
tfjs-core/src/io/browser_files.ts:35
↓ 2 callersFunctiondelay
(ms: number)
scripts/publish-npm.ts:124
↓ 2 callersFunctiondeleteDatabase
()
tfjs-core/src/io/indexed_db.ts:42
↓ 2 callersMethoddeleteSavedModel
(savedModelId: number)
tfjs-node/src/tfjs_binding.ts:56
↓ 2 callersFunctiondenseBincount
(args: { inputs: DenseBincountInputs, backend: MathBackendCPU, attrs: DenseBincountAttrs })
tfjs-backend-cpu/src/kernels/DenseBincount.ts:23
↓ 2 callersFunctiondepthwiseConv2d
(args: DepthwiseConv2DLayerArgs)
tfjs-layers/src/exports_layers.ts:445
↓ 2 callersFunctiondepthwiseConv2dNativeBackpropFilter
(args: { inputs: DepthwiseConv2dNativeBackpropFilterInputs, backend: MathBackendCPU, attrs: DepthwiseCon
tfjs-backend-cpu/src/kernels/DepthwiseConv2dNativeBackpropFilter.ts:23
↓ 2 callersFunctiondepthwiseConv2dNativeBackpropInput
(args: { inputs: DepthwiseConv2dNativeBackpropInputInputs, backend: MathBackendCPU, attrs: DepthwiseConv
tfjs-backend-cpu/src/kernels/DepthwiseConv2dNativeBackpropInput.ts:23
↓ 2 callersFunctiondepthwiseConv2dNativeImpl
( input: TensorInfo, filter: TensorInfo, convInfo: backend_util.Conv2DInfo, backend: NodeJSKernelBacke
tfjs-node/src/kernels/DepthwiseConv2dNative.ts:45
↓ 2 callersFunctionderXBatch
(x: Tensor, indices: Tensor, dy: Tensor)
tfjs-core/src/gradients/GatherV2_grad.ts:37
← previousnext →1,301–1,400 of 6,022, ranked by callers