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
↓ 8 callers
Function
cast
(x: Tensor, dtype: tfc.DataType)
tfjs-layers/src/backend/tfjs_backend.ts:73
↓ 8 callers
Method
compile
* Compiles the inference graph and returns the minimal set of nodes that are * required for execution, in the correct execution order. * @return
tfjs-converter/src/executor/graph_executor.ts:167
↓ 8 callers
Function
computeProbs
( events: Float32Array|Uint8Array|Int32Array, numOutcomes: number)
tfjs-core/src/ops/multinomial_test.ts:123
↓ 8 callers
Function
confusionMatrix
( labels: Tensor1D, predictions: Tensor1D, numClasses?: number, weights?: Tensor1D)
tfjs-vis/src/util/math.ts:171
↓ 8 callers
Function
createDummyModel
(inputSize: number)
tfjs-layers/src/engine/training_test.ts:2221
↓ 8 callers
Function
executeOp
( node: Node, tensorMap: NamedTensorsMap, context: ExecutionContext, resourceManager?: ResourceManager
tfjs-converter/src/operations/operation_executor.ts:55
↓ 8 callers
Function
expectNumbersClose
(a: number, e: number, epsilon?: number)
tfjs-core/src/test_util.ts:127
↓ 8 callers
Function
generateCaseInputs
(totalSizeTensor: number, totalSizeFilter: number)
tfjs-core/src/ops/fused/fused_conv2d_test.ts:22
↓ 8 callers
Function
generateDropoutMask
(args: { ones: () => tfc.Tensor, rate: number, training?: boolean, count?: number, dropoutFunc?: Fun
tfjs-layers/src/layers/recurrent.ts:2105
↓ 8 callers
Function
getExecutionSubgraph
( inputs: NamedTensorMap, outputs: Node[], weightMap: NamedTensorsMap, initNodes?: Node[])
tfjs-converter/src/executor/model_analysis.ts:41
↓ 8 callers
Method
getMap
* Returns the singleton instance of the map.
tfjs-core/src/serialization.ts:144
↓ 8 callers
Method
getNumFreeBuffers
()
tfjs-backend-webgpu/src/buffer_manager.ts:94
↓ 8 callers
Method
getNumUsedTextures
()
tfjs-backend-webgl/src/texture_manager.ts:154
↓ 8 callers
Function
getTFDType
(dataType: tf.DataType)
tfjs-node/src/nodejs_kernel_backend.ts:710
↓ 8 callers
Function
getUid
(prefix = '')
tfjs-layers/src/backend/state.ts:33
↓ 8 callers
Function
grad
* Provided `f(x)`, returns another function `g(x, dy?)`, which gives the * gradient of `f(x)` with respect to `x`. * * If `dy` is provided, the gra
tfjs-core/src/gradients.ts:59
↓ 8 callers
Function
hasExtension
(gl: WebGLRenderingContext, extensionName: string)
tfjs-backend-webgl/src/webgl_util.ts:558
↓ 8 callers
Function
hashLen16
(u: Long, v: Long, mul = hexToLong('9ddfea08eb382d69'))
tfjs-core/src/hash_util.ts:59
↓ 8 callers
Function
imag
(args: {inputs: ImagInputs, backend: MathBackendCPU})
tfjs-backend-cpu/src/kernels/Imag.ts:22
↓ 8 callers
Function
isDataArray
(x: Tensor|Tensor[]| {[inputName: string]: Tensor})
tfjs-layers/src/engine/training.ts:57
↓ 8 callers
Method
isInf
(this: T)
tfjs-core/src/public/chained_ops/is_inf.ts:25
↓ 8 callers
Function
iteratorFromZipped
( iterators: IteratorContainer, mismatchMode: ZipMismatchMode = ZipMismatchMode.FAIL)
tfjs-data/src/iterators/lazy_iterator.ts:136
↓ 8 callers
Function
linechart
( container: Drawable, data: XYPlotData, opts: XYPlotOptions = {})
tfjs-vis/src/render/linechart.ts:60
↓ 8 callers
Method
load
* Load a model from local storage. * * See the documentation to `browserLocalStorage` for details on the saved * artifacts. * * @return
tfjs-react-native/src/async_storage_io.ts:131
↓ 8 callers
Method
load
()
tfjs-node/src/io/file_system.ts:128
↓ 8 callers
Function
loadLayersModel
( pathOrIOHandler: string|io.IOHandler, options?: io.LoadOptions)
tfjs-layers/src/models.ts:248
↓ 8 callers
Function
makeTensor
( values: TensorLike|WebGLData|WebGPUData, shape: number[], inferredShape: number[], dtype?: DataType)
tfjs-core/src/ops/tensor_ops_util.ts:25
↓ 8 callers
Function
mean
(values: TypedArray|number[])
tfjs-core/src/ops/rand_util.ts:245
↓ 8 callers
Function
monitorPromisesProgress
( promises: Array<Promise<T>>, onProgress: OnProgressCallback, startFraction?: number, endFraction?: n
tfjs-core/src/io/progress.ts:30
↓ 8 callers
Function
ones
( shape: ShapeMap[R], dtype: DataType = 'float32')
tfjs-core/src/ops/ones.ts:39
↓ 8 callers
Function
parseAxisParam
( axis: number|number[], shape: number[])
tfjs-core/src/util_base.ts:378
↓ 8 callers
Function
permuteAxesAndTranspose
( x: TensorInfo, axis: number|number[], backend: BackendWasm)
tfjs-backend-wasm/src/kernels/kernel_utils.ts:30
↓ 8 callers
Function
printTime
(elapsed)
e2e/benchmarks/browserstack-benchmark/index.js:676
↓ 8 callers
Method
profile
(query: () => (TensorContainer | Promise<TensorContainer>))
tfjs-core/src/engine.ts:976
↓ 8 callers
Function
pyListRepeat
(value: any, numValues: number)
tfjs-layers/src/utils/generic_utils.ts:24
↓ 8 callers
Function
randomNormal
( shape: Shape, mean = 0.0, stddev = 1.0, dtype?: 'float32'|'int32', seed?: number)
tfjs-layers/src/backend/tfjs_backend.ts:365
↓ 8 callers
Method
resetStates
* Reset the state tensors of the RNN. * * If the `states` argument is `undefined` or `null`, will set the * state tensor(s) of the RNN to all
tfjs-layers/src/layers/recurrent.ts:562
↓ 8 callers
Function
reverse
( args: {inputs: ReverseInputs, backend: MathBackendCPU, attrs: ReverseAttrs})
tfjs-backend-cpu/src/kernels/Reverse.ts:24
↓ 8 callers
Function
scatterplot
( container: Drawable, data: XYPlotData, opts: XYPlotOptions = {})
tfjs-vis/src/render/scatterplot.ts:46
↓ 8 callers
Function
setWasmPaths
( prefixOrFileMap: string|{[key in WasmBinaryName]?: string}, usePlatformFetch = false)
tfjs-backend-wasm/src/backend_wasm.ts:479
↓ 8 callers
Function
standardizeWeights
( y: Tensor, sampleWeight?: Tensor, classWeight?: ClassWeight, sampleWeightMode?: 'temporal')
tfjs-layers/src/engine/training_utils.ts:125
↓ 8 callers
Method
tick
(tickConfig: {placeholderForLossesAndMetrics: string})
tfjs-node/src/callbacks_test.ts:29
↓ 8 callers
Function
toTexture
( gl: WebGL2RenderingContext, imageTensor: tf.Tensor3D, texture?: WebGLTexture)
tfjs-react-native/src/camera/camera.ts:109
↓ 8 callers
Method
transformGraph
( graph: tensorflow.IGraphDef, signature: tensorflow.ISignatureDef = {})
tfjs-converter/src/operations/operation_mapper.ts:74
↓ 8 callers
Function
unaryKernelFuncFromImpl
( name: string, unaryImpl: SimpleUnaryImpl<I, O>, dtype?: DataTypeFor<O>)
tfjs-backend-cpu/src/utils/unary_utils.ts:53
↓ 7 callers
Method
RepositoryName
FullName after removing the local path to the repository. If we have a real absolute path name here we can try to do something smart: detecti
tfjs-backend-wasm/tools/cpplint.py:1111
↓ 7 callers
Function
any
( args: {inputs: AnyInputs, backend: MathBackendCPU, attrs: AnyAttrs})
tfjs-backend-cpu/src/kernels/Any.ts:25
↓ 7 callers
Function
arrayProduct
(arr: number[])
tfjs-backend-webgpu/src/webgpu_util.ts:19
↓ 7 callers
Method
as2D
(rows: number, columns: number)
tfjs-core/src/public/chained_ops/as2d.ts:24
↓ 7 callers
Method
asScalar
()
tfjs-core/src/public/chained_ops/as_scalar.ts:25
↓ 7 callers
Function
askUserForVersions
(validVersions: string[], packageName: string)
scripts/release_notes/release_notes.ts:97
↓ 7 callers
Function
barchart
( container: Drawable, data: Array<{index: number; value: number;}>, opts: BarChartOpts = {})
tfjs-vis/src/render/barchart.ts:46
↓ 7 callers
Method
batchNorm
( mean: Tensor<R>|Tensor1D|TensorLike, variance: Tensor<R>|Tensor1D|TensorLike, offset
tfjs-core/src/public/chained_ops/batchnorm.ts:23
↓ 7 callers
Function
biasActivationSnippet
( hasBias: boolean, activation: backend_util.Activation)
tfjs-backend-webgpu/src/activation_util.ts:70
↓ 7 callers
Method
bufferSync
(t: TensorInfo)
tfjs-backend-webgl/src/backend_webgl.ts:473
↓ 7 callers
Method
build
(inputShape?: Shape|Shape[])
tfjs-layers/src/models.ts:589
↓ 7 callers
Method
build
(inputShape: Shape|Shape[])
tfjs-layers/src/layers/merge.ts:977
↓ 7 callers
Method
call
(inputs: Tensor|Tensor[], kwargs: Kwargs)
tfjs-layers/src/models.ts:582
↓ 7 callers
Function
cast
( args: {inputs: CastInputs, attrs: CastAttrs, backend: BackendWasm})
tfjs-backend-wasm/src/kernels/Cast.ts:23
↓ 7 callers
Function
complex
( args: {inputs: ComplexInputs, backend: MathBackendWebGL})
tfjs-backend-webgl/src/kernels/Complex.ts:32
↓ 7 callers
Function
confusionMatrix
( container: Drawable, data: ConfusionMatrixData, opts: ConfusionMatrixOptions = {})
tfjs-vis/src/render/confusion_matrix.ts:69
↓ 7 callers
Function
createFunctionalModel
()
tfjs-layers/src/engine/training_test.ts:2862
↓ 7 callers
Function
dataTypeToGPUType
(type: DataType, component = 1)
tfjs-backend-webgpu/src/webgpu_program.ts:782
↓ 7 callers
Function
disposeIntermediateTensorInfoOrNull
( backend: MathBackendWebGL, tensorInfo: TensorInfo)
tfjs-backend-webgl/src/kernels/TopK.ts:28
↓ 7 callers
Function
elu
(args?: ELULayerArgs)
tfjs-layers/src/exports_layers.ts:102
↓ 7 callers
Method
endComputePassEncoder
()
tfjs-backend-webgpu/src/backend_webgpu.ts:317
↓ 7 callers
Function
envSatisfiesConstraints
( env: Environment, testEnv: TestEnv, constraints: Constraints)
tfjs-core/src/jasmine_util.ts:73
↓ 7 callers
Function
executeOp
(node: Node, tensorMap: NamedTensorsMap, context: ExecutionContext, ops = tfOps)
tfjs-converter/src/operations/executors/reduction_executor.ts:29
↓ 7 callers
Function
executeOp
(node: Node, tensorMap: NamedTensorsMap, context: ExecutionContext, ops = tfOps)
tfjs-converter/src/operations/executors/normalization_executor.ts:29
↓ 7 callers
Function
executeOp
( node: Node, tensorMap: NamedTensorsMap, context: ExecutionContext, resourceManager: ResourceManager,
tfjs-converter/src/operations/executors/dynamic_executor.ts:52
↓ 7 callers
Function
fill
* Creates a `tf.Tensor` filled with a scalar value. * * ```js * tf.fill([2, 2], 4).print(); * ``` * * @param shape An array of integers defining
tfjs-core/src/ops/fill.ts:40
↓ 7 callers
Function
flatten
(arr: any)
tfjs-core/src/ops/local_response_normalization_test.ts:27
↓ 7 callers
Function
format
({match, index, input}: {match: string, index: number, input: string})
tfjs-layers/src/layers/nlp/match_all_polyfill_test.ts:20
↓ 7 callers
Method
fromConfig
@nocollapse
tfjs-core/src/optimizers/sgd_optimizer.ts:96
↓ 7 callers
Function
generateNotes
(repositories: util.Repo[])
scripts/release_notes/release_notes.ts:247
↓ 7 callers
Function
get
(identifier: string|LossOrMetricFn)
tfjs-layers/src/metrics.ts:149
↓ 7 callers
Method
getConfig
()
tfjs-layers/src/layers/core.ts:292
↓ 7 callers
Function
getEnumKeyFromValue
(object: any, value: number)
tfjs-node/src/saved_model.ts:55
↓ 7 callers
Function
getFlatOffsetUniformName
(texName: string)
tfjs-backend-webgl/src/shader_compiler.ts:888
↓ 7 callers
Function
getFloat16Decoder
()
tfjs-core/src/io/io_utils.ts:720
↓ 7 callers
Function
getImageType
(content: Uint8Array)
tfjs-node/src/image.ts:270
↓ 7 callers
Function
getPredictFnForModel
* Wrap the model's predict function (`model.predict` for tf.LayersModel * and `model.executeAsync` for tf.GraphModel) with the input. * * @param mo
e2e/benchmarks/benchmark_util.js:152
↓ 7 callers
Function
getTaggedVersions
(packageName: string)
scripts/release_notes/release_notes.ts:122
↓ 7 callers
Function
getTensorsInContainer
(result: TensorContainer)
tfjs-core/src/tensor_util.ts:54
↓ 7 callers
Function
getUnaryOpString
(type: UnaryOpType, useVec4?: boolean)
tfjs-backend-webgpu/src/unary_op_util.ts:219
↓ 7 callers
Function
grad
(dy: Tensor4D, saved: Tensor[])
tfjs-core/src/ops/fused/conv2d.ts:242
↓ 7 callers
Method
gradients
* Returns gradients of `f` with respect to each of the `xs`. The gradients * returned are of the same length as `xs`, but some might be null if `f`
tfjs-core/src/engine.ts:1105
↓ 7 callers
Method
histogram
* Write a histogram summary, for later analysis in TensorBoard's 'Histograms' * and 'Distributions' dashboards (data written using this API will ap
tfjs-node/src/tensorboard.ts:75
↓ 7 callers
Function
insertionPoint
(i: number)
tfjs-core/src/backends/non_max_suppression_util_test.ts:21
↓ 7 callers
Function
isControlFlow
(node: Node)
tfjs-converter/src/executor/model_analysis.ts:337
↓ 7 callers
Function
isDataDict
(x: Tensor|Tensor[]| {[inputName: string]: Tensor})
tfjs-layers/src/engine/training.ts:65
↓ 7 callers
Function
isIterable
(obj: any)
tfjs-data/src/util/deep_map.ts:255
↓ 7 callers
Function
isSliceContinous
( shape: number[], begin: number[], size: number[])
tfjs-core/src/ops/slice_util.ts:352
↓ 7 callers
Method
iterator
()
tfjs-layers/src/engine/dataset_fakes.ts:209
↓ 7 callers
Method
loadSync
* Synchronously construct the in memory weight map and * compile the inference graph. * * @doc {heading: 'Models', subheading: 'Classes', ign
tfjs-converter/src/executor/graph_model.ts:176
↓ 7 callers
Function
makeBatches
( size: number, batchSize: number)
tfjs-layers/src/engine/training_tensors.ts:228
↓ 7 callers
Method
onEpochEnd
(epoch: number, logs: Logs)
tfjs-layers/src/base_callbacks_test.ts:218
↓ 7 callers
Function
perClassAccuracy
( labels: Tensor1D, predictions: Tensor1D, numClasses?: number)
tfjs-vis/src/util/math.ts:281
↓ 7 callers
Function
pool2d
( x: Tensor, poolSize: [number, number], strides?: [number, number], padding?: PaddingMode, dataFormat
tfjs-layers/src/layers/pooling.ts:42
↓ 7 callers
Function
pow
(x: Tensor, a: Tensor|number)
tfjs-layers/src/backend/tfjs_backend.ts:549
← previous
next →
601–700 of 6,022, ranked by callers