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
↓ 1 callersFunctioncheckLossAndTargetCompatibility
* Validation on the compatibility of targes and loss functions. * * This helps prevent users from using loss functions incorrectly. * * @param tar
tfjs-layers/src/engine/training.ts:230
↓ 1 callersMethodcheckManifestAndWeightFiles
* Check the compatibility between weights manifest and weight files.
tfjs-core/src/io/browser_files.ts:214
↓ 1 callersMethodcheckMapInputs
( inputTensorNames: string[], modelInputNames: string[])
tfjs-tflite/src/tflite_model.ts:279
↓ 1 callersFunctioncheckMasksShapes
( inputs: Tensor, paddingMask: Tensor, attentionMask: Tensor)
tfjs-layers/src/layers/nlp/modeling/transformer_layer_utils.ts:27
↓ 1 callersFunctioncheckNoneSymbolic
(tensors: MaybeSymbolic| MaybeSymbolic[])
tfjs-layers/src/engine/topology.ts:1673
↓ 1 callersFunctioncheckOutputTypeConversion
( originalOutputType: TFLiteDataType, expectedConvertedType: DataType)
tfjs-tflite/src/tflite_model_test.ts:128
↓ 1 callersFunctioncheckPromises
(promises: Array<Promise<T>>)
tfjs-core/src/io/progress.ts:50
↓ 1 callersFunctioncheckPublishable
(packageJsonPath: string)
scripts/release-util.ts:675
↓ 1 callersFunctioncheckShapesMatch
(x: HasShape, y: HasShape)
tfjs-layers/src/variables.ts:133
↓ 1 callersMethodcheckTensorForDisposalWithNodeLiveUntilInfo
( node: Node, tensorMap: NamedTensorsMap, context: ExecutionContext, tensorsToKeep: Set<number>, o
tfjs-converter/src/executor/graph_executor.ts:382
↓ 1 callersFunctioncheckWebGLError
(gl: WebGLRenderingContext)
tfjs-backend-webgl/src/webgl_util.ts:31
↓ 1 callersFunctionchmodRecursive
* Recursively change permissions of files to 775. * * @param rootPath The path where permissions are changed.
link-package/build_deps.ts:243
↓ 1 callersMethodchooseIndex
()
tfjs-data/src/iterators/lazy_iterator.ts:1169
↓ 1 callersFunctioncleanDeps
* Deletes the deps directory if it exists, and creates a fresh deps folder.
tfjs-node/scripts/install.js:128
↓ 1 callersMethodcleanUp
()
tfjs-tflite/src/tflite_model_test.ts:62
↓ 1 callersFunctionclearDisplay
()
tfjs-backend-webgpu/benchmarks/util.js:75
↓ 1 callersMethodcloneAndKeepTensor
(tensor: Tensor)
tfjs-converter/src/executor/graph_executor.ts:193
↓ 1 callersMethodclose
()
tfjs-data/src/util/test_utils.ts:117
↓ 1 callersFunctioncloseModal
()
tfjs-backend-webgpu/benchmarks/util.js:65
↓ 1 callersFunctioncollectInputShape
* Collects the input shape(s) of a list of `tf.Tensor`s or * `tf.SymbolicTensor`s. * * TODO(michaelterry): Update PyKeras docs (backport). * * @p
tfjs-layers/src/engine/topology.ts:1591
↓ 1 callersFunctioncombineLocations
( outputLoc: number[], reduceLoc: number[], axes: number[])
tfjs-core/src/ops/axis_util.ts:33
↓ 1 callersFunctioncomparer
(idx, asc)
tfjs-backend-webgpu/perf/tune.js:344
↓ 1 callersMethodcompute
()
tfjs-backend-cpu/src/kernels/RaggedTensorToTensor_impl.ts:316
↓ 1 callersMethodcompute
(data: Uint8Array[], splits: Int32Array)
tfjs-backend-cpu/src/kernels/StringNGrams_impl.ts:130
↓ 1 callersMethodcomputeAttention
* Applies Dot-product attention with query, key, value tensors. * * This function defines the computation inside `call` with projected * mult
tfjs-layers/src/layers/nlp/multihead_attention.ts:663
↓ 1 callersMethodcomputeAttentionMask
* Computes the attention mask. * * * The `query`'s mask is reshaped from [B, T] to [B, T, 1]. * * The `value`'s mask is reshaped from [B, S]
tfjs-layers/src/layers/nlp/multihead_attention.ts:805
↓ 1 callersFunctioncomputeBaseline
()
tfjs-data/demo/boston-housing/index.ts:157
↓ 1 callersMethodcomputeCausalMask
* Computes a causal mask (e.g., for masked self-attention layers). * * For example, if query and value both contain sequences of length 4, *
tfjs-layers/src/layers/nlp/multihead_attention.ts:856
↓ 1 callersFunctioncomputeConv3DInfo
( inShape: [number, number, number, number, number], filterShape: [number, number, number, number, num
tfjs-core/src/ops/conv_util.ts:276
↓ 1 callersFunctioncomputeFans
* Computes the number of input and output units for a weight shape. * @param shape Shape of weight. * @param dataFormat data format to use for convo
tfjs-layers/src/initializers.ts:264
↓ 1 callersFunctioncomputeFloat16ExponentTable
* Computes exponent table for casting Float16 to Float32 * See http://www.fox-toolkit.org/ftp/fasthalffloatconversion.pdf * * @returns Uint32Array,
tfjs-core/src/io/io_utils.ts:679
↓ 1 callersFunctioncomputeFloat16MantisaTable
* Computes mantisa table for casting Float16 to Float32 * See http://www.fox-toolkit.org/ftp/fasthalffloatconversion.pdf * * @returns Uint32Array,
tfjs-core/src/io/io_utils.ts:645
↓ 1 callersFunctioncomputeFloat16OffsetTable
* Computes offset table for casting Float16 to Float32 * See http://www.fox-toolkit.org/ftp/fasthalffloatconversion.pdf * * @returns Uint32Array, 6
tfjs-core/src/io/io_utils.ts:702
↓ 1 callersFunctioncomputeMaxSizePerColumn
( vals: TypedArray|string[], shape: number[], dtype: DataType, strides: number[])
tfjs-core/src/tensor_format.ts:46
↓ 1 callersFunctioncomputeOutAndReduceShapes
( aShape: number[], axes: number[])
tfjs-core/src/ops/axis_util.ts:49
↓ 1 callersFunctioncomputeOutShape
(inShape: number[], perm: number[])
tfjs-backend-wasm/src/kernels/Transpose.ts:81
↓ 1 callersMethodcomputeOutputShape
(inputShape: Shape|Shape[])
tfjs-layers/src/layers/embeddings.ts:152
↓ 1 callersMethodcomputeOutputShape
(inputShape: Shape|Shape[])
tfjs-layers/src/layers/wrappers.ts:318
↓ 1 callersFunctioncomputeOutputShape2D
( inShape: [number, number], fieldSize: number, stride: number, zeroPad?: number, roundingMode?: 'floo
tfjs-core/src/ops/conv_util.ts:350
↓ 1 callersFunctioncomputeOutputShape4D
( inShape: [number, number, number, number], filterShape: [number, number, number], outChannels: numbe
tfjs-core/src/ops/conv_util.ts:367
↓ 1 callersMethodcomputeSelfAttentionMask
( decoderSequence: Tensor, decoderPaddingMask: Tensor, decoderAttentionMask: Tensor, useCasual
tfjs-layers/src/layers/nlp/modeling/transformer_decoder.ts:440
↓ 1 callersFunctioncomputeTensors2D
( inputs: ConcatInputs, axis: number, backend: WebGPUBackend)
tfjs-backend-webgpu/src/kernels/Concat_impl.ts:140
↓ 1 callersFunctioncomputeTensors2D
( inputs: ConcatInputs, axis: number, backend: MathBackendWebGL)
tfjs-backend-webgl/src/kernels/Concat_impl.ts:148
↓ 1 callersFunctioncomputeWorkgroupInfoForMatMul
( dimAOuter: number, dimInner: number, dimBOuter: number, transposeA = false)
tfjs-backend-webgpu/src/webgpu_util.ts:67
↓ 1 callersFunctionconcat
( args: {inputs: ConcatInputs, attrs: ConcatAttrs, backend: WebGPUBackend})
tfjs-backend-webgpu/src/kernels/Concat.ts:25
↓ 1 callersFunctionconcat
( args: {inputs: ConcatInputs, attrs: ConcatAttrs, backend: MathBackendWebGL})
tfjs-backend-webgl/src/kernels/Concat.ts:24
↓ 1 callersFunctionconcat
( args: {inputs: ConcatInputs, backend: BackendWasm, attrs: ConcatAttrs})
tfjs-backend-wasm/src/kernels/Concat.ts:25
↓ 1 callersFunctionconcatImpl
( inputs: Array<{vals: BackendValues, shape: number[]}>, outShape: number[], dtype: DataType, simplyCo
tfjs-backend-cpu/src/kernels/Concat_impl.ts:20
↓ 1 callersFunctionconcatImpl
( inputs: ConcatInputs, axis: number, backend: WebGPUBackend)
tfjs-backend-webgpu/src/kernels/Concat_impl.ts:29
↓ 1 callersFunctionconcatImpl
( inputs: ConcatInputs, axis: number, backend: MathBackendWebGL)
tfjs-backend-webgl/src/kernels/Concat_impl.ts:32
↓ 1 callersFunctionconfig
({ plugins = [], output = {}, external = [], visualize = false, tsCompilerOptions = {} })
tfjs-backend-webgpu/rollup.config.js:41
↓ 1 callersMethodconnect
(node: {})
tfjs-data/src/util/test_utils.ts:129
↓ 1 callersMethodcontrol_flow
(self, x, y, z)
e2e/integration_tests/convert_predict.py:321
↓ 1 callersFunctionconv1d
(args: ConvLayerArgs)
tfjs-layers/src/exports_layers.ts:246
↓ 1 callersFunctionconv1d
( x: Tensor, kernel: Tensor, strides = 1, padding = 'valid', dataFormat?: DataFormat, dilationRate = 1
tfjs-layers/src/layers/convolutional.ts:141
↓ 1 callersFunctionconv2D
( args: {inputs: Conv2DInputs, backend: MathBackendCPU, attrs: Conv2DAttrs})
tfjs-backend-cpu/src/kernels/Conv2D.ts:23
↓ 1 callersFunctionconv2DBackpropFilterImpl
( dy: TensorInfo, filter: TensorInfo, convInfo: backend_util.Conv2DInfo, backend: NodeJSKernelBackend)
tfjs-node/src/kernels/Conv2DBackpropFilter.ts:41
↓ 1 callersFunctionconv2DBackpropInputImpl
( dy: TensorInfo, filter: TensorInfo, convInfo: backend_util.Conv2DInfo, backend: NodeJSKernelBackend)
tfjs-node/src/kernels/Conv2DBackpropInput.ts:40
↓ 1 callersFunctionconv2d
( x: Tensor, kernel: Tensor, strides = [1, 1], padding = 'valid', dataFormat?: DataFormat, dilationRat
tfjs-layers/src/layers/convolutional.ts:161
↓ 1 callersFunctionconv2dByMatMul
({ x, filter, convInfo, backend, bias = null, preluActivationWeights = null, leakyreluAlpha = 0,
tfjs-backend-webgpu/src/kernels/Conv2D_impl.ts:70
↓ 1 callersFunctionconv2dCommonSnippet
( isChannelsLast: boolean, fitAOuter: boolean, fitBOuter: boolean, fitInner: boolean, addBias = false,
tfjs-backend-webgpu/src/conv2d_mm_webgpu.ts:25
↓ 1 callersFunctionconv2dTranspose
(args: ConvLayerArgs)
tfjs-layers/src/exports_layers.ts:307
↓ 1 callersFunctionconv2dTransposeCommonSnippet
(innerElementSize = 4)
tfjs-backend-webgpu/src/conv_backprop_mm_webgpu.ts:24
↓ 1 callersFunctionconv2dWithIm2Col
({ x, filter, convInfo, backend, bias = null, preluActivationWeights = null, leakyreluAlpha = 0,
tfjs-backend-webgpu/src/kernels/Conv2D_impl.ts:175
↓ 1 callersFunctionconv3d
( x: Tensor, kernel: Tensor, strides = [1, 1, 1], padding = 'valid', dataFormat?: DataFormat, dilation
tfjs-layers/src/layers/convolutional.ts:228
↓ 1 callersFunctionconvert
(argv: string[])
tfjs-converter/tools/pb2json_converter.ts:35
↓ 1 callersFunctionconvert
(arguments)
tfjs-converter/python/tensorflowjs/converters/converter.py:883
↓ 1 callersFunctionconvertConv2DDataFormat
(dataFormat: 'NHWC'|'NCHW')
tfjs-core/src/ops/conv_util.ts:588
↓ 1 callersFunctionconvertDocStringToDocInfoObject
(docString: string)
tfjs-core/scripts/test_snippets/util.ts:179
↓ 1 callersFunctionconvertMantissa
(i: number)
tfjs-core/src/io/io_utils.ts:646
↓ 1 callersFunctionconvertNDArrayScalarsInConfig
* Replace ndarray-style scalar objects in serialization objects with numbers. * * Background: In some versions of tf.keras, certain scalar values in
tfjs-layers/src/utils/generic_utils.ts:160
↓ 1 callersMethodconvertTensorMapToTensorsMap
(map: NamedTensorMap)
tfjs-converter/src/executor/graph_model.ts:567
↓ 1 callersFunctionconvertToRGBA
( data: number[], shape: number[], dtype: string, alpha = 1)
tfjs-core/src/ops/draw_test.ts:42
↓ 1 callersFunctionconvertToRowMajor
(inputValues: number[][]| tf.Tensor2D)
tfjs-vis/src/render/heatmap.ts:255
↓ 1 callersMethodconvertValue
Handles proper rounding for non-floating-point numbers.
tfjs-core/src/ops/rand_util.ts:169
↓ 1 callersMethodconvertValue
(value: number)
tfjs-core/src/ops/rand_util.ts:207
↓ 1 callersFunctionconvert_tf_hub_module_v1
Freeze the TF-Hub module and check compatibility with Tensorflow.js. Optimize and convert the TF-Hub module to Tensorflow.js format, if it passes
tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2.py:1042
↓ 1 callersFunctionconvert_tf_saved_model
Freeze the SavedModel and check the model compatibility with Tensorflow.js. Optimize and convert the model to Tensorflow.js format, when the model
tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2.py:940
↓ 1 callersMethodcopyBuffer
(srcBuffer: GPUBuffer)
tfjs-backend-webgpu/src/backend_webgpu.ts:542
↓ 1 callersFunctioncopyRecursive
* Recursively copy a file or directory. * * @param src The source directory. * @param dest The destination to copy src to.
link-package/build_deps.ts:206
↓ 1 callersFunctioncountTrainableParams
(model: Container)
tfjs-layers/src/utils/layer_utils.ts:89
↓ 1 callersMethodcreateAnalyser
()
tfjs-data/src/util/test_utils.ts:113
↓ 1 callersMethodcreateAndWaitForFence
()
tfjs-backend-webgl/src/gpgpu_context.ts:282
↓ 1 callersMethodcreateCameraTexture
* Creates a WebGL texture that is updated by the underlying platform to * contain the contents of the camera.
tfjs-react-native/src/camera/camera_stream.tsx:221
↓ 1 callersFunctioncreateCanvas
(webGLVersion: number)
tfjs-backend-webgl/src/canvas_util.ts:72
↓ 1 callersFunctioncreateData
()
tfjs-layers/src/layers/merge_test.ts:426
↓ 1 callersMethodcreateFence
(gl: WebGLRenderingContext)
tfjs-backend-webgl/src/gpgpu_context.ts:287
↓ 1 callersMethodcreateFloat16MatrixTexture
(rows: number, columns: number)
tfjs-backend-webgl/src/gpgpu_context.ts:202
↓ 1 callersMethodcreateFloat16PackedMatrixTexture
(rows: number, columns: number)
tfjs-backend-webgl/src/gpgpu_context.ts:230
↓ 1 callersFunctioncreateHalfFloatTextureAndBindToFramebuffer
( // tslint:disable-next-line:no-any gl: WebGLRenderingContext, textureHalfFloatExtension: any)
tfjs-backend-webgl/src/webgl_util.ts:670
↓ 1 callersFunctioncreateIndices
* Computes the update indices by iterating through all indices from * `startIndices` to `startIndices + updates.shape`.
tfjs-layers/src/layers/nlp/utils.ts:52
↓ 1 callersFunctioncreateInputTensors
* Create a list of input tensors. * * @private * @param inputsData An array with each element being the value to * create a tensor. * @param i
tfjs-inference/src/index.ts:221
↓ 1 callersFunctioncreateInstantiateWasmFunc
(path: string)
tfjs-backend-wasm/src/backend_wasm.ts:238
↓ 1 callersFunctioncreateMask
()
tfjs-layers/src/layers/recurrent.ts:2117
↓ 1 callersMethodcreateMediaStreamSource
()
tfjs-data/src/util/test_utils.ts:109
↓ 1 callersFunctioncreateNamedTensorMap
* Create a NamedTensorMap from an output node name. * @param outputNodeName Output node name. * @param modelJson The parsed model.json. * @param du
e2e/benchmarks/local-benchmark/dump.js:112
↓ 1 callersFunctioncreateNestedArray
( offset: number, shape: number[], a: TypedArray, isComplex = false)
tfjs-core/src/util_base.ts:581
↓ 1 callersMethodcreateOrVerifyDirectory
* For each item in `this.path`, creates a directory at the path or verify * that the path exists as a directory.
tfjs-node/src/io/file_system.ts:206
↓ 1 callersMethodcreateSummaryFileWriter
( resourceHandle: Tensor, logdir: string, maxQueue?: number, flushMillis?: number, filenameSuffix?
tfjs-node/src/nodejs_kernel_backend.ts:517
↓ 1 callersFunctioncreateTab
(browserConf)
e2e/benchmarks/browserstack-benchmark/index.js:431
← previousnext →2,001–2,100 of 6,022, ranked by callers