MCPcopy Create free account

hub / github.com/tensorflow/tfjs / functions

Functions6,022 in github.com/tensorflow/tfjs

↓ 2 callersMethodmapOutputs
(outputs: string[])
tfjs-converter/src/executor/graph_executor.ts:755
↓ 2 callersMethodmapSignatureEntries
(entries: {[k: string]: tensorflow.ITensorInfo})
tfjs-converter/src/operations/operation_mapper.ts:171
↓ 2 callersFunctionmapTFDtypeToJSDtype
(tfDtype: string)
tfjs-node/src/saved_model.ts:439
↓ 2 callersMethodmaskedSoftmax
( attentionScores: Tensor, attentionMask?: Tensor )
tfjs-layers/src/layers/nlp/multihead_attention.ts:623
↓ 2 callersFunctionmatMulReadFnSource
( transposeA: boolean, transposeB: boolean, fitAOuter = false, fitBOuter = false, fitInner = false, co
tfjs-backend-webgpu/src/matmul_packed_webgpu.ts:24
↓ 2 callersFunctionmaxPool
( args: {inputs: MaxPoolInputs, backend: MathBackendCPU, attrs: MaxPoolAttrs})
tfjs-backend-cpu/src/kernels/MaxPool.ts:24
↓ 2 callersFunctionmaxPoolPositions
( xValues: TypedArray, xShape: number[], dtype: DataType, convInfo: backend_util.Conv2DInfo, flattenPo
tfjs-backend-cpu/src/utils/pool_utils.ts:89
↓ 2 callersFunctionmeanSquaredLogErrorFor1DArray
(x: number[], y: number[])
tfjs-layers/src/losses_test.ts:78
↓ 2 callersFunctionmergeBatchSizeAndShape
( batchSize: number, shape: Shape|{[name: string]: Shape})
tfjs-layers/src/engine/dataset_fakes.ts:59
↓ 2 callersMethodmove
( dataId: DataId, values: BackendValues, shape: number[], dtype: DataType, refCount: number)
tfjs-core/src/backends/backend.ts:37
↓ 2 callersFunctionnewReadIterator
()
tfjs-data/src/iterators/lazy_iterator_test.ts:283
↓ 2 callersMethodnextBatch
(batchSize, data, index)
tfjs-vis/demos/mnist_internals/data.js:123
↓ 2 callersMethodnextBatch
(batchSize, data, index)
tfjs-vis/demos/mnist/data.js:123
↓ 2 callersMethodnextDataId
()
tfjs-backend-webgpu/src/backend_webgpu.ts:132
↓ 2 callersMethodnextTestBatch
(batchSize)
tfjs-vis/demos/mnist_internals/data.js:115
↓ 2 callersMethodnextValue
Returns next sample from a Gaussian distribution.
tfjs-core/src/ops/rand_util.ts:69
↓ 2 callersMethodnormalizeInputs
(inputs: Tensor|Tensor[]| NamedTensorMap)
tfjs-converter/src/executor/graph_model.ts:410
↓ 2 callersMethodnormalizeOutputs
(outputs: string|string[])
tfjs-converter/src/executor/graph_model.ts:448
↓ 2 callersFunctionnormalize_weight_name
Remove suffix ":0" (if present) from weight name.
tfjs-converter/python/tensorflowjs/converters/keras_h5_conversion.py:37
↓ 2 callersFunctionnow
()
tfjs-core/src/util.ts:85
↓ 2 callersMethodonBatchEnd
(batch: number, logs?: Logs)
tfjs-layers/src/engine/training_test.ts:1489
↓ 2 callersMethodonEpochEnd
(epoch: number, logs?: UnresolvedLogs)
tfjs-layers/src/engine/training_test.ts:1462
↓ 2 callersFunctionones
()
tfjs-layers/src/exports_initializers.ts:27
↓ 2 callersFunctionopNameToFileName
(opName: string)
tfjs/tools/custom_module/util.ts:37
↓ 2 callersFunctionoutput_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 callersFunctionpadCommon
(shape: number[], fillZero = false)
tfjs-backend-webgpu/src/pad_webgpu.ts:21
↓ 2 callersMethodparseRow
(line: string, validateElementCount = true)
tfjs-data/src/datasets/csv_dataset.ts:292
↓ 2 callersFunctionparseStringParam
(s: []|string, keepCase: boolean)
tfjs-converter/src/operations/operation_mapper.ts:450
↓ 2 callersFunctionparseTensorShapeParam
(shape: tensorflow.ITensorShape)
tfjs-converter/src/operations/operation_mapper.ts:542
↓ 2 callersFunctionperClassAccuracy
( container: Drawable, classAccuracy: Array<{accuracy: number, count: number}>, classLabels?: stri
tfjs-vis/src/show/quality.ts:51
↓ 2 callersFunctionpool3d
( xValues: TypedArray, xShape: number[], dtype: DataType, strides: number[], convInfo: backend_util.Co
tfjs-backend-cpu/src/utils/pool_utils.ts:156
↓ 2 callersFunctionpoolImpl
( x: TensorInfo, convInfo: backend_util.Conv2DInfo, poolType: PoolType, backend: WebGPUBackend)
tfjs-backend-webgpu/src/kernels/Pool_impl.ts:29
↓ 2 callersMethodpopulateURLFlags
()
tfjs-core/src/environment.ts:161
↓ 2 callersFunctionpredict
(inputData)
e2e/benchmarks/local-benchmark/tflite_worker.js:39
↓ 2 callersMethodpredictLoop
* 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 callersMethodpredictOnBatch
* 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 callersFunctionprepareInputs
()
tfjs-backend-webgl/src/kernels/FusedConv2D.ts:57
↓ 2 callersFunctionpreprocessConv3DInput
( x: Tensor, dataFormat: DataFormat)
tfjs-layers/src/layers/convolutional.ts:58
↓ 2 callersFunctionprintMemory
(bytes)
e2e/benchmarks/browserstack-benchmark/index.js:680
↓ 2 callersFunctionpromiseCheckWrapper
(acturalValPromise: Promise<{}>, truthVal: {})
tfjs-core/src/profiler_test.ts:50
↓ 2 callersFunctionqr2d
(x: Tensor2D, fullMatrices = false)
tfjs-core/src/ops/linalg/qr.ts:120
↓ 2 callersFunctionquantizationTest
(quantizationDtype: 'uint8'|'uint16')
tfjs-core/src/io/weights_loader_test.ts:442
↓ 2 callersFunctionr2Score
(yTrue: Tensor, yPred: Tensor)
tfjs-layers/src/metrics.ts:113
↓ 2 callersFunctionrange
(start: number, end: number)
tfjs-data/src/iterators/file_chunk_iterator_test.ts:22
↓ 2 callersMethodread
(dataId: DataId)
tfjs-core/src/backends/backend.ts:33
↓ 2 callersMethodreadToGPU
(dataId: DataId, options?: DataToGPUOptions)
tfjs-core/src/tensor.ts:188
↓ 2 callersFunctionreadToLength
(reader: ReadableStreamDefaultReader<ArrayBuffer>, initialData: ArrayBuffer,
tfjs-core/src/io/io_utils.ts:294
↓ 2 callersMethodregister
* Registers the class as serializable.
tfjs-core/src/serialization.ts:154
↓ 2 callersMethodregisterManager
* 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 callersMethodreleaseResource
(dataId: DataId)
tfjs-backend-webgpu/src/backend_webgpu.ts:224
↓ 2 callersFunctionremoveClass
(selector: string, className: string)
tfjs-tflite/demo/src/script.ts:152
↓ 2 callersMethodremoveDataId
(dataId: DataId, backend: KernelBackend)
tfjs-core/src/engine.ts:910
↓ 2 callersFunctionremoveItems
(keys: LocalStorageKeys)
tfjs-core/src/io/local_storage.ts:98
↓ 2 callersMethodrenderMainScreen
()
tfjs-react-native/integration_rn59/App.tsx:87
↓ 2 callersFunctionresizeBilinear
(args: { inputs: ResizeBilinearInputs, backend: MathBackendCPU, attrs: ResizeBilinearAttrs })
tfjs-backend-cpu/src/kernels/ResizeBilinear.ts:23
↓ 2 callersFunctionroundUpToPow2
(num: number)
tfjs-backend-webgpu/src/kernels/TopK.ts:35
↓ 2 callersFunctionroundUpToPow2
(num: number)
tfjs-backend-webgl/src/kernels/TopK.ts:35
↓ 2 callersFunctionrun
(model: tf.Sequential)
tfjs-data/demo/boston-housing/index.ts:125
↓ 2 callersFunctionrun
()
tfjs-backend-webgpu/perf/tune.js:299
↓ 2 callersFunctionrunBenchmarkFromFile
* 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 callersMethodrunKernelFunc
* Internal helper method to execute a kernel Func * * Use `runKernel` to execute kernels from outside of engine.
tfjs-core/src/engine.ts:595
↓ 2 callersFunctionrunSwap
(dir: number, inc: number, shape: number[])
tfjs-backend-webgpu/src/kernels/TopK.ts:101
↓ 2 callersFunctionrunSwap
(dir: number, inc: number, shape: number[])
tfjs-backend-webgl/src/kernels/TopK.ts:124
↓ 2 callersFunctionsearchSorted
(args: { inputs: SearchSortedInputs, backend: MathBackendCPU, attrs: SearchSortedAttrs })
tfjs-backend-cpu/src/kernels/SearchSorted.ts:24
↓ 2 callersFunctionsendChatMsg
(url, msg)
scripts/cloud_funcs/sync_reactnative/index.js:61
↓ 2 callersMethodserialNext
()
tfjs-data/src/iterators/lazy_iterator.ts:584
↓ 2 callersFunctionsetBackend
(backendName: string)
tfjs-core/src/globals.ts:291
↓ 2 callersMethodsetColumnNames
()
tfjs-data/src/datasets/csv_dataset.ts:82
↓ 2 callersMethodsetFastWeightInitDuringBuild
(value: boolean)
tfjs-layers/src/layers/wrappers.ts:551
↓ 2 callersMethodsetFastWeightInitDuringBuild
(value: boolean)
tfjs-layers/src/layers/recurrent.ts:806
↓ 2 callersMethodsetModelInputFromTensor
( modelInput: TFLiteWebModelRunnerTensorInfo, tensor: Tensor)
tfjs-tflite/src/tflite_model.ts:189
↓ 2 callersMethodsetNumThreads
(value: number)
tfjs-tflite/src/types/image_segmenter.ts:34
↓ 2 callersMethodsetNumThreads
(value: number)
tfjs-tflite/src/types/object_detector.ts:31
↓ 2 callersMethodsetNumThreads
(value: number)
tfjs-tflite/src/types/image_classifier.ts:31
↓ 2 callersMethodsetOutputMatrixTextureDriver
( outputMatrixTextureMaybePacked: WebGLTexture, width: number, height: number)
tfjs-backend-webgl/src/gpgpu_context.ts:655
↓ 2 callersMethodsetResourceIdToCapturedInput
(outputs: Tensor[])
tfjs-converter/src/executor/graph_model.ts:477
↓ 2 callersMethodsetTabs
(surfaceList: SurfaceInfoStrict[])
tfjs-vis/src/components/visor.tsx:156
↓ 2 callersMethodsetTimeoutCustom
(functionRef: Function, delay: number)
tfjs-core/src/platforms/platform.ts:50
↓ 2 callersMethodsetUp
(self)
tfjs-converter/python/tensorflowjs/converters/converter_test.py:42
↓ 2 callersFunctionsetWebGLContext
( webGLVersion: number, gl: WebGLRenderingContext)
tfjs-backend-webgl/src/canvas_util.ts:36
↓ 2 callersMethodsetWeights
* 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 callersFunctionsetupFakeWeightFiles
(fileBufferMap: { [filename: string]: string|Float32Array|Int32Array|ArrayBuffer|Uint8Array| Uint16Arr
tfjs-node/src/io/node_http_test.ts:66
↓ 2 callersFunctionshowBackendFlagSettingsAndReturnTunableFlagControllers
* 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 callersFunctionshowDiff
(newSize, masterSize)
scripts/bundle-size-util.js:26
↓ 2 callersFunctionsparseSegmentReduce
( input: TensorInfo, indices: TensorInfo, segmentIds: TensorInfo, isSum = false, backend: WebGPUBacken
tfjs-backend-webgpu/src/kernel_utils/sparse_segment_reduce.ts:25
↓ 2 callersFunctionsparseSegmentReduction
( args: { backend: BackendWasm, inputs: SparseSegmentSumInputs|SparseSegmentMeanInputs, },
tfjs-backend-wasm/src/kernels/SparseSegmentReduction.ts:44
↓ 2 callersFunctionsparseSegmentReductionImpl
( input: TypedArray, inputShape: number[], inputDType: DataType, indices: TypedArray, segmentIds: Type
tfjs-backend-cpu/src/kernels/SparseSegmentReduction_impl.ts:20
↓ 2 callersFunctionsquare_if_positive
(v)
e2e/integration_tests/convert_predict.py:208
↓ 2 callersFunctionstandardizeArgs
( inputs: Tensor|Tensor[]|SymbolicTensor|SymbolicTensor[], initialState: Tensor|Tensor[]|SymbolicTenso
tfjs-layers/src/layers/recurrent.ts:59
↓ 2 callersFunctionstandardizeCallbacks
( callbacks: BaseCallback|BaseCallback[]|CustomCallbackArgs| CustomCallbackArgs[], yieldEvery: Yie
tfjs-layers/src/base_callbacks.ts:476
↓ 2 callersFunctionstandardizeDataIteratorOutput
* 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 callersFunctionstandardizeSampleOrClassWeights
( xWeight: ClassWeight|ClassWeight[]|ClassWeightMap, outputNames: string[], weightType: 'sampleWeight'
tfjs-layers/src/engine/training_utils.ts:34
↓ 2 callersMethodstandardizeUserDataXY
( x: Tensor|Tensor[]|{[inputName: string]: Tensor}, y: Tensor|Tensor[]|{[inputName: string]: Tenso
tfjs-layers/src/engine/training.ts:1145
↓ 2 callersFunctionstart
()
tfjs-tflite/demo/src/script.ts:30
↓ 2 callersFunctionstridesOrDilationsArePositive
(values: number| number[])
tfjs-core/src/ops/conv_util.ts:576
↓ 2 callersFunctionstringLiteralArray
(a: T[])
tfjs-layers/src/keras_format/utils.ts:26
↓ 2 callersMethodstylize
(contentImage: string, styleImage: string)
tfjs-react-native/integration_rn59/components/webcam/webcam_demo.tsx:127
↓ 2 callersFunctionsummaryFileWriter
( logdir: string, maxQueue = 10, flushMillis = 120000, filenameSuffix = '.v2')
tfjs-node/src/tensorboard.ts:122
↓ 2 callersMethodsurfaceId
(label: string, tab: string)
tfjs-vis/src/components/visor.tsx:152
↓ 2 callersMethodsyncData
* Await the values of all losses and metrics.
tfjs-layers/src/base_callbacks.ts:316
← previousnext →1,601–1,700 of 6,022, ranked by callers