Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/tensorflow/tfjs
/ functions
Functions
6,022 in github.com/tensorflow/tfjs
⨍
Functions
6,022
◇
Types & classes
1,932
↳
Endpoints
2
↓ 2 callers
Method
mapOutputs
(outputs: string[])
tfjs-converter/src/executor/graph_executor.ts:755
↓ 2 callers
Method
mapSignatureEntries
(entries: {[k: string]: tensorflow.ITensorInfo})
tfjs-converter/src/operations/operation_mapper.ts:171
↓ 2 callers
Function
mapTFDtypeToJSDtype
(tfDtype: string)
tfjs-node/src/saved_model.ts:439
↓ 2 callers
Method
maskedSoftmax
( attentionScores: Tensor, attentionMask?: Tensor )
tfjs-layers/src/layers/nlp/multihead_attention.ts:623
↓ 2 callers
Function
matMulReadFnSource
( transposeA: boolean, transposeB: boolean, fitAOuter = false, fitBOuter = false, fitInner = false, co
tfjs-backend-webgpu/src/matmul_packed_webgpu.ts:24
↓ 2 callers
Function
maxPool
( args: {inputs: MaxPoolInputs, backend: MathBackendCPU, attrs: MaxPoolAttrs})
tfjs-backend-cpu/src/kernels/MaxPool.ts:24
↓ 2 callers
Function
maxPoolPositions
( xValues: TypedArray, xShape: number[], dtype: DataType, convInfo: backend_util.Conv2DInfo, flattenPo
tfjs-backend-cpu/src/utils/pool_utils.ts:89
↓ 2 callers
Function
meanSquaredLogErrorFor1DArray
(x: number[], y: number[])
tfjs-layers/src/losses_test.ts:78
↓ 2 callers
Function
mergeBatchSizeAndShape
( batchSize: number, shape: Shape|{[name: string]: Shape})
tfjs-layers/src/engine/dataset_fakes.ts:59
↓ 2 callers
Method
move
( dataId: DataId, values: BackendValues, shape: number[], dtype: DataType, refCount: number)
tfjs-core/src/backends/backend.ts:37
↓ 2 callers
Function
newReadIterator
()
tfjs-data/src/iterators/lazy_iterator_test.ts:283
↓ 2 callers
Method
nextBatch
(batchSize, data, index)
tfjs-vis/demos/mnist_internals/data.js:123
↓ 2 callers
Method
nextBatch
(batchSize, data, index)
tfjs-vis/demos/mnist/data.js:123
↓ 2 callers
Method
nextDataId
()
tfjs-backend-webgpu/src/backend_webgpu.ts:132
↓ 2 callers
Method
nextTestBatch
(batchSize)
tfjs-vis/demos/mnist_internals/data.js:115
↓ 2 callers
Method
nextValue
Returns next sample from a Gaussian distribution.
tfjs-core/src/ops/rand_util.ts:69
↓ 2 callers
Method
normalizeInputs
(inputs: Tensor|Tensor[]| NamedTensorMap)
tfjs-converter/src/executor/graph_model.ts:410
↓ 2 callers
Method
normalizeOutputs
(outputs: string|string[])
tfjs-converter/src/executor/graph_model.ts:448
↓ 2 callers
Function
normalize_weight_name
Remove suffix ":0" (if present) from weight name.
tfjs-converter/python/tensorflowjs/converters/keras_h5_conversion.py:37
↓ 2 callers
Function
now
()
tfjs-core/src/util.ts:85
↓ 2 callers
Method
onBatchEnd
(batch: number, logs?: Logs)
tfjs-layers/src/engine/training_test.ts:1489
↓ 2 callers
Method
onEpochEnd
(epoch: number, logs?: UnresolvedLogs)
tfjs-layers/src/engine/training_test.ts:1462
↓ 2 callers
Function
ones
()
tfjs-layers/src/exports_initializers.ts:27
↓ 2 callers
Function
opNameToFileName
(opName: string)
tfjs/tools/custom_module/util.ts:37
↓ 2 callers
Function
output_path_exists
Check the existence of the output path. Args: output_path: input path of the model. Returns: bool: return true when the output directory e
tfjs-converter/python/tensorflowjs/converters/wizard.py:202
↓ 2 callers
Function
padCommon
(shape: number[], fillZero = false)
tfjs-backend-webgpu/src/pad_webgpu.ts:21
↓ 2 callers
Method
parseRow
(line: string, validateElementCount = true)
tfjs-data/src/datasets/csv_dataset.ts:292
↓ 2 callers
Function
parseStringParam
(s: []|string, keepCase: boolean)
tfjs-converter/src/operations/operation_mapper.ts:450
↓ 2 callers
Function
parseTensorShapeParam
(shape: tensorflow.ITensorShape)
tfjs-converter/src/operations/operation_mapper.ts:542
↓ 2 callers
Function
perClassAccuracy
( container: Drawable, classAccuracy: Array<{accuracy: number, count: number}>, classLabels?: stri
tfjs-vis/src/show/quality.ts:51
↓ 2 callers
Function
pool3d
( xValues: TypedArray, xShape: number[], dtype: DataType, strides: number[], convInfo: backend_util.Co
tfjs-backend-cpu/src/utils/pool_utils.ts:156
↓ 2 callers
Function
poolImpl
( x: TensorInfo, convInfo: backend_util.Conv2DInfo, poolType: PoolType, backend: WebGPUBackend)
tfjs-backend-webgpu/src/kernels/Pool_impl.ts:29
↓ 2 callers
Method
populateURLFlags
()
tfjs-core/src/environment.ts:161
↓ 2 callers
Function
predict
(inputData)
e2e/benchmarks/local-benchmark/tflite_worker.js:39
↓ 2 callers
Method
predictLoop
* Helper method to loop over some data in batches. * * Porting Note: Not using the functional approach in the Python equivalent * due to th
tfjs-layers/src/engine/training.ts:1033
↓ 2 callers
Method
predictOnBatch
* Returns predictions for a single batch of samples. * * @param x: Input samples, as a Tensor, or list of Tensors (if the model * has multi
tfjs-layers/src/models.ts:801
↓ 2 callers
Function
prepareInputs
()
tfjs-backend-webgl/src/kernels/FusedConv2D.ts:57
↓ 2 callers
Function
preprocessConv3DInput
( x: Tensor, dataFormat: DataFormat)
tfjs-layers/src/layers/convolutional.ts:58
↓ 2 callers
Function
printMemory
(bytes)
e2e/benchmarks/browserstack-benchmark/index.js:680
↓ 2 callers
Function
promiseCheckWrapper
(acturalValPromise: Promise<{}>, truthVal: {})
tfjs-core/src/profiler_test.ts:50
↓ 2 callers
Function
qr2d
(x: Tensor2D, fullMatrices = false)
tfjs-core/src/ops/linalg/qr.ts:120
↓ 2 callers
Function
quantizationTest
(quantizationDtype: 'uint8'|'uint16')
tfjs-core/src/io/weights_loader_test.ts:442
↓ 2 callers
Function
r2Score
(yTrue: Tensor, yPred: Tensor)
tfjs-layers/src/metrics.ts:113
↓ 2 callers
Function
range
(start: number, end: number)
tfjs-data/src/iterators/file_chunk_iterator_test.ts:22
↓ 2 callers
Method
read
(dataId: DataId)
tfjs-core/src/backends/backend.ts:33
↓ 2 callers
Method
readToGPU
(dataId: DataId, options?: DataToGPUOptions)
tfjs-core/src/tensor.ts:188
↓ 2 callers
Function
readToLength
(reader: ReadableStreamDefaultReader<ArrayBuffer>, initialData: ArrayBuffer,
tfjs-core/src/io/io_utils.ts:294
↓ 2 callers
Method
register
* Registers the class as serializable.
tfjs-core/src/serialization.ts:154
↓ 2 callers
Method
registerManager
* Register a save-handler router. * * @param saveRouter A function that maps a URL-like string onto an instance * of `IOHandler` with the `sa
tfjs-core/src/io/model_management.ts:59
↓ 2 callers
Method
releaseResource
(dataId: DataId)
tfjs-backend-webgpu/src/backend_webgpu.ts:224
↓ 2 callers
Function
removeClass
(selector: string, className: string)
tfjs-tflite/demo/src/script.ts:152
↓ 2 callers
Method
removeDataId
(dataId: DataId, backend: KernelBackend)
tfjs-core/src/engine.ts:910
↓ 2 callers
Function
removeItems
(keys: LocalStorageKeys)
tfjs-core/src/io/local_storage.ts:98
↓ 2 callers
Method
renderMainScreen
()
tfjs-react-native/integration_rn59/App.tsx:87
↓ 2 callers
Function
resizeBilinear
(args: { inputs: ResizeBilinearInputs, backend: MathBackendCPU, attrs: ResizeBilinearAttrs })
tfjs-backend-cpu/src/kernels/ResizeBilinear.ts:23
↓ 2 callers
Function
roundUpToPow2
(num: number)
tfjs-backend-webgpu/src/kernels/TopK.ts:35
↓ 2 callers
Function
roundUpToPow2
(num: number)
tfjs-backend-webgl/src/kernels/TopK.ts:35
↓ 2 callers
Function
run
(model: tf.Sequential)
tfjs-data/demo/boston-housing/index.ts:125
↓ 2 callers
Function
run
()
tfjs-backend-webgpu/perf/tune.js:299
↓ 2 callers
Function
runBenchmarkFromFile
* Runs a benchmark with a preconfigured file * * @param file Relative filepath to preset benchmark configuration * @param runBenchmark Function to
e2e/benchmarks/browserstack-benchmark/app.js:462
↓ 2 callers
Method
runKernelFunc
* Internal helper method to execute a kernel Func * * Use `runKernel` to execute kernels from outside of engine.
tfjs-core/src/engine.ts:595
↓ 2 callers
Function
runSwap
(dir: number, inc: number, shape: number[])
tfjs-backend-webgpu/src/kernels/TopK.ts:101
↓ 2 callers
Function
runSwap
(dir: number, inc: number, shape: number[])
tfjs-backend-webgl/src/kernels/TopK.ts:124
↓ 2 callers
Function
searchSorted
(args: { inputs: SearchSortedInputs, backend: MathBackendCPU, attrs: SearchSortedAttrs })
tfjs-backend-cpu/src/kernels/SearchSorted.ts:24
↓ 2 callers
Function
sendChatMsg
(url, msg)
scripts/cloud_funcs/sync_reactnative/index.js:61
↓ 2 callers
Method
serialNext
()
tfjs-data/src/iterators/lazy_iterator.ts:584
↓ 2 callers
Function
setBackend
(backendName: string)
tfjs-core/src/globals.ts:291
↓ 2 callers
Method
setColumnNames
()
tfjs-data/src/datasets/csv_dataset.ts:82
↓ 2 callers
Method
setFastWeightInitDuringBuild
(value: boolean)
tfjs-layers/src/layers/wrappers.ts:551
↓ 2 callers
Method
setFastWeightInitDuringBuild
(value: boolean)
tfjs-layers/src/layers/recurrent.ts:806
↓ 2 callers
Method
setModelInputFromTensor
( modelInput: TFLiteWebModelRunnerTensorInfo, tensor: Tensor)
tfjs-tflite/src/tflite_model.ts:189
↓ 2 callers
Method
setNumThreads
(value: number)
tfjs-tflite/src/types/image_segmenter.ts:34
↓ 2 callers
Method
setNumThreads
(value: number)
tfjs-tflite/src/types/object_detector.ts:31
↓ 2 callers
Method
setNumThreads
(value: number)
tfjs-tflite/src/types/image_classifier.ts:31
↓ 2 callers
Method
setOutputMatrixTextureDriver
( outputMatrixTextureMaybePacked: WebGLTexture, width: number, height: number)
tfjs-backend-webgl/src/gpgpu_context.ts:655
↓ 2 callers
Method
setResourceIdToCapturedInput
(outputs: Tensor[])
tfjs-converter/src/executor/graph_model.ts:477
↓ 2 callers
Method
setTabs
(surfaceList: SurfaceInfoStrict[])
tfjs-vis/src/components/visor.tsx:156
↓ 2 callers
Method
setTimeoutCustom
(functionRef: Function, delay: number)
tfjs-core/src/platforms/platform.ts:50
↓ 2 callers
Method
setUp
(self)
tfjs-converter/python/tensorflowjs/converters/converter_test.py:42
↓ 2 callers
Function
setWebGLContext
( webGLVersion: number, gl: WebGLRenderingContext)
tfjs-backend-webgl/src/canvas_util.ts:36
↓ 2 callers
Method
setWeights
* Sets the weights of the model. * * @param weights Should be a list of Tensors with shapes and types matching * the output of `model.getWe
tfjs-layers/src/models.ts:679
↓ 2 callers
Function
setupFakeWeightFiles
(fileBufferMap: { [filename: string]: string|Float32Array|Int32Array|ArrayBuffer|Uint8Array| Uint16Arr
tfjs-node/src/io/node_http_test.ts:66
↓ 2 callers
Function
showBackendFlagSettingsAndReturnTunableFlagControllers
* Show flag settings for the given backend under the UI element of * `folderController`. * * @param {dat.gui.GUI} folderController * @param {strin
e2e/benchmarks/local-benchmark/index.js:109
↓ 2 callers
Function
showDiff
(newSize, masterSize)
scripts/bundle-size-util.js:26
↓ 2 callers
Function
sparseSegmentReduce
( input: TensorInfo, indices: TensorInfo, segmentIds: TensorInfo, isSum = false, backend: WebGPUBacken
tfjs-backend-webgpu/src/kernel_utils/sparse_segment_reduce.ts:25
↓ 2 callers
Function
sparseSegmentReduction
( args: { backend: BackendWasm, inputs: SparseSegmentSumInputs|SparseSegmentMeanInputs, },
tfjs-backend-wasm/src/kernels/SparseSegmentReduction.ts:44
↓ 2 callers
Function
sparseSegmentReductionImpl
( input: TypedArray, inputShape: number[], inputDType: DataType, indices: TypedArray, segmentIds: Type
tfjs-backend-cpu/src/kernels/SparseSegmentReduction_impl.ts:20
↓ 2 callers
Function
square_if_positive
(v)
e2e/integration_tests/convert_predict.py:208
↓ 2 callers
Function
standardizeArgs
( inputs: Tensor|Tensor[]|SymbolicTensor|SymbolicTensor[], initialState: Tensor|Tensor[]|SymbolicTenso
tfjs-layers/src/layers/recurrent.ts:59
↓ 2 callers
Function
standardizeCallbacks
( callbacks: BaseCallback|BaseCallback[]|CustomCallbackArgs| CustomCallbackArgs[], yieldEvery: Yie
tfjs-layers/src/base_callbacks.ts:476
↓ 2 callers
Function
standardizeDataIteratorOutput
* Standardize the output of a dataset iterator for use by * LayersModel.fitDataset(). * * @param model: A `tf.LayersModel` object. * @param iterat
tfjs-layers/src/engine/training_dataset.ts:202
↓ 2 callers
Function
standardizeSampleOrClassWeights
( xWeight: ClassWeight|ClassWeight[]|ClassWeightMap, outputNames: string[], weightType: 'sampleWeight'
tfjs-layers/src/engine/training_utils.ts:34
↓ 2 callers
Method
standardizeUserDataXY
( x: Tensor|Tensor[]|{[inputName: string]: Tensor}, y: Tensor|Tensor[]|{[inputName: string]: Tenso
tfjs-layers/src/engine/training.ts:1145
↓ 2 callers
Function
start
()
tfjs-tflite/demo/src/script.ts:30
↓ 2 callers
Function
stridesOrDilationsArePositive
(values: number| number[])
tfjs-core/src/ops/conv_util.ts:576
↓ 2 callers
Function
stringLiteralArray
(a: T[])
tfjs-layers/src/keras_format/utils.ts:26
↓ 2 callers
Method
stylize
(contentImage: string, styleImage: string)
tfjs-react-native/integration_rn59/components/webcam/webcam_demo.tsx:127
↓ 2 callers
Function
summaryFileWriter
( logdir: string, maxQueue = 10, flushMillis = 120000, filenameSuffix = '.v2')
tfjs-node/src/tensorboard.ts:122
↓ 2 callers
Method
surfaceId
(label: string, tab: string)
tfjs-vis/src/components/visor.tsx:152
↓ 2 callers
Method
syncData
* Await the values of all losses and metrics.
tfjs-layers/src/base_callbacks.ts:316
← previous
next →
1,601–1,700 of 6,022, ranked by callers