MCPcopy Create free account

hub / github.com/tensorflow/tfjs / functions

Functions6,022 in github.com/tensorflow/tfjs

↓ 8 callersFunctioncast
(x: Tensor, dtype: tfc.DataType)
tfjs-layers/src/backend/tfjs_backend.ts:73
↓ 8 callersMethodcompile
* 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 callersFunctioncomputeProbs
( events: Float32Array|Uint8Array|Int32Array, numOutcomes: number)
tfjs-core/src/ops/multinomial_test.ts:123
↓ 8 callersFunctionconfusionMatrix
( labels: Tensor1D, predictions: Tensor1D, numClasses?: number, weights?: Tensor1D)
tfjs-vis/src/util/math.ts:171
↓ 8 callersFunctioncreateDummyModel
(inputSize: number)
tfjs-layers/src/engine/training_test.ts:2221
↓ 8 callersFunctionexecuteOp
( node: Node, tensorMap: NamedTensorsMap, context: ExecutionContext, resourceManager?: ResourceManager
tfjs-converter/src/operations/operation_executor.ts:55
↓ 8 callersFunctionexpectNumbersClose
(a: number, e: number, epsilon?: number)
tfjs-core/src/test_util.ts:127
↓ 8 callersFunctiongenerateCaseInputs
(totalSizeTensor: number, totalSizeFilter: number)
tfjs-core/src/ops/fused/fused_conv2d_test.ts:22
↓ 8 callersFunctiongenerateDropoutMask
(args: { ones: () => tfc.Tensor, rate: number, training?: boolean, count?: number, dropoutFunc?: Fun
tfjs-layers/src/layers/recurrent.ts:2105
↓ 8 callersFunctiongetExecutionSubgraph
( inputs: NamedTensorMap, outputs: Node[], weightMap: NamedTensorsMap, initNodes?: Node[])
tfjs-converter/src/executor/model_analysis.ts:41
↓ 8 callersMethodgetMap
* Returns the singleton instance of the map.
tfjs-core/src/serialization.ts:144
↓ 8 callersMethodgetNumFreeBuffers
()
tfjs-backend-webgpu/src/buffer_manager.ts:94
↓ 8 callersMethodgetNumUsedTextures
()
tfjs-backend-webgl/src/texture_manager.ts:154
↓ 8 callersFunctiongetTFDType
(dataType: tf.DataType)
tfjs-node/src/nodejs_kernel_backend.ts:710
↓ 8 callersFunctiongetUid
(prefix = '')
tfjs-layers/src/backend/state.ts:33
↓ 8 callersFunctiongrad
* 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 callersFunctionhasExtension
(gl: WebGLRenderingContext, extensionName: string)
tfjs-backend-webgl/src/webgl_util.ts:558
↓ 8 callersFunctionhashLen16
(u: Long, v: Long, mul = hexToLong('9ddfea08eb382d69'))
tfjs-core/src/hash_util.ts:59
↓ 8 callersFunctionimag
(args: {inputs: ImagInputs, backend: MathBackendCPU})
tfjs-backend-cpu/src/kernels/Imag.ts:22
↓ 8 callersFunctionisDataArray
(x: Tensor|Tensor[]| {[inputName: string]: Tensor})
tfjs-layers/src/engine/training.ts:57
↓ 8 callersMethodisInf
(this: T)
tfjs-core/src/public/chained_ops/is_inf.ts:25
↓ 8 callersFunctioniteratorFromZipped
( iterators: IteratorContainer, mismatchMode: ZipMismatchMode = ZipMismatchMode.FAIL)
tfjs-data/src/iterators/lazy_iterator.ts:136
↓ 8 callersFunctionlinechart
( container: Drawable, data: XYPlotData, opts: XYPlotOptions = {})
tfjs-vis/src/render/linechart.ts:60
↓ 8 callersMethodload
* 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 callersMethodload
()
tfjs-node/src/io/file_system.ts:128
↓ 8 callersFunctionloadLayersModel
( pathOrIOHandler: string|io.IOHandler, options?: io.LoadOptions)
tfjs-layers/src/models.ts:248
↓ 8 callersFunctionmakeTensor
( values: TensorLike|WebGLData|WebGPUData, shape: number[], inferredShape: number[], dtype?: DataType)
tfjs-core/src/ops/tensor_ops_util.ts:25
↓ 8 callersFunctionmean
(values: TypedArray|number[])
tfjs-core/src/ops/rand_util.ts:245
↓ 8 callersFunctionmonitorPromisesProgress
( promises: Array<Promise<T>>, onProgress: OnProgressCallback, startFraction?: number, endFraction?: n
tfjs-core/src/io/progress.ts:30
↓ 8 callersFunctionones
( shape: ShapeMap[R], dtype: DataType = 'float32')
tfjs-core/src/ops/ones.ts:39
↓ 8 callersFunctionparseAxisParam
( axis: number|number[], shape: number[])
tfjs-core/src/util_base.ts:378
↓ 8 callersFunctionpermuteAxesAndTranspose
( x: TensorInfo, axis: number|number[], backend: BackendWasm)
tfjs-backend-wasm/src/kernels/kernel_utils.ts:30
↓ 8 callersFunctionprintTime
(elapsed)
e2e/benchmarks/browserstack-benchmark/index.js:676
↓ 8 callersMethodprofile
(query: () => (TensorContainer | Promise<TensorContainer>))
tfjs-core/src/engine.ts:976
↓ 8 callersFunctionpyListRepeat
(value: any, numValues: number)
tfjs-layers/src/utils/generic_utils.ts:24
↓ 8 callersFunctionrandomNormal
( shape: Shape, mean = 0.0, stddev = 1.0, dtype?: 'float32'|'int32', seed?: number)
tfjs-layers/src/backend/tfjs_backend.ts:365
↓ 8 callersMethodresetStates
* 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 callersFunctionreverse
( args: {inputs: ReverseInputs, backend: MathBackendCPU, attrs: ReverseAttrs})
tfjs-backend-cpu/src/kernels/Reverse.ts:24
↓ 8 callersFunctionscatterplot
( container: Drawable, data: XYPlotData, opts: XYPlotOptions = {})
tfjs-vis/src/render/scatterplot.ts:46
↓ 8 callersFunctionsetWasmPaths
( prefixOrFileMap: string|{[key in WasmBinaryName]?: string}, usePlatformFetch = false)
tfjs-backend-wasm/src/backend_wasm.ts:479
↓ 8 callersFunctionstandardizeWeights
( y: Tensor, sampleWeight?: Tensor, classWeight?: ClassWeight, sampleWeightMode?: 'temporal')
tfjs-layers/src/engine/training_utils.ts:125
↓ 8 callersMethodtick
(tickConfig: {placeholderForLossesAndMetrics: string})
tfjs-node/src/callbacks_test.ts:29
↓ 8 callersFunctiontoTexture
( gl: WebGL2RenderingContext, imageTensor: tf.Tensor3D, texture?: WebGLTexture)
tfjs-react-native/src/camera/camera.ts:109
↓ 8 callersMethodtransformGraph
( graph: tensorflow.IGraphDef, signature: tensorflow.ISignatureDef = {})
tfjs-converter/src/operations/operation_mapper.ts:74
↓ 8 callersFunctionunaryKernelFuncFromImpl
( name: string, unaryImpl: SimpleUnaryImpl<I, O>, dtype?: DataTypeFor<O>)
tfjs-backend-cpu/src/utils/unary_utils.ts:53
↓ 7 callersMethodRepositoryName
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 callersFunctionany
( args: {inputs: AnyInputs, backend: MathBackendCPU, attrs: AnyAttrs})
tfjs-backend-cpu/src/kernels/Any.ts:25
↓ 7 callersFunctionarrayProduct
(arr: number[])
tfjs-backend-webgpu/src/webgpu_util.ts:19
↓ 7 callersMethodas2D
(rows: number, columns: number)
tfjs-core/src/public/chained_ops/as2d.ts:24
↓ 7 callersMethodasScalar
()
tfjs-core/src/public/chained_ops/as_scalar.ts:25
↓ 7 callersFunctionaskUserForVersions
(validVersions: string[], packageName: string)
scripts/release_notes/release_notes.ts:97
↓ 7 callersFunctionbarchart
( container: Drawable, data: Array<{index: number; value: number;}>, opts: BarChartOpts = {})
tfjs-vis/src/render/barchart.ts:46
↓ 7 callersMethodbatchNorm
( mean: Tensor<R>|Tensor1D|TensorLike, variance: Tensor<R>|Tensor1D|TensorLike, offset
tfjs-core/src/public/chained_ops/batchnorm.ts:23
↓ 7 callersFunctionbiasActivationSnippet
( hasBias: boolean, activation: backend_util.Activation)
tfjs-backend-webgpu/src/activation_util.ts:70
↓ 7 callersMethodbufferSync
(t: TensorInfo)
tfjs-backend-webgl/src/backend_webgl.ts:473
↓ 7 callersMethodbuild
(inputShape?: Shape|Shape[])
tfjs-layers/src/models.ts:589
↓ 7 callersMethodbuild
(inputShape: Shape|Shape[])
tfjs-layers/src/layers/merge.ts:977
↓ 7 callersMethodcall
(inputs: Tensor|Tensor[], kwargs: Kwargs)
tfjs-layers/src/models.ts:582
↓ 7 callersFunctioncast
( args: {inputs: CastInputs, attrs: CastAttrs, backend: BackendWasm})
tfjs-backend-wasm/src/kernels/Cast.ts:23
↓ 7 callersFunctioncomplex
( args: {inputs: ComplexInputs, backend: MathBackendWebGL})
tfjs-backend-webgl/src/kernels/Complex.ts:32
↓ 7 callersFunctionconfusionMatrix
( container: Drawable, data: ConfusionMatrixData, opts: ConfusionMatrixOptions = {})
tfjs-vis/src/render/confusion_matrix.ts:69
↓ 7 callersFunctioncreateFunctionalModel
()
tfjs-layers/src/engine/training_test.ts:2862
↓ 7 callersFunctiondataTypeToGPUType
(type: DataType, component = 1)
tfjs-backend-webgpu/src/webgpu_program.ts:782
↓ 7 callersFunctiondisposeIntermediateTensorInfoOrNull
( backend: MathBackendWebGL, tensorInfo: TensorInfo)
tfjs-backend-webgl/src/kernels/TopK.ts:28
↓ 7 callersFunctionelu
(args?: ELULayerArgs)
tfjs-layers/src/exports_layers.ts:102
↓ 7 callersMethodendComputePassEncoder
()
tfjs-backend-webgpu/src/backend_webgpu.ts:317
↓ 7 callersFunctionenvSatisfiesConstraints
( env: Environment, testEnv: TestEnv, constraints: Constraints)
tfjs-core/src/jasmine_util.ts:73
↓ 7 callersFunctionexecuteOp
(node: Node, tensorMap: NamedTensorsMap, context: ExecutionContext, ops = tfOps)
tfjs-converter/src/operations/executors/reduction_executor.ts:29
↓ 7 callersFunctionexecuteOp
(node: Node, tensorMap: NamedTensorsMap, context: ExecutionContext, ops = tfOps)
tfjs-converter/src/operations/executors/normalization_executor.ts:29
↓ 7 callersFunctionexecuteOp
( node: Node, tensorMap: NamedTensorsMap, context: ExecutionContext, resourceManager: ResourceManager,
tfjs-converter/src/operations/executors/dynamic_executor.ts:52
↓ 7 callersFunctionfill
* 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 callersFunctionflatten
(arr: any)
tfjs-core/src/ops/local_response_normalization_test.ts:27
↓ 7 callersFunctionformat
({match, index, input}: {match: string, index: number, input: string})
tfjs-layers/src/layers/nlp/match_all_polyfill_test.ts:20
↓ 7 callersMethodfromConfig
@nocollapse
tfjs-core/src/optimizers/sgd_optimizer.ts:96
↓ 7 callersFunctiongenerateNotes
(repositories: util.Repo[])
scripts/release_notes/release_notes.ts:247
↓ 7 callersFunctionget
(identifier: string|LossOrMetricFn)
tfjs-layers/src/metrics.ts:149
↓ 7 callersMethodgetConfig
()
tfjs-layers/src/layers/core.ts:292
↓ 7 callersFunctiongetEnumKeyFromValue
(object: any, value: number)
tfjs-node/src/saved_model.ts:55
↓ 7 callersFunctiongetFlatOffsetUniformName
(texName: string)
tfjs-backend-webgl/src/shader_compiler.ts:888
↓ 7 callersFunctiongetFloat16Decoder
()
tfjs-core/src/io/io_utils.ts:720
↓ 7 callersFunctiongetImageType
(content: Uint8Array)
tfjs-node/src/image.ts:270
↓ 7 callersFunctiongetPredictFnForModel
* 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 callersFunctiongetTaggedVersions
(packageName: string)
scripts/release_notes/release_notes.ts:122
↓ 7 callersFunctiongetTensorsInContainer
(result: TensorContainer)
tfjs-core/src/tensor_util.ts:54
↓ 7 callersFunctiongetUnaryOpString
(type: UnaryOpType, useVec4?: boolean)
tfjs-backend-webgpu/src/unary_op_util.ts:219
↓ 7 callersFunctiongrad
(dy: Tensor4D, saved: Tensor[])
tfjs-core/src/ops/fused/conv2d.ts:242
↓ 7 callersMethodgradients
* 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 callersMethodhistogram
* 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 callersFunctioninsertionPoint
(i: number)
tfjs-core/src/backends/non_max_suppression_util_test.ts:21
↓ 7 callersFunctionisControlFlow
(node: Node)
tfjs-converter/src/executor/model_analysis.ts:337
↓ 7 callersFunctionisDataDict
(x: Tensor|Tensor[]| {[inputName: string]: Tensor})
tfjs-layers/src/engine/training.ts:65
↓ 7 callersFunctionisIterable
(obj: any)
tfjs-data/src/util/deep_map.ts:255
↓ 7 callersFunctionisSliceContinous
( shape: number[], begin: number[], size: number[])
tfjs-core/src/ops/slice_util.ts:352
↓ 7 callersMethoditerator
()
tfjs-layers/src/engine/dataset_fakes.ts:209
↓ 7 callersMethodloadSync
* 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 callersFunctionmakeBatches
( size: number, batchSize: number)
tfjs-layers/src/engine/training_tensors.ts:228
↓ 7 callersMethodonEpochEnd
(epoch: number, logs: Logs)
tfjs-layers/src/base_callbacks_test.ts:218
↓ 7 callersFunctionperClassAccuracy
( labels: Tensor1D, predictions: Tensor1D, numClasses?: number)
tfjs-vis/src/util/math.ts:281
↓ 7 callersFunctionpool2d
( x: Tensor, poolSize: [number, number], strides?: [number, number], padding?: PaddingMode, dataFormat
tfjs-layers/src/layers/pooling.ts:42
↓ 7 callersFunctionpow
(x: Tensor, a: Tensor|number)
tfjs-layers/src/backend/tfjs_backend.ts:549
← previousnext →601–700 of 6,022, ranked by callers