MCPcopy Index your code

hub / github.com/tensorflow/tfjs / functions

Functions6,022 in github.com/tensorflow/tfjs

⨍ Functions 6,022◇ Types & classes 1,932↳ Endpoints 2
↓ 85 callersMethodbuffer
( shape: ShapeMap[R], dtype: D, values?: DataTypeMap[D])
tfjs-core/src/tensor.ts:196
↓ 85 callersMethodfit
* Trains the model for a fixed number of epochs (iterations on a * dataset). * * ```js * const model = tf.sequential({ * layers: [t
tfjs-layers/src/engine/training.ts:1464
↓ 85 callersMethodsum
(axis?: number|number[], keepDims?: boolean)
tfjs-core/src/public/chained_ops/sum.ts:23
↓ 84 callersMethodcast
(x: T, dtype: DataType)
tfjs-core/src/tensor.ts:195
↓ 84 callersMethodregisterFlag
( flagName: string, evaluationFn: FlagEvaluationFn, setHook?: (value: FlagValue) => void)
tfjs-core/src/environment.ts:70
↓ 82 callersMethodgather
( this: T, indices: Tensor|TensorLike, axis?: number, batchDims?: number)
tfjs-core/src/public/chained_ops/gather.ts:24
↓ 80 callersMethodall
(this: T, axis?: number|number[], keepDims?: boolean)
tfjs-core/src/public/chained_ops/all.ts:23
↓ 80 callersFunctionreshape
(args: { inputs: ReshapeInputs, backend: MathBackendWebGL, attrs: ReshapeAttrs })
tfjs-backend-webgl/src/kernels/Reshape.ts:24
↓ 80 callersMethodsub
(b: Tensor|TensorLike)
tfjs-core/src/public/chained_ops/sub.ts:23
↓ 79 callersFunctionsum
(arr: number[])
tfjs-core/src/util_base.ts:106
↓ 78 callersMethoddispose
()
tfjs-layers/src/engine/training.ts:1916
↓ 77 callersMethoddecode
Decode the provided bytes into a string using the provided encoding.
tfjs-core/src/platforms/platform.ts:48
↓ 77 callersMethoddepthwiseConv2d
( filter: Tensor4D|TensorLike4D, strides: [number, number]|number, pad: 'valid'|'same'|number,
tfjs-core/src/public/chained_ops/depthwise_conv2d.ts:23
↓ 76 callersMethoddot
(b: Tensor|TensorLike)
tfjs-core/src/public/chained_ops/dot.ts:23
↓ 75 callersFunctioncast
( args: {inputs: CastInputs, backend: MathBackendCPU, attrs: CastAttrs})
tfjs-backend-cpu/src/kernels/Cast.ts:49
↓ 75 callersMethoddispose
()
tfjs-backend-webgl/src/gpgpu_context.ts:168
↓ 75 callersMethodmakeTensorInfo
( shape: number[], dtype: DataType, values?: BackendValues|string[])
tfjs-backend-webgl/src/backend_webgl.ts:783
↓ 75 callersMethodreset
* Resets the engine state. Removes all backends but does not remove * registered backend factories.
tfjs-core/src/engine.ts:1265
↓ 74 callersMethodcompile
* See `LayersModel.compile`. * * @param args
tfjs-layers/src/models.ts:813
↓ 74 callersFunctionflatDispatchLayout
(shape: number[])
tfjs-backend-webgpu/src/webgpu_util.ts:142
↓ 72 callersFunctioncreateNumericArrayAttrFromIndex
(inputIndex: number)
tfjs-converter/src/operations/executors/test_helper.ts:53
↓ 70 callersMethodfitDataset
* Trains the model using a dataset object. * * @param dataset A dataset object. Its `iterator()` method is expected * to generate a dataset
tfjs-layers/src/engine/training.ts:1791
↓ 69 callersMethodfind
* Looks up keys in a hash table, outputs the corresponding values. * * Performs batch lookups, for every element in the key tensor, `find` *
tfjs-converter/src/executor/hash_table.ts:126
↓ 68 callersMethodflatten
()
tfjs-core/src/public/chained_ops/flatten.ts:24
↓ 68 callersMethodload
()
tfjs-converter/demo/mobilenet/mobilenet.js:32
↓ 65 callersFunctionassertNotComplex
( tensor: TensorInfo|TensorInfo[], opName: string)
tfjs-backend-cpu/src/cpu_util.ts:20
↓ 65 callersMethodcreate
* The factory function to create a NLClassifier instance. * * @param model The path to load the TFLite model from, or the model content *
tfjs-tflite/src/types/nl_classifier.ts:39
↓ 65 callersMethodnow
* Returns the current high-resolution time in milliseconds relative to an * arbitrary time in the past. It works across different platforms (node.j
tfjs-core/src/platforms/platform.ts:40
↓ 64 callersFunctiongetExactlyOneTensor
(xs: Tensor|Tensor[])
tfjs-layers/src/utils/types_utils.ts:47
↓ 64 callersFunctionreshape
( args: {inputs: ReshapeInputs, backend: WebGPUBackend, attrs: ReshapeAttrs})
tfjs-backend-webgpu/src/kernels/Reshape.ts:22
↓ 63 callersFunctioncreateStrAttr
(str: string)
tfjs-converter/src/operations/executors/test_helper.ts:29
↓ 63 callersMethoddiv
(b: Tensor|TensorLike)
tfjs-core/src/public/chained_ops/div.ts:24
↓ 62 callersFunctionassert
(expr: boolean, msg: () => string)
tfjs-core/src/util_base.ts:151
↓ 62 callersMethodexecute
* Single Execute the inference for the input tensors and return activation * values for specified output node names without batching. * * @pa
tfjs-core/src/model_types.ts:101
↓ 60 callersMethodload
* Load model artifacts via HTTP request(s). * * See the documentation to `tf.io.http` for details on the saved * artifacts. * * @return
tfjs-core/src/io/http.ts:188
↓ 59 callersMethodequal
(b: Tensor|TensorLike)
tfjs-core/src/public/chained_ops/equal.ts:23
↓ 58 callersFunctionconvertPythonicToTs
( pythonicConfig: PyJsonValue, key?: string)
tfjs-layers/src/utils/serialization_utils.ts:44
↓ 58 callersFunctionexpectValuesInRange
( actual: TypedArray|number[], low: number, high: number)
tfjs-core/src/test_util.ts:149
↓ 57 callersFunctioncreateNumericArrayAttr
(value: number[])
tfjs-converter/src/operations/executors/test_helper.ts:49
↓ 57 callersMethodmean
(axis?: number|number[], keepDims?: boolean)
tfjs-core/src/public/chained_ops/mean.ts:23
↓ 56 callersMethodexecuteOp
( opName: string, opAttrs: TFEOpAttr[], inputTensorIds: number[], numOutputs: number)
tfjs-node/src/tfjs_binding.ts:48
↓ 55 callersMethodfit
* Trains the model for a fixed number of epochs (iterations on a dataset). * * ```js * const model = tf.sequential({ * layers: [tf.layer
tfjs-layers/src/models.ts:867
↓ 55 callersMethodonesLike
(this: T)
tfjs-core/src/public/chained_ops/ones_like.ts:25
↓ 53 callersFunctioncreateDtypeAttr
(dtype: string)
tfjs-converter/src/operations/executors/test_helper.ts:72
↓ 53 callersMethodstridedSlice
( this: Tensor, begin: number[], end: number[], strides: number[], beginMask?: number, endMask
tfjs-core/src/public/chained_ops/strided_slice.ts:25
↓ 52 callersFunctionserializeRegularizer
(constraint: Regularizer)
tfjs-layers/src/regularizers.ts:119
↓ 51 callersMethodindexToLoc
(index: number)
tfjs-core/src/tensor.ts:133
↓ 50 callersMethodapply
( inputs: Tensor | SymbolicTensor, kwargs?: Kwargs )
tfjs-layers/src/layers/nlp/multihead_attention.ts:695
↓ 50 callersMethodsave
(modelArtifacts: ModelArtifacts)
tfjs-core/src/io/http.ts:89
↓ 49 callersMethodgreater
(b: Tensor|TensorLike)
tfjs-core/src/public/chained_ops/greater.ts:23
↓ 49 callersMethodload
* Loads the model and weight files, construct the in memory weight map and * compile the inference graph.
tfjs-converter/src/executor/graph_model.ts:144
↓ 49 callersMethodnext
()
tfjs-data/src/iterators/lazy_iterator.ts:710
↓ 48 callersFunctionconvertTsToPythonic
( tsConfig: serialization.ConfigDictValue, key?: string)
tfjs-layers/src/utils/serialization_utils.ts:90
↓ 48 callersMethodlocToIndex
(locs: number[])
tfjs-core/src/tensor.ts:120
↓ 47 callersFunctioncreateBoolAttr
(value: boolean)
tfjs-converter/src/operations/executors/test_helper.ts:37
↓ 47 callersMethodexpandDims
(axis?: number)
tfjs-core/src/public/chained_ops/expand_dims.ts:23
↓ 47 callersMethodgetSaveHandlers
* Look up IOHandler for saving, given a URL-like string. * * @param url * @returns If only one match is found, an instance of IOHandler with
tfjs-core/src/io/router_registry.ts:70
↓ 47 callersFunctionserializeInitializer
(initializer: Initializer)
tfjs-layers/src/initializers.ts:631
↓ 47 callersMethodsize
* The number of items in the hash table.
tfjs-converter/src/executor/hash_table.ts:60
↓ 46 callersFunctionbuffer
( shape: ShapeMap[R], dtype: D = 'float32' as D, values?: DataTypeMap[D])
tfjs-core/src/ops/buffer.ts:48
↓ 46 callersMethodnotEqual
(b: Tensor|TensorLike)
tfjs-core/src/public/chained_ops/not_equal.ts:23
↓ 46 callersMethodresizeBilinear
( newShape2D: [number, number], alignCorners?: boolean, halfPixelCenters?: boolean)
tfjs-core/src/public/chained_ops/resize_bilinear.ts:23
↓ 45 callersFunctionError
Logs the fact we've found a lint error. We log where the error was found, and also our confidence in the error, that is, how certain we are this
tfjs-backend-wasm/tools/cpplint.py:1216
↓ 45 callersMethodgetClassName
()
tfjs-tflite/src/types/common.ts:26
↓ 44 callersMethodresizeNearestNeighbor
( newShape2D: [number, number], alignCorners?: boolean, halfFloatCenters?: boolean)
tfjs-core/src/public/chained_ops/resize_nearest_neighbor.ts:23
↓ 44 callersMethodtile
(b: number[])
tfjs-core/src/public/chained_ops/tile.ts:24
↓ 43 callersMethodrelu
()
tfjs-core/src/public/chained_ops/relu.ts:23
↓ 42 callersFunctioncreateTensorsAttr
( index: number, paramLength: number)
tfjs-converter/src/operations/executors/test_helper.ts:67
↓ 42 callersMethodgetNumOfTFTensors
()
tfjs-node/src/nodejs_kernel_backend.ts:699
↓ 42 callersMethodgreaterEqual
(b: Tensor|TensorLike)
tfjs-core/src/public/chained_ops/greater_equal.ts:23
↓ 42 callersMethodless
(b: Tensor|TensorLike)
tfjs-core/src/public/chained_ops/less.ts:23
↓ 42 callersMethodsave
* Save the configuration and/or weights of the GraphModel. * * An `IOHandler` is an object that has a `save` method of the proper * signature
tfjs-converter/src/executor/graph_model.ts:280
↓ 42 callersFunctionserializeConstraint
(constraint: Constraint)
tfjs-layers/src/constraints.ts:234
↓ 42 callersFunctiontypeSnippet
(component: number, type = 'f32')
tfjs-backend-webgpu/src/webgpu_program.ts:98
↓ 42 callersFunctionzerosLike
( x: Tensor, dtype?: DataType, name?: string)
tfjs-layers/src/variables.ts:178
↓ 41 callersFunction$
(cmd: string, env: Record<string, string> = {})
scripts/release-util.ts:236
↓ 41 callersMethodlessEqual
(b: Tensor|TensorLike)
tfjs-core/src/public/chained_ops/less_equal.ts:23
↓ 41 callersFunctionslice
( args: {inputs: SliceInputs, backend: MathBackendCPU, attrs: SliceAttrs})
tfjs-backend-cpu/src/kernels/Slice.ts:58
↓ 41 callersMethodsqueeze
(axis?: number[])
tfjs-core/src/public/chained_ops/squeeze.ts:23
↓ 40 callersMethodapply
( decoderSequence: Tensor|SymbolicTensor, kwargs?: TransformerDecoderOptions)
tfjs-layers/src/layers/nlp/modeling/transformer_decoder.ts:311
↓ 40 callersMethodcos
(this: T)
tfjs-core/src/public/chained_ops/cos.ts:25
↓ 40 callersFunctionexpectArrayInMeanStdRange
( actual: TypedArray|number[], expectedMean: number, expectedStdDev: number, epsilon?: number)
tfjs-core/src/ops/rand_util.ts:233
↓ 40 callersFunctionf
(...vals: number[])
tfjs-core/src/ops/local_response_normalization_test.ts:59
↓ 39 callersFunctioncreateDenseModelAndData
( useBias = false, kernelRegularizer?: string|Regularizer, biasRegularizer?: string|Regulari
tfjs-layers/src/engine/training_test.ts:567
↓ 39 callersMethoddispose
()
tfjs-layers/src/layers/core.ts:108
↓ 39 callersFunctionexecuteOp
( node: Node, tensorMap: NamedTensorsMap, context: ExecutionContext)
tfjs-converter/src/operations/executors/control_executor.ts:28
↓ 39 callersFunctionfetch
( path: string, requestInits?: RequestInit)
tfjs-core/src/util.ts:105
↓ 39 callersMethodmaxPool
( filterSize: [number, number]|number, strides: [number, number]|number, pad: 'valid'|'same'|n
tfjs-core/src/public/chained_ops/max_pool.ts:24
↓ 39 callersMethodpop
* Removes the last layer in the model. * * @exception TypeError if there are no layers in the model.
tfjs-layers/src/models.ts:562
↓ 39 callersFunctionrandomUniform
(args: RandomUniformArgs)
tfjs-layers/src/exports_initializers.ts:49
↓ 39 callersMethodsetWeights
(weights: Tensor[])
tfjs-layers/src/layers/wrappers.ts:311
↓ 38 callersMethodencode
* Encode the provided string into an array of bytes using the provided * encoding.
tfjs-core/src/platforms/platform.ts:46
↓ 38 callersMethodmakeTensorInfo
( shape: number[], dtype: DataType, values?: BackendValues|string[])
tfjs-backend-webgpu/src/backend_webgpu.ts:704
↓ 38 callersMethodregisterBackend
( backendName: string, factory: () => KernelBackend | Promise<KernelBackend>, priority = 1)
tfjs-core/src/engine.ts:260
↓ 38 callersMethodwhere
(condition: Tensor|TensorLike, x: Tensor|TensorLike)
tfjs-core/src/public/chained_ops/where.ts:24
↓ 37 callersMethodadd
* Adds a layer instance on top of the layer stack. * * ```js * const model = tf.sequential(); * model.add(tf.layers.dense({units: 8, inp
tfjs-layers/src/models.ts:452
↓ 37 callersMethodatan2
(b: Tensor|TensorLike)
tfjs-core/src/public/chained_ops/atan2.ts:23
↓ 37 callersMethodconv1d
( filter: Tensor3D|TensorLike3D, stride: number, pad: 'valid'|'same'|number|ExplicitPadding, d
tfjs-core/src/public/chained_ops/conv1d.ts:24
↓ 37 callersMethodexecuteSingleInput
* Executes an op that has a single input and output. * * Helper function to wrap executeSingleOutput in a particular case. * @param name The
tfjs-node/src/nodejs_kernel_backend.ts:177
↓ 37 callersFunctiongetConstraint
(identifier: ConstraintIdentifier| serialization.ConfigDict|Constraint)
tfjs-layers/src/constraints.ts:247
← previousnext →101–200 of 6,022, ranked by callers