MCPcopy Create free account

hub / github.com/tensorflow/tfjs / functions

Functions6,022 in github.com/tensorflow/tfjs

↓ 5 callersMethodcall
(inputs: Tensor|Tensor[], kwargs: Kwargs)
tfjs-layers/src/layers/core.ts:265
↓ 5 callersMethodcall
(inputs: Tensor|Tensor[], kwargs: Kwargs)
tfjs-layers/src/layers/recurrent.ts:702
↓ 5 callersMethodcall
( query: Tensor, kwargs: MultiHeadAttentionOptions )
tfjs-layers/src/layers/nlp/multihead_attention.ts:710
↓ 5 callersMethodcallAndPackArgs
* Calls the layer and returns extra information like the paddingMask used to * pack the sequence, the label data, and the sample weights used.
tfjs-layers/src/layers/nlp/models/gpt2/gpt2_preprocessor.ts:205
↓ 5 callersMethodcheckCompileCompletionAsync
()
tfjs-backend-webgpu/src/backend_webgpu.ts:325
↓ 5 callersFunctioncheckPoolMode
(value?: string)
tfjs-layers/src/common.ts:35
↓ 5 callersMethodcloneTensorList
(tensors: Tensor[])
tfjs-converter/src/executor/graph_executor.ts:205
↓ 5 callersMethodcompileAndRun
( program: GPGPUProgram, inputs: TensorInfo[], outputDtype?: DataType, customUniformValues?: numbe
tfjs-backend-webgl/src/backend_webgl.ts:1017
↓ 5 callersFunctioncomputeStrides
(shape: number[])
tfjs-core/src/util_base.ts:565
↓ 5 callersFunctionconvertToTensorArray
( arg: Array<T|TensorLike>, argName: string, functionName: string, parseAsDtype: DataType|'numeric'|'s
tfjs-core/src/tensor_util_env.ts:133
↓ 5 callersFunctioncreateAndConfigureTexture
( gl: WebGLRenderingContext, width: number, height: number, internalFormat: number, textureFormat: num
tfjs-backend-webgl/src/gpgpu_util.ts:53
↓ 5 callersMethodcreateBufferFromTexture
( texture: WebGLTexture, rows: number, columns: number)
tfjs-backend-webgl/src/gpgpu_context.ts:273
↓ 5 callersFunctioncreateComplexTuples
(vals: Array<{}>| TypedArray)
tfjs-core/src/tensor_format.ts:190
↓ 5 callersFunctioncreateDenseCategoricalModelAndData
(useBias = false)
tfjs-layers/src/engine/training_test.ts:580
↓ 5 callersFunctioncreateGPUBufferFromData
( device: GPUDevice, data: number[], dtype: tf.DataType, bufferUsage = GPUBufferUsage.COPY_DST | GPUBu
tfjs-backend-webgpu/src/backend_webgpu_test.ts:510
↓ 5 callersMethoddata
Gets the direct access to the underlying buffer.
tfjs-tflite/src/types/tflite_web_model_runner.ts:125
↓ 5 callersFunctiondescribeAllEnvs
(testName: string, tests: () => void)
tfjs-data/src/util/test_utils.ts:72
↓ 5 callersMethoddispose
()
tfjs-core/src/engine.ts:161
↓ 5 callersMethoddownloadFloat32MatrixFromBuffer
(buffer: WebGLBuffer, size: number)
tfjs-backend-webgl/src/gpgpu_context.ts:268
↓ 5 callersMethodexecuteFunctionAsync
( inputs: Tensor[], tensorArrayMap: TensorArrayMap, tensorListMap: TensorListMap)
tfjs-converter/src/executor/types.ts:26
↓ 5 callersFunctionexecuteInWorker
* Helper function for executing scripts in a webworker. We use * webworkers to get isolated contexts for tests for custom bundles. * * @param progr
e2e/integration_tests/custom_bundle_test.ts:314
↓ 5 callersFunctionexecuteOp
(node: Node, tensorMap: NamedTensorsMap, context: ExecutionContext, ops = tfOps)
tfjs-converter/src/operations/executors/evaluation_executor.ts:29
↓ 5 callersFunctionfingerPrintHash
(hash: Long)
tfjs-core/src/hash_util_test.ts:146
↓ 5 callersFunctionflatten
( arr: T|RecursiveArray<T>, result: T[] = [], skipTypedArray = false)
tfjs-core/src/util.ts:165
↓ 5 callersMethodfloorDiv
(b: Tensor|TensorLike)
tfjs-core/src/public/chained_ops/floorDiv.ts:23
↓ 5 callersMethodgetAsync
(flagName: string)
tfjs-core/src/environment.ts:87
↓ 5 callersMethodgetConfig
()
tfjs-layers/src/layers/advanced_activations.ts:254
↓ 5 callersMethodgetConfig
()
tfjs-layers/src/layers/recurrent.ts:813
↓ 5 callersMethodgetConfig
()
tfjs-layers/src/layers/preprocessing/center_crop.ts:119
↓ 5 callersMethodgetHashTableById
* Get the actual `HashTable` by its handle tensor's id. * @param id The id of the handle tensor.
tfjs-converter/src/executor/resource_manager.ts:55
↓ 5 callersMethodgetImageData
(x: number, y: number, width: number, height: number)
tfjs-core/src/ops/from_pixels_test.ts:23
↓ 5 callersMethodgetInputTensorIds
(tensors: Array<TensorInfo|Int64Scalar>)
tfjs-node/src/nodejs_kernel_backend.ts:129
↓ 5 callersFunctiongetLossOrMetricName
(fn: string|LossOrMetricFn)
tfjs-layers/src/metrics.ts:176
↓ 5 callersFunctiongetRegisteredOp
(name: string)
tfjs-converter/src/operations/custom_op/register.ts:67
↓ 5 callersFunctiongetString
(length: number)
tfjs-core/src/hash_util_test.ts:156
↓ 5 callersFunctiongetTestImageAsTensor4d
()
tfjs-core/src/image_test_util.ts:27
↓ 5 callersFunctiongetY
()
tfjs-layers/src/engine/training_test.ts:75
↓ 5 callersFunctiongradFunc
(dy: tf.Tensor)
tfjs-core/src/gradients_test.ts:324
↓ 5 callersFunctioninferElementShape
( listElementShape: number|number[], tensors: Tensor[], elementShape: number|number[])
tfjs-converter/src/executor/tensor_utils.ts:66
↓ 5 callersFunctioninput_format_string
(base, target_format, detected_format)
tfjs-converter/python/tensorflowjs/converters/wizard.py:342
↓ 5 callersFunctioninterpretAxis
* Interpretable potentially negative axis index. * * For example, given axis = -1, and dim = 3, this function will return 2. * * @param axis The a
tfjs-layers/src/layers/merge.ts:871
↓ 5 callersFunctionisDataTensor
(x: Tensor|Tensor[]|{[inputName: string]: Tensor}| {[inputName: string]: Tensor[]
tfjs-layers/src/engine/training.ts:49
↓ 5 callersFunctionisLocalPath
(source: any)
tfjs-data/src/util/source_util.ts:22
↓ 5 callersFunctionmapActivationToShaderProgram
( activation: backend_util.Activation, packed = false)
tfjs-backend-webgl/src/kernel_utils/kernel_funcs_utils.ts:184
↓ 5 callersFunctionmergePaddingAndAttentionMask
( inputs: Tensor, paddingMask: Tensor, attentionMask: Tensor)
tfjs-layers/src/layers/nlp/modeling/transformer_layer_utils.ts:93
↓ 5 callersFunctionmin
(array: number[]|Float32Array)
tfjs-layers/src/utils/math_utils.ts:61
↓ 5 callersFunctionminimum
(config?: SymbolicTensor[]|Tensor[]|LayerArgs)
tfjs-layers/src/layers/merge.ts:628
↓ 5 callersFunctionmultiply
( args: {inputs: BinaryInputs, backend: MathBackendWebGL})
tfjs-backend-webgl/src/kernels/Multiply.ts:31
↓ 5 callersFunctionnormalizeBatchInTraining
( x: Tensor, gamma: Tensor, beta: Tensor, reductionAxes: number[], epsilon = 1e-3)
tfjs-layers/src/layers/normalization.ts:158
↓ 5 callersFunctiononesLike
( x: Tensor, dtype?: DataType, name?: string)
tfjs-layers/src/variables.ts:206
↓ 5 callersFunctionparseTupleParam
(param: number|number[])
tfjs-core/src/ops/conv_util.ts:395
↓ 5 callersFunctionpool3d
( x: Tensor5D, poolSize: [number, number, number], strides?: [number, number, number], padding?: Paddi
tfjs-layers/src/layers/pooling.ts:95
↓ 5 callersMethodprofileKernel
(kernelName: string, inputs: NamedTensorMap, f: () => Tensor[])
tfjs-core/src/profiler.ts:40
↓ 5 callersFunctionpurgeLocalStorageArtifacts
()
tfjs-core/src/io/local_storage.ts:40
↓ 5 callersFunctionreadWithFillValue
( imageVals: TypedArray, imageHeight: number, imageWidth: number, batchStride: number, rowStride: numb
tfjs-backend-cpu/src/kernels/Transform.ts:200
↓ 5 callersFunctionregisterClass
( cls: SerializableConstructor<T>, pkg?: string, name?: string)
tfjs-core/src/serialization.ts:234
↓ 5 callersMethodreleaseBuffer
(buffer: GPUBuffer, reuse = true)
tfjs-backend-webgpu/src/buffer_manager.ts:62
↓ 5 callersFunctionresetWasmPath
()
tfjs-backend-wasm/src/backend_wasm.ts:508
↓ 5 callersMethodsetOutputMatrixTexture
( outputMatrixTexture: WebGLTexture, rows: number, columns: number)
tfjs-backend-webgl/src/gpgpu_context.ts:420
↓ 5 callersMethodsetPlatform
(platformName: string, platform: Platform)
tfjs-core/src/environment.ts:58
↓ 5 callersFunctionsetWasmPath
(path: string, usePlatformFetch = false)
tfjs-backend-wasm/src/backend_wasm.ts:441
↓ 5 callersFunctionslice
( args: {inputs: SliceInputs, backend: WebGPUBackend, attrs: SliceAttrs})
tfjs-backend-webgpu/src/kernels/Slice.ts:24
↓ 5 callersFunctionslice
( args: {inputs: SliceInputs, backend: MathBackendWebGL, attrs: SliceAttrs})
tfjs-backend-webgl/src/kernels/Slice.ts:54
↓ 5 callersFunctionsliceAlongFirstAxis
( array: Tensor, start: number, size: number)
tfjs-layers/src/backend/tfjs_backend.ts:150
↓ 5 callersFunctionsliceArrays
( arrays: Tensor|Tensor[], start: number, stop: number)
tfjs-layers/src/engine/training_tensors.ts:178
↓ 5 callersMethodstartScope
* Start a scope. Use this with endScope() to achieve the same functionality * as scope() without the need for a function closure.
tfjs-core/src/engine.ts:1054
↓ 5 callersFunctionstep
( args: {backend: BackendWasm, inputs: StepInputs, attrs: StepAttrs})
tfjs-backend-wasm/src/kernels/Step.ts:36
↓ 5 callersMethodstop
()
tfjs-data/src/iterators/webcam_iterator.ts:206
↓ 5 callersFunctiontoArrayBuffer
(buf: Buffer|Buffer[])
tfjs-node/src/io/io_utils.ts:32
↓ 5 callersFunctiontoCamelCase
(identifier: string)
tfjs-layers/src/utils/generic_utils.ts:123
↓ 5 callersFunctiontoNestedArray
( shape: number[], a: TypedArray, isComplex = false)
tfjs-core/src/util_base.ts:601
↓ 5 callersFunctionvalToString
( val: number|string|[number, number], pad: number, dtype: DataType)
tfjs-core/src/tensor_format.ts:69
↓ 5 callersMethodverifyArgs
(args: ConvLayerArgs)
tfjs-layers/src/layers/convolutional.ts:664
↓ 4 callersFunctionCleanseComments
Removes //-comments and single-line C-style /* */ comments. Args: line: A line of C++ source. Returns: The line with single-line comment
tfjs-backend-wasm/tools/cpplint.py:1408
↓ 4 callersFunctionIsHeaderExtension
(file_extension)
tfjs-backend-wasm/tools/cpplint.py:580
↓ 4 callersFunctionPathSplitToList
Returns the path split into a list by the separator. Args: path: An absolute or relative path (e.g. '/a/b/c/' or '../a') Returns: A list
tfjs-backend-wasm/tools/cpplint.py:1784
↓ 4 callersFunctionReverseCloseExpression
If input points to ) or } or ] or >, finds the position that opens it. If lines[linenum][pos] points to a ')' or '}' or ']' or '>', finds the lin
tfjs-backend-wasm/tools/cpplint.py:1718
↓ 4 callersFunctionTensorValue3x4
()
tfjs-core/src/ops/sparse/sparse_segment_sum_test.ts:22
↓ 4 callersFunctionTensorValue3x4Float
()
tfjs-core/src/ops/sparse/sparse_segment_mean_test.ts:22
↓ 4 callersFunction_build_signature_def
(frozen_graph, input_nodes, output_nodes, signature_def=None)
tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2.py:492
↓ 4 callersFunction_ensure_json_dict
(item)
tfjs-converter/python/tensorflowjs/converters/keras_h5_conversion.py:155
↓ 4 callersFunctionapplyActivation
( backend: MathBackendCPU, x: TensorInfo, activation: backend_util.Activation, preluActivationWeights?
tfjs-backend-cpu/src/utils/fused_utils.ts:29
↓ 4 callersMethodas1D
()
tfjs-core/src/public/chained_ops/as1d.ts:24
↓ 4 callersFunctionbatchMatMulImpl
({ a, b, transposeA, transposeB, backend, bias = null, preluActivationWeights = null, leakyrel
tfjs-backend-webgpu/src/kernels/BatchMatMul_impl.ts:43
↓ 4 callersFunctionbatchMatMulImpl
({ a, b, transposeA, transposeB, backend, bias = null, preluActivationWeights = null, leakyrel
tfjs-backend-webgl/src/kernels/BatchMatMul_impl.ts:46
↓ 4 callersFunctionbatchNorm
(args: { inputs: FusedBatchNormInputs, backend: MathBackendCPU, attrs: FusedBatchNormAttrs })
tfjs-backend-cpu/src/kernels/BatchNorm.ts:23
↓ 4 callersFunctionbrowserDownloads
(fileNamePrefix = 'model')
tfjs-core/src/io/browser_files.ts:299
↓ 4 callersMethodbuildFromSignature
* Builds layers and variables. * * Once the method is called, this.builtFromSignature will be set to true.
tfjs-layers/src/layers/nlp/multihead_attention.ts:469
↓ 4 callersFunctionbuildProjectionEquation
* Builds an einsum equation for projections inside multi-head attention.
tfjs-layers/src/layers/nlp/multihead_attention.ts:102
↓ 4 callersFunctionbuildVec
(x: string[])
tfjs-backend-webgl/src/shader_compiler_util.ts:83
↓ 4 callersFunctionbytesToUnicode
()
tfjs-layers/src/layers/nlp/tokenizers_utils.ts:25
↓ 4 callersMethodcallAndReturnCache
* Exactly like `call` except also returns the updated cache.
tfjs-layers/src/layers/nlp/modeling/cached_multihead_attention.ts:117
↓ 4 callersFunctioncheck
(tensor: tf.Tensor, name: string, val: number)
tfjs-backend-wasm/src/index_test.ts:80
↓ 4 callersFunctioncheckArrayLengths
( inputs: Tensor[], targets: Tensor[], weights?: Tensor[])
tfjs-layers/src/engine/training.ts:194
↓ 4 callersFunctioncheckBidirectionalMergeMode
(value?: string)
tfjs-layers/src/layers/wrappers.ts:213
↓ 4 callersFunctioncheckComputationForErrors
( vals: DataTypeMap[D], dtype: D, kernelName: string)
tfjs-core/src/profiler.ts:94
↓ 4 callersFunctioncheckDistribution
(value?: string)
tfjs-layers/src/initializers.ts:25
↓ 4 callersFunctioncheckFanMode
(value?: string)
tfjs-layers/src/initializers.ts:21
↓ 4 callersFunctioncheckGrads
(grads: Tensor[])
tfjs-core/src/gradients.ts:379
↓ 4 callersFunctioncheckInterpolationFormat
(value?: string)
tfjs-layers/src/common.ts:26
← previousnext →801–900 of 6,022, ranked by callers