MCPcopy Create free account

hub / github.com/tensorflow/tfjs / functions

Functions6,022 in github.com/tensorflow/tfjs

↓ 1 callersMethod_getNoiseShape
(inputs: Tensor|Tensor[])
tfjs-layers/src/layers/noise.ts:160
↓ 1 callersFunction_get_affine_quantization_range
Computes quantization range to ensure that zero is represented if covered. Gymnastics with nudged zero point is to ensure that real zero maps to an
tfjs-converter/python/tensorflowjs/quantization.py:175
↓ 1 callersFunction_get_generated_by
(topology)
tfjs-converter/python/tensorflowjs/converters/keras_h5_conversion.py:364
↓ 1 callersFunction_get_requirements
Reads the requirements file and returns the packages
tfjs-converter/python/setup.py:23
↓ 1 callersFunction_get_resource_ids_maps
Generates dictionaries that map tensor names to the loaded saved model resource id, allowing for matching of initializer outputs to inference inp
tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2.py:653
↓ 1 callersFunction_get_resource_initializer_concrete_function
Create a tf.function that creates and initializes all the resources used by the model. For more information on resources, please see the TensorFlow
tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2.py:597
↓ 1 callersFunction_get_weights_manifest_for_group
Gets the weights entries manifest JSON for a group. Args: group: A list of weight entries. Returns: An list of manifest entries (dicts) t
tfjs-converter/python/tensorflowjs/write_weights.py:311
↓ 1 callersFunction_initialize_output_dictionary
Prepopulate required fields for all data foramts. Args: h5file: Valid h5file object. Returns: A dictionary with common fields sets, shar
tfjs-converter/python/tensorflowjs/converters/keras_h5_conversion.py:133
↓ 1 callersFunction_isNavigatorDefined
()
tfjs-core/src/device_util.ts:19
↓ 1 callersFunction_is_assets_required
(model_ops)
tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2.py:398
↓ 1 callersFunction_layers_mobilenet
()
e2e/integration_tests/convert_predict.py:472
↓ 1 callersFunction_load_model
(saved_model_dir, saved_model_tags)
tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2.py:578
↓ 1 callersFunction_parse_metadata_map
(metadata_arg)
tfjs-converter/python/tensorflowjs/converters/converter.py:588
↓ 1 callersFunction_parse_quantization_dtype_map
(float16, uint8, uint16, quantization_bytes)
tfjs-converter/python/tensorflowjs/converters/converter.py:561
↓ 1 callersFunction_quantize_entry
Quantizes the weights in the entry, returning a new entry. The weights are quantized by linearly re-scaling the values between the minimum and ma
tfjs-converter/python/tensorflowjs/write_weights.py:153
↓ 1 callersFunction_remove_unused_control_flow_inputs
(input_graph_def)
tfjs-converter/python/tensorflowjs/converters/tf_saved_model_conversion_v2.py:351
↓ 1 callersMethod_saveRNNKerasModelForTest
(self, path)
tfjs-converter/python/tensorflowjs/converters/keras_tfjs_loader_test.py:60
↓ 1 callersFunction_serialize_numeric_array
Serializes a numeric numpy array into bytes. Args: data: A numeric numpy array. Returns: bytes of the array to be serialized on disk.
tfjs-converter/python/tensorflowjs/write_weights.py:230
↓ 1 callersFunction_serialize_string_array
Serializes a numpy array of dtype `string` into bytes. Each string value is preceded by 4 bytes which denote a 32-bit unsigned integer in little
tfjs-converter/python/tensorflowjs/write_weights.py:195
↓ 1 callersFunction_shard_group_bytes_to_disk
Shards the concatenated bytes for a group to disk. Args: write_dir: The directory to write the files to. group_index: The index for the gro
tfjs-converter/python/tensorflowjs/write_weights.py:277
↓ 1 callersFunction_stack_group_bytes
Stacks the bytes for a weight group into a flat byte array. Args: group: A list of weight entries. Returns: A type: (group_bytes, total_b
tfjs-converter/python/tensorflowjs/write_weights.py:241
↓ 1 callersFunction_standardize_input_output_formats
Standardize input and output formats. Args: input_format: Input format as a string. output_format: Output format as a string. Returns:
tfjs-converter/python/tensorflowjs/converters/converter.py:534
↓ 1 callersMethodabs
(x: T)
tfjs-backend-webgl/src/backend_webgl.ts:766
↓ 1 callersMethodacquireTexture
( width: number, height: number, format: GPUTextureFormat, usage: GPUTextureUsageFlags)
tfjs-backend-webgpu/src/texture_manager.ts:29
↓ 1 callersMethodacquireTexture
( texShape: [number, number], texType: TextureUsage, dtype: DataType, isPacked: boolean)
tfjs-backend-webgl/src/backend_webgl.ts:1203
↓ 1 callersFunctionadd
(args?: LayerArgs)
tfjs-layers/src/exports_layers.ts:710
↓ 1 callersMethodadd
(func)
e2e/benchmarks/browserstack-benchmark/promise_queue.js:9
↓ 1 callersFunctionaddHistogramSelector
( items: string[], parent: HTMLElement, // tslint:disable-next-line:no-any selectionHandler: (item
tfjs-vis/src/show/model.ts:204
↓ 1 callersMethodaddItemToPoll
(isDoneFn: () => boolean, resolveFn: () => void)
tfjs-backend-webgl/src/gpgpu_context.ts:606
↓ 1 callersFunctionaddLoaderElement
* Add a loader element under the tab page. * * @param {string} tabId
e2e/benchmarks/browserstack-benchmark/index.js:500
↓ 1 callersFunctionall
( args: {inputs: AllInputs, backend: MathBackendCPU, attrs: AllAttrs})
tfjs-backend-cpu/src/kernels/All.ts:25
↓ 1 callersFunctionappendRow
(result)
tfjs-backend-webgpu/perf/tune.js:3
↓ 1 callersFunctionargReduce
( backend: MathBackendWebGL, x: TensorInfo, reduceType: 'max'|'min', bestIndicesA: TensorInfo = null)
tfjs-backend-webgl/src/kernel_utils/arg_min_max.ts:25
↓ 1 callersFunctionargReducePacked
( backend: MathBackendWebGL, x: TensorInfo, reduceType: 'max'|'min', bestIndicesA: TensorInfo = null)
tfjs-backend-webgl/src/kernel_utils/arg_min_max.ts:53
↓ 1 callersFunctionarraysEqual
(n1: number[]|TypedArray, n2: number[]|TypedArray)
tfjs-vis/src/util/utils.ts:36
↓ 1 callersFunctionarraysEqualWithNull
(n1: number[], n2: number[])
tfjs-core/src/util_base.ts:197
↓ 1 callersMethodas4D
( rows: number, columns: number, depth: number, depth2: number)
tfjs-core/src/public/chained_ops/as4d.ts:24
↓ 1 callersMethodassertDictContainsSubsetAlmostEqual
(self, d1, d2)
tfjs-converter/python/tensorflowjs/quantization_test.py:27
↓ 1 callersFunctionassertFeedCompatibility
* Helper function to check the dtype and shape compatibility of a feed value.
tfjs-layers/src/engine/executor.ts:28
↓ 1 callersFunctionassertShapesMatch
( shapeA: number[], shapeB: number[], errorMessagePrefix = '')
tfjs-vis/src/util/utils.ts:29
↓ 1 callersFunctionassertTypesMatch
(a: Tensor, b: Tensor)
tfjs-core/src/tensor_util.ts:31
↓ 1 callersFunctionavgPoolGrad
(args: { inputs: AvgPoolGradInputs, backend: MathBackendCPU, attrs: AvgPoolGradAttrs })
tfjs-backend-cpu/src/kernels/AvgPoolGrad.ts:22
↓ 1 callersFunctionbackendEncodeImage
(imageData: Uint8Array)
tfjs-node/src/image.ts:227
↓ 1 callersFunctionbatchConcat
* Assembles a list of same-shaped numbers, number arrays, or Tensors * into a single new Tensor where axis 0 is the batch dimension.
tfjs-data/src/dataset.ts:691
↓ 1 callersFunctionbatchDot
(x: Tensor, y: Tensor, axes: number|[number, number])
tfjs-layers/src/layers/merge.ts:878
↓ 1 callersFunctionbatchMatMul
(args: { inputs: BatchMatMulInputs, attrs: BatchMatMulAttrs, backend: MathBackendCPU })
tfjs-backend-cpu/src/kernels/BatchMatMul.ts:25
↓ 1 callersFunctionbenchmarkCodeSnippet
(benchmarkParameters)
e2e/benchmarks/browserstack-benchmark/benchmark_models.js:91
↓ 1 callersFunctionbenchmarkModel
(benchmarkParameters)
e2e/benchmarks/browserstack-benchmark/benchmark_models.js:61
↓ 1 callersFunctionbilinearInterpolation
( imageVals: TypedArray, imageHeight: number, imageWidth: number, batchStride: number, rowStride: numb
tfjs-backend-cpu/src/kernels/Transform.ts:224
↓ 1 callersFunctionbinaryInsert
( arr: T[], element: T, comparator?: (a: T, b: T) => number)
tfjs-core/src/backends/non_max_suppression_util.ts:31
↓ 1 callersFunctionbinarySearch
( arr: T[], target: T, comparator?: (a: T, b: T) => number)
tfjs-core/src/backends/non_max_suppression_util.ts:53
↓ 1 callersFunctionbinarySearch_
( arr: T[], target: T, comparator: (a: T, b: T) => number)
tfjs-core/src/backends/non_max_suppression_util.ts:69
↓ 1 callersFunctionbincount
(args: { inputs: BincountInputs, backend: MathBackendCPU, attrs: BincountAttrs })
tfjs-backend-cpu/src/kernels/Bincount.ts:23
↓ 1 callersFunctionbincountReduceImpl
( xBuf: TensorBuffer<R>, weightsBuf: TensorBuffer<R>, size: number, binaryOutput = false)
tfjs-backend-cpu/src/kernels/Bincount_impl.ts:46
↓ 1 callersMethodbindKeys
* Binds the ~ (tilde) key to toggle the visor. * * This is called by default when the visor is initially created. * * @doc {heading: 'Viso
tfjs-vis/src/visor.ts:220
↓ 1 callersMethodbindKeys
()
tfjs-vis/src/components/visor.tsx:144
↓ 1 callersFunctionbindTextureUnit
( gl: WebGLRenderingContext, texture: WebGLTexture, textureUnit: number)
tfjs-backend-webgl/src/webgl_util.ts:246
↓ 1 callersMethodbpeMerge
* Perform byte-pair merge for each word in the inputs.
tfjs-layers/src/layers/nlp/tokenizers.ts:444
↓ 1 callersMethodbpeMergeAndUpdateCache
* Process unseen tokens and add to cache.
tfjs-layers/src/layers/nlp/tokenizers.ts:479
↓ 1 callersMethodbpeMergeOneStep
* Perform one step of byte-pair merge.
tfjs-layers/src/layers/nlp/tokenizers.ts:348
↓ 1 callersFunctionbroadcastNormalizeBatchInTraining
* Broadcasting batch normalization for use in training (not inference). * * The input is normalized to zero mean and unit variance along the * `red
tfjs-layers/src/layers/normalization.ts:119
↓ 1 callersMethodbuckets
* Group data into histogram buckets. * * @param data A `Tensor` of any shape. Must be castable to `float32` * @param bucketCount Optional pos
tfjs-node/src/nodejs_kernel_backend.ts:630
↓ 1 callersFunctionbuild
* Calls node-gyp for Node.js Tensorflow binding after lib is downloaded.
tfjs-node/scripts/install.js:171
↓ 1 callersMethodbuild
(inputShape: Shape|Shape[])
tfjs-layers/src/layers/recurrent.ts:491
↓ 1 callersMethodbuild
* * @param inputShape decoderSequenceShape or * [decoderSequenceShape, encoderSequenceShape]
tfjs-layers/src/layers/nlp/modeling/transformer_decoder.ts:252
↓ 1 callersMethodbuildAttention
* Builds multi-head dot-product attention computations. * * This function builds attributes necessary for `computeAttention` to * customize a
tfjs-layers/src/layers/nlp/multihead_attention.ts:599
↓ 1 callersFunctionbuildAttentionEquation
* Builds einsum equations for the attention computation. * * Query, key, value inputs after projection are expected to have the shape as: * `(bs, <
tfjs-layers/src/layers/nlp/multihead_attention.ts:60
↓ 1 callersFunctionbuildDenseSpec
( sparse: StridedSliceSparseSpec, dense: StridedSliceDenseSpec)
tfjs-core/src/ops/slice_util.ts:631
↓ 1 callersFunctionbuildDetectedObjects
( width: number, height: number, boxes: Float32Array, boxScores: number[], boxLabels: number[], select
tfjs-automl/src/object_detection.ts:150
↓ 1 callersFunctionbuildMapOfGraph
(tensor: SymbolicTensor, finishedNodes: Node[], nodesInProgress: Node[], layer?: Layer, nodeIndex?: n
tfjs-layers/src/engine/container.ts:262
↓ 1 callersFunctionbytesFromStringArray
(arr: Uint8Array[])
tfjs-core/src/util_base.ts:510
↓ 1 callersFunctionbytesPerElement
(dtype: DataType)
tfjs-core/src/util_base.ts:492
↓ 1 callersFunctioncacheUrl
(url: string)
tfjs-backend-wasm/src/test_util.ts:24
↓ 1 callersMethodcalculateFirstParentOutputIndex
* The outputIndex represents the index in the output tensor * where the first element of a particular dimension would be written. * If it is -1,
tfjs-backend-cpu/src/kernels/RaggedTensorToTensor_impl.ts:151
↓ 1 callersFunctioncalculateMostLikelyLabels
( scores: Float32Array, numBoxes: number, numClasses: number)
tfjs-automl/src/object_detection.ts:126
↓ 1 callersMethodcalculateOutputIndex
( dimension: number, parentOutputIndex: number[], outputIndexMultiplier: number, outputSize: numbe
tfjs-backend-cpu/src/kernels/RaggedTensorToTensor_impl.ts:273
↓ 1 callersMethodcalculateOutputIndexRowSplit
( rowSplit: TypedArray, parentOutputIndex: number[], outputIndexMultiplier: number, outputSize: nu
tfjs-backend-cpu/src/kernels/RaggedTensorToTensor_impl.ts:171
↓ 1 callersMethodcalculateOutputIndexValueRowID
( valueRowIds: TypedArray, parentOutputIndex: number[], outputIndexMultiplier: number, outputSize:
tfjs-backend-cpu/src/kernels/RaggedTensorToTensor_impl.ts:220
↓ 1 callersMethodcalculateOutputSize
(firstDim: number)
tfjs-backend-cpu/src/kernels/RaggedTensorToTensor_impl.ts:117
↓ 1 callersFunctioncalculateResultSnippet
(transposeA: boolean, innerElementSize: number, rowPerThread: number, tileInner: number)
tfjs-backend-webgpu/src/matmul_packed_webgpu.ts:109
↓ 1 callersFunctioncall
(outputs: Tensor, cache: Tensor)
tfjs-layers/src/layers/nlp/modeling/transformer_decoder_test.ts:148
↓ 1 callersFunctioncall
(outputs: Tensor, cache: Tensor)
tfjs-layers/src/layers/nlp/modeling/cached_multihead_attention_test.ts:61
↓ 1 callersMethodcall
(inputs: Tensor|Tensor[], kwargs: Kwargs)
tfjs-layers/src/layers/wrappers.ts:436
↓ 1 callersMethodcall
( inputs: Tensor|Tensor[], kwargs: GPT2PreprocessorOptions)
tfjs-layers/src/layers/nlp/models/gpt2/gpt2_preprocessor.ts:162
↓ 1 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_causal_lm_preprocessor.ts:80
↓ 1 callersFunctioncameraWithTensors
( // tslint:disable-next-line: variable-name CameraComponent: React.ComponentType<T> )
tfjs-react-native/src/camera/camera_stream.tsx:173
↓ 1 callersFunctioncanWrapPixelsToImageBitmap
(pixels: PixelData|ImageData| HTMLImageElement|HTMLCanvasElement|
tfjs-core/src/ops/browser.ts:194
↓ 1 callersFunctioncartoonize
( tfliteModel: tflite.TFLiteModel, ele: HTMLImageElement|HTMLVideoElement)
tfjs-tflite/demo/src/script.ts:115
↓ 1 callersFunctioncastImpl
( values: TypedArray, shape: number[], inputType: DataType, dtype: DataType)
tfjs-backend-cpu/src/kernels/Cast.ts:27
↓ 1 callersFunctioncategoricalCrossentropy
( target: Tensor, output: Tensor, fromLogits = false)
tfjs-layers/src/losses.ts:110
↓ 1 callersMethodcenterCrop
(inputs: Tensor3D | Tensor4D, hBuffer: number, wBuffer: number, height: number, width: number, inp
tfjs-layers/src/layers/preprocessing/center_crop.ts:36
↓ 1 callersFunctioncenterCropAndResize
Center crops an image
tfjs-automl/src/img_classification.ts:83
↓ 1 callersFunctioncheckAllSymbolic
(tensors: MaybeSymbolic|MaybeSymbolic[])
tfjs-layers/src/engine/topology.ts:1661
↓ 1 callersFunctioncheckBrowserStackAccount
()
e2e/benchmarks/browserstack-benchmark/app.js:42
↓ 1 callersMethodcheckCompileCompletion
()
tfjs-backend-webgl/src/backend_webgl.ts:1222
↓ 1 callersMethodcheckCompletionAsync_
(binary: GPGPUBinary)
tfjs-backend-webgl/src/backend_webgl.ts:1251
↓ 1 callersFunctioncheckFn
()
tfjs-core/src/util_test.ts:268
↓ 1 callersMethodcheckForDuplicate
(callbackConstructor: BaseCallbackConstructor)
tfjs-layers/src/base_callbacks.ts:540
↓ 1 callersMethodcheckForLossMismatch
( loss: string|string[]|{[outputName: string]: string}|LossOrMetricFn| LossOrMetricFn[]|{[output
tfjs-layers/src/layers/nlp/models/task.ts:44
↓ 1 callersFunctioncheckFraction
(startFraction: number, endFraction: number)
tfjs-core/src/io/progress.ts:56
← previousnext →1,901–2,000 of 6,022, ranked by callers