Avorter | Déclenche une exception pour abandonner le processus lorsqu'il est appelé. |
Tous | Calcule le "et logique" des éléments à travers les dimensions d'un tenseur. |
ToutÀTous <T> | Une opération pour échanger des données entre des répliques de TPU. |
Table de hachage anonyme | Crée une table de hachage anonyme non initialisée. |
AnonymousIteratorV2 | Un conteneur pour une ressource d'itérateur. |
AnonymousIteratorV3 | Un conteneur pour une ressource d'itérateur. |
AnonymousMemoryCache | |
AnonymousMultiDeviceIteratorAnonymousMultiDeviceIteratorAnonymousMultiDeviceIterator | Un conteneur pour une ressource d'itérateur multi-périphérique. |
AnonymousMultiDeviceIteratorV3 | Un conteneur pour une ressource d'itérateur multi-périphérique. |
AnonymousMutableDenseHashTable | Crée une table de hachage mutable anonyme vide qui utilise des tenseurs comme magasin de sauvegarde. |
AnonymousMutableHashTable | Crée une table de hachage mutable anonyme vide. |
AnonymousMutableHashTableOfTensors | Crée une table de hachage mutable anonyme vide de valeurs vectorielles. |
AnonymousRandomSeedGenerator | |
AnonymousSeedGenerator | |
N'importe quel | Calcule le "ou logique" des éléments à travers les dimensions d'un tenseur. |
AppliquerAdagradV2 <T> | Mettez à jour '*var' selon le schéma adagrad. |
ApproxTopK <T étend le nombre> | Renvoie les valeurs min/max k et leurs indices de l'opérande d'entrée de manière approximative. |
AssertCardinalityDataset | |
AssertNextDataset | Une transformation qui affirme quelles transformations se produisent ensuite. |
AssertPrevDataset | Une transformation qui affirme quelles transformations se sont produites précédemment. |
Affirmer que | Affirme que la condition donnée est vraie. |
Affecter <T> | Mettez à jour 'ref' en lui attribuant 'value'. |
AffecterAjouter <T> | Mettez à jour 'ref' en y ajoutant 'value'. |
AssignAddVariableOp | Ajoute une valeur à la valeur actuelle d'une variable. |
AssignSub <T> | Mettez à jour 'ref' en soustrayant 'value'. |
AssignSubVariableOp | Soustrait une valeur de la valeur actuelle d'une variable. |
AssignVariableOp | Affecte une nouvelle valeur à une variable. |
AssignVariableXlaConcatND | Concats le tenseur d'entrée dans toutes les dimensions. |
AutoShardDataset | Crée un jeu de données qui fragmente le jeu de données d'entrée. |
BandedTriangularSolve <T> | |
Barrière | Définit une barrière qui persiste à travers différentes exécutions de graphique. |
BarrièreFermer | Ferme la barrière donnée. |
TailleBarrièreIncomplète | Calcule le nombre d'éléments incomplets dans la barrière donnée. |
BarrièreInsérerBeaucoup | Pour chaque clé, attribue la valeur respective au composant spécifié. |
TailleBarrièrePrêt | Calcule le nombre d'éléments complets dans la barrière donnée. |
BarrièrePrendreBeaucoup | Prend le nombre donné d'éléments terminés d'une barrière. |
Grouper | Regroupe tous les tenseurs d'entrée de manière non déterministe. |
BatchMatMulV2 <T> | Multiplie les tranches de deux tenseurs par lots. |
BatchMatMulV3 <V> | Multiplie les tranches de deux tenseurs par lots. |
LotVersEspace <T> | BatchToSpace pour les tenseurs 4-D de type T. |
LotVersEspaceNd <T> | BatchToSpace pour les tenseurs ND de type T. |
BesselI0 <T étend le nombre> | |
BesselI1 <T prolonge le nombre> | |
BesselJ0 <T prolonge le nombre> | |
BesselJ1 <T prolonge le nombre> | |
BesselK0 <T prolonge le nombre> | |
BesselK0e <T prolonge le nombre> | |
BesselK1 <T prolonge le nombre> | |
BesselK1e <T prolonge le nombre> | |
BesselY0 <T prolonge le nombre> | |
BesselY1 <T prolonge le nombre> | |
Bitcast <U> | Bitcaste un tenseur d'un type à un autre sans copier les données. |
BlockLSTM <T étend le nombre> | Calcule la propagation directe des cellules LSTM pour tous les pas de temps. |
BlockLSTMGrad <T étend le nombre> | Calcule la propagation vers l'arrière de la cellule LSTM pour toute la séquence temporelle. |
BlockLSTMGradV2 <T étend le nombre> | Calcule la propagation vers l'arrière de la cellule LSTM pour toute la séquence temporelle. |
BlockLSTMV2 <T étend le nombre> | Calcule la propagation directe des cellules LSTM pour tous les pas de temps. |
BoostedTreesStatistiques agrégées | Regroupe le résumé des statistiques accumulées pour le lot. |
BoostedTreesBucketize | Regroupez chaque fonctionnalité en fonction des limites du compartiment. |
BoostedTreesCalculateBestFeatureSplit | Calcule les gains pour chaque fonctionnalité et renvoie les meilleures informations de fractionnement possibles pour la fonctionnalité. |
BoostedTreesCalculateBestFeatureSplitV2 | Calcule les gains pour chaque fonctionnalité et renvoie les meilleures informations de fractionnement possibles pour chaque nœud. |
BoostedTreesCalculateBestGainsPerFeature | Calcule les gains pour chaque fonctionnalité et renvoie les meilleures informations de fractionnement possibles pour la fonctionnalité. |
BoostedTreesCenterBias | Calcule l'a priori à partir des données d'apprentissage (le biais) et remplit le premier nœud avec l'a priori des logits. |
BoostedTreesCreateEnsemble | Crée un modèle d'ensemble d'arbres et lui renvoie un handle. |
BoostedTreesCreateQuantileStreamResource | Créez la ressource pour les flux quantiles. |
BoostedTreesDeserializeEnsemble | Désérialise une configuration d'ensemble d'arborescence sérialisée et remplace l'arborescence actuelle ensemble. |
BoostedTreesEnsembleResourceHandleOp | Crée un handle vers un BoostedTreesEnsembleResource |
BoostedTreesExampleDebugOutputs | Sorties de débogage/d'interprétabilité du modèle pour chaque exemple. |
BoostedTreesFlushQuantileSummaries | Videz les résumés quantiles de chaque ressource de flux quantile. |
BoostedTreesGetEnsembleStates | Récupère le jeton de tampon de ressource d'ensemble d'arbres, le nombre d'arbres et les statistiques de croissance. |
BoostedTreesMakeQuantileSummaries | Fait le résumé des quantiles pour le lot. |
BoostedTreesMakeStatsSummary | Fait le résumé des statistiques accumulées pour le lot. |
BoostedTreesPredict | Exécute plusieurs prédicteurs d'ensemble de régression additive sur les instances d'entrée et calcule les logits. |
BoostedTreesQuantileStreamResourceAddSummaries | Ajoutez les résumés quantiles à chaque ressource de flux quantile. |
BoostedTreesQuantileStreamResourceDeserialize | Désérialisez les limites de compartiment et l'indicateur prêt dans le QuantileAccumulator actuel. |
BoostedTreesQuantileStreamResourceFlush | Videz les récapitulatifs d'une ressource de flux quantile. |
BoostedTreesQuantileStreamResourceGetBucketBoundaries | Générez les limites de compartiment pour chaque fonctionnalité en fonction des résumés accumulés. |
BoostedTreesQuantileStreamResourceHandleOp | Crée un handle vers un BoostedTreesQuantileStreamResource. |
BoostedTreesSerializeEnsemble | Sérialise l'ensemble d'arbres en un proto. |
BoostedTreesSparseAggregateStats | Regroupe le résumé des statistiques accumulées pour le lot. |
BoostedTreesSparseCalculateBestFeatureSplit | Calcule les gains pour chaque fonctionnalité et renvoie les meilleures informations de fractionnement possibles pour la fonctionnalité. |
BoostedTreesFormationPrédire | Exécute plusieurs prédicteurs d'ensemble de régression additive sur les instances d'entrée et calcule la mise à jour des logits mis en cache. |
BoostedTreesUpdateEnsemble | Met à jour l'ensemble d'arbres en ajoutant une couche au dernier arbre en cours de croissance ou en démarrant un nouvel arbre. |
BoostedTreesUpdateEnsembleV2 | Met à jour l'ensemble d'arbres en ajoutant une couche au dernier arbre en cours de croissance ou en démarrant un nouvel arbre. |
BroadcastDynamicShape <T étend le nombre> | Renvoie la forme de s0 op s1 avec diffusion. |
BroadcastGradientArgs <T étend le nombre> | Renvoie les indices de réduction pour calculer les gradients de s0 op s1 avec diffusion. |
Diffuser vers <T> | Diffusez un tableau pour une forme compatible. |
Regrouper | Regroupe les « entrées » en fonction des « limites ». |
CSRSparseMatrixComponents <T> | Lit les composants CSR au lot `index`. |
CSRSparseMatrixToDense <T> | Convertir un CSRSparseMatrix (éventuellement groupé) en dense. |
CSRSparseMatrixToSparseTensor <T> | Convertit un CSRSparesMatrix (éventuellement par lot) en un SparseTensor. |
CSVDataset | |
CSVDatasetV2 | |
Perte CCTV2 | Calcule la perte CTC (probabilité logarithmique) pour chaque entrée de lot. |
CacheDatasetV2 | |
CheckNumericsV2 <T étend le nombre> | Vérifie un tenseur pour les valeurs NaN, -Inf et +Inf. |
Choisir le jeu de données le plus rapide | |
ClipParValeur <T> | Découpe les valeurs de tenseur à un minimum et un maximum spécifiés. |
CollateTPUEmbeddingMemory | Une opération qui fusionne les protos de configuration de la mémoire codés en chaîne de tous les hôtes. |
CollectiveAllToAllV2 <T étend Number> | Échange mutuellement plusieurs tenseurs de type et de forme identiques. |
CollectiveAllToAllV3 <T étend le nombre> | Échange mutuellement plusieurs tenseurs de type et de forme identiques. |
CollectiveAssignGroupV2 | Attribuez des clés de groupe en fonction de l'attribution de groupe. |
CollectiveBcastRecvV2 <U> | Reçoit une valeur de tenseur diffusée depuis un autre appareil. |
CollectiveBcastSendV2 <T> | Diffuse une valeur de tenseur vers un ou plusieurs autres appareils. |
CollectiveGather <T étend le nombre> | Accumule mutuellement plusieurs tenseurs de type et de forme identiques. |
CollectiveGatherV2 <T étend le nombre> | Accumule mutuellement plusieurs tenseurs de type et de forme identiques. |
CollectiveInitializeCommunicator | Initialise un groupe pour les opérations collectives. |
CollectivePermute <T> | Une opération pour permuter les Tensors entre les instances de TPU répliquées. |
CollectiveReduceScatterV2 <T étend le nombre> | Réduit mutuellement plusieurs tenseurs de type et de forme identiques et disperse le résultat. |
CollectiveReduceV2 <T étend le nombre> | Réduit mutuellement plusieurs tenseurs de type et de forme identiques. |
CollectiveReduceV3 <T étend le nombre> | Réduit mutuellement plusieurs tenseurs de type et de forme identiques. |
CombinéNonMaxSuppression | Sélectionne avidement un sous-ensemble de boîtes englobantes dans l'ordre décroissant du score, Cette opération effectue non_max_suppression sur les entrées par lot, dans toutes les classes. |
CompositeTensorVariantFromComponents | Encode une valeur `ExtensionType` dans un Tensor scalaire `variant`. |
CompositeTensorVariantToComponents | Décode un Tensor scalaire `variant` en une valeur `ExtensionType`. |
CompressElement | Compresse un élément de jeu de données. |
ComputeBatchSize | Calcule la taille de lot statique d'un ensemble de données sans lots partiels. |
ComputeDedupDataTupleMask | Un op calcule le masque de tuple des données de déduplication à partir du noyau d'intégration. |
Concat <T> | Concatène les tenseurs le long d'une dimension. |
ConfigureAndInitializeGlobalTPU | Une opération qui met en place les structures centralisées d'un système TPU distribué. |
ConfigureDistributedTPU | Configure les structures centralisées pour un système TPU distribué. |
ConfigureTPUEmbedding | Configure TPUEmbedding dans un système TPU distribué. |
ConfigureTPUEmbeddingHost | Op qui configure le logiciel TPUEmbedding sur un hôte. |
ConfigureTPUEmbeddingMemory | Op qui configure le logiciel TPUEmbedding sur un hôte. |
ConnectTPUEmbeddingHosts | Une opération qui configure la communication entre les instances du logiciel hôte TPUEmbedding après que ConfigureTPUEmbeddingHost a été appelé sur chaque hôte. |
Constante <T> | Un opérateur produisant une valeur constante. |
ConsumeMutexLock | Cette opération consomme un verrou créé par `MutexLock`. |
ControlTrigger | Ne fait rien. |
Conv2DBackpropFilterV2 <T étend le nombre> | Calcule les gradients de convolution par rapport au filtre. |
Conv2DBackpropInputV2 <T étend le nombre> | Calcule les gradients de convolution par rapport à l'entrée. |
Copier <T> | Copiez un tenseur de CPU à CPU ou de GPU à GPU. |
CopierHôte <T> | Copiez un tenseur sur l'hôte. |
Copier vers maillage <T> | |
CopyToMeshGrad <T> | |
CountUpTo <T étend le nombre> | Incrémente 'ref' jusqu'à ce qu'il atteigne 'limit'. |
CrossReplicaSum <T étend le nombre> | Une opération pour additionner les entrées sur les instances de TPU répliquées. |
CudnnRNNBackpropV3 <T étend le nombre> | Étape Backprop de CudnnRNNV3. |
CudnnRNNCanonicalToParamsV2 <T étend le nombre> | Convertit les paramètres CudnnRNN de la forme canonique à la forme utilisable. |
CudnnRNNParamsToCanonicalV2 <T étend le nombre> | Récupère les paramètres CudnnRNN sous forme canonique. |
CudnnRNNV3 <T étend le nombre> | Un RNN soutenu par cuDNN. |
CumulativeLogsumexp <T étend le nombre> | Calculez le produit cumulé du tenseur "x" le long de "l'axe". |
DTensorRestoreV2 | |
DTensorSetGlobalTPUArray | Une opération qui informe un hôte des identifiants globaux de tous les TPU du système. |
DataServiceDataset | Crée un ensemble de données qui lit les données du service tf.data. |
DataServiceDatasetV2 | Crée un ensemble de données qui lit les données du service tf.data. |
Ensemble de donnéesCardinalité | Renvoie la cardinalité de `input_dataset`. |
DatasetFromGraph | Crée un jeu de données à partir du `graph_def` donné. |
DatasetToGraphV2 | Renvoie un GraphDef sérialisé représentant `input_dataset`. |
Dawsn <T prolonge le nombre> | |
DebugGradientIdentity <T> | Opération d'identité pour le débogage de gradient. |
DebugGradientRefIdentity <T> | Opération d'identité pour le débogage de gradient. |
DebugIdentity <T> | Fournit un mappage d'identité du tenseur d'entrée de type non Ref pour le débogage. |
DebugIdentityV2 <T> | Debug Identity V2 Op. |
DebugNanCount | Débogage NaN Valeur Compteur Op. |
Résumé numérique de débogage | Op. récapitulatif numérique de débogage. |
DebugNumericSummaryV2 <U étend le nombre> | Résumé numérique de débogage V2 Op. |
DecodeImage <T étend le nombre> | Fonction pour decode_bmp, decode_gif, decode_jpeg et decode_png. |
DecodePaddedRaw <T étend le nombre> | Réinterprète les octets d'une chaîne comme un vecteur de nombres. |
DécoderProto | L'op extrait les champs d'un message de tampons de protocole sérialisé en tenseurs. |
DeepCopy <T> | Crée une copie de `x`. |
SupprimerItérateur | Un conteneur pour une ressource d'itérateur. |
Supprimer le cache de mémoire | |
SupprimerMultiDeviceIterator | Un conteneur pour une ressource d'itérateur. |
SupprimerRandomSeedGenerator | |
SupprimerSeedGenerator | |
DeleteSessionTensor | Supprime le tenseur spécifié par son handle dans la session. |
DenseBincount <U étend le nombre> | Compte le nombre d'occurrences de chaque valeur dans un tableau d'entiers. |
DenseCountSparseOutput <U étend le nombre> | Effectue un comptage de bacs de sortie clairsemé pour une entrée tf.tensor. |
DenseToCSRSparseMatrixDenseToCSRSparseMatrix | Convertit un tenseur dense en un CSRSparseMatrix (éventuellement groupé). |
DestroyResourceOp | Supprime la ressource spécifiée par le handle. |
DétruireVariableTemporaire <T> | Détruit la variable temporaire et renvoie sa valeur finale. |
DeviceIndex | Renvoie l'index du périphérique exécuté par l'opération. |
DirectedInterleaveDatasetDirectedInterleaveDatasetDirectedInterleaveDataset | Un substitut à `InterleaveDataset` sur une liste fixe de `N` ensembles de données. |
Désactiver la copie en lecture | Désactive le mode copie en lecture. |
DistribuéEnregistrer | |
DrawBoundingBoxesV2 <T étend le nombre> | Dessinez des cadres de délimitation sur un lot d'images. |
DummyIterationCounter | |
DummyMemoryCache | |
DummySeedGenerator | |
DynamicEnqueueTPUEmbeddingArbitraryTensorBatch | Facilite le portage du code qui utilise tf.nn.embedding_lookup_sparse(). |
DynamicPartition <T> | Partitionne les "données" en tenseurs "num_partitions" en utilisant les indices des "partitions". |
Point Dynamique <T> | Entrelacez les valeurs des tenseurs "data" dans un seul tenseur. |
ModifierDistance | Calcule la distance d'édition de Levenshtein (éventuellement normalisée). |
Eig <U> | Calcule la décomposition propre d'une ou plusieurs matrices carrées. |
Einsum <T> | Contraction du tenseur selon la convention de sommation d'Einstein. |
Vide <T> | Crée un tenseur avec la forme donnée. |
EmptyTensorList | Crée et renvoie une liste de tenseurs vide. |
EmptyTensorMap | Crée et renvoie une carte de tenseurs vide. |
EncodeProto | L'op sérialise les messages protobuf fournis dans les tenseurs d'entrée. |
EnqueueTPUEmbeddingArbitraryTensorBatch | Facilite le portage du code qui utilise tf.nn.embedding_lookup_sparse(). |
EnqueueTPUEmbeddingBatch | Opération qui met en file d'attente une liste de Tensors de lot d'entrée dans TPUEmbedding. |
EnqueueTPUEmbeddingIntegerBatch | Opération qui met en file d'attente une liste de Tensors de lot d'entrée dans TPUEmbedding. |
EnqueueTPUEmbeddingRaggedTensorBatch | Facilite le portage du code qui utilise tf.nn.embedding_lookup(). |
EnqueueTPUEmbeddingSparseBatch | Une opération qui met en file d'attente les index d'entrée TPUEmbedding à partir d'un SparseTensor. |
EnqueueTPUEmbeddingSparseTensorBatch | Facilite le portage du code qui utilise tf.nn.embedding_lookup_sparse(). |
AssurerForme <T> | Garantit que la forme du tenseur correspond à la forme attendue. |
Entrez <T> | Crée ou trouve un cadre enfant et met les "données" à la disposition du cadre enfant. |
Erfinv <T étend le nombre> | |
Norme Euclidienne <T> | Calcule la norme euclidienne des éléments sur les dimensions d'un tenseur. |
ExécuterTPUEmbeddingPartitioner | Une opération qui exécute le partitionneur TPUEmbedding sur la configuration centrale périphérique et calcule la taille HBM (en octets) requise pour l'opération TPUEmbedding. |
Quitter <T> | Quitte le cadre en cours vers son cadre parent. |
DévelopperDims <T> | Insère une dimension de 1 dans la forme d'un tenseur. |
ExperimentalAutoShardDatasetExperimentalAutoShardDataset | Crée un jeu de données qui fragmente le jeu de données d'entrée. |
ExperimentalBytesProducedStatsDataset | Enregistre la taille en octets de chaque élément de `input_dataset` dans un StatsAggregator. |
ExperimentalChooseFastestDataset | |
Cardinalité expérimentale de l'ensemble de données | Renvoie la cardinalité de `input_dataset`. |
ExperimentalDatasetToTFRecordExperimentalDatasetToTFRecordExperimentalDatasetToTFRecordExperimentalDatasetToTFRecord | Écrit l'ensemble de données donné dans le fichier donné en utilisant le format TFRecord. |
ExperimentalDenseToSparseBatchDatasetExperimentalDenseToSparseBatchDataset | Crée un ensemble de données qui regroupe les éléments d'entrée dans un SparseTensor. |
ExperimentalLatencyStatsDataset | Enregistre la latence de production des éléments `input_dataset` dans un StatsAggregator. |
ExperimentalMatchingFilesDatasetExperimentalMatchingFilesDataset | |
ExperimentalMaxIntraOpParallelismDataset | Crée un jeu de données qui remplace le parallélisme intra-op maximal. |
ExperimentalParseExampleDatasetExperimentalParseExampleDatasetExperimentalParseExampleDataset | Transforme `input_dataset` contenant des protos `Example` en tant que vecteurs de DT_STRING en un ensemble de données d'objets `Tensor` ou `SparseTensor` représentant les entités analysées. |
ExperimentalPrivateThreadPoolDatasetExperimentalPrivateThreadPoolDataset | Crée un ensemble de données qui utilise un pool de threads personnalisé pour calculer `input_dataset`. |
ExperimentalRandomDataset | Crée un ensemble de données qui renvoie des nombres pseudo-aléatoires. |
ExperimentalRebatchDatasetExperimentalRebatchDataset | Crée un jeu de données qui modifie la taille du lot. |
ExperimentalSetStatsAggregatorDatasetExperimentalSetStatsAggregatorDatasetExperimentalSetStatsAggregatorDataset | |
ExperimentalSlidingWindowDataset | Crée un ensemble de données qui passe une fenêtre glissante sur `input_dataset`. |
ExperimentalSqlDatasetExperimentalSqlDataset | Crée un jeu de données qui exécute une requête SQL et émet des lignes du jeu de résultats. |
ExperimentalStatsAggregatorHandleExperimentalStatsAggregatorHandleExperimentalStatsAggregatorHandle | Crée une ressource de gestionnaire de statistiques. |
ExperimentalStatsAggregatorSummary | Produit un résumé de toutes les statistiques enregistrées par le gestionnaire de statistiques donné. |
ExperimentalUnbatchDatasetExperimentalUnbatchDatasetExperimentalUnbatchDataset | Un ensemble de données qui divise les éléments de son entrée en plusieurs éléments. |
Expint <T prolonge le nombre> | |
ExtraireGlimpseV2 | Extrait un aperçu du tenseur d'entrée. |
ExtractVolumePatches <T étend le nombre> | Extrayez les `patchs` de `input` et placez-les dans la dimension de sortie `"depth"`. |
FileSystemSetConfigurationFileSystemSetConfiguration | Définir la configuration du système de fichiers. |
Remplir <U> | Crée un tenseur rempli d'une valeur scalaire. |
Finaliser l'ensemble de données | Crée un ensemble de données en appliquant tf.data.Options à `input_dataset`. |
FinaliserTPUEmbedding | Une opération qui finalise la configuration de TPUEmbedding. |
Empreinte digitale | Génère des valeurs d'empreintes digitales. |
FresnelCos <T étend le nombre> | |
FresnelSin <T étend le nombre> | |
FusedBatchNormGradV3 <T étend le nombre, U étend le nombre> | Gradient pour la normalisation par lots. |
FusedBatchNormV3 <T étend le nombre, U étend le nombre> | Normalisation par lots. |
GRUBlockCell <T étend le nombre> | Calcule la propagation directe des cellules GRU pour 1 pas de temps. |
GRUBlockCellGrad <T étend le nombre> | Calcule la rétropropagation des cellules GRU pour 1 pas de temps. |
Rassemblez <T> | Rassemblez les tranches de l'axe `params` `axis` en fonction des `indices`. |
RassemblerNd <T> | Rassemblez les tranches de `params` dans un Tensor avec la forme spécifiée par `indices`. |
GenerateBoundingBoxProposalsGenerateBoundingBoxProposals | Cette opération produit une région d'intérêts à partir de boîtes englobantes données (bbox_deltas) encodées par rapport aux ancres selon l'eq.2 dans arXiv : 1506.01497 L'op sélectionne les boîtes de notation supérieures `pre_nms_topn`, les décode en fonction des ancres, applique une suppression non maximale sur les boîtes qui se chevauchent avec une valeur d'intersection sur union (iou) supérieure à `nms_threshold`, en supprimant les boîtes où le côté le plus court est inférieur à ` taille_min`. |
GetElementAtIndex | Obtient l'élément à l'index spécifié dans un ensemble de données. |
ObtenirOptions | Renvoie les tf.data.Options attachées à `input_dataset`. |
GetSessionHandle | Stocke le tenseur d'entrée dans l'état de la session en cours. |
GetSessionTensor <T> | Récupère la valeur du tenseur spécifié par son handle. |
Dégradés | Ajoute des opérations pour calculer les dérivées partielles de la somme de y s par rapport à x s, c'est-à d(y_1 + y_2 + ...)/dx_1, d(y_1 + y_2 + ...)/dx_2... Si les valeurs Options.dx() sont définies, elles correspondent aux dérivées partielles symboliques initiales d'une fonction de perte L wrt |
GarantieConst <T> | Donne une garantie au runtime TF que le tenseur d'entrée est une constante. |
Table de hachage | Crée une table de hachage non initialisée. |
HistogramFixedWidth <U étend le nombre> | Renvoie l'histogramme des valeurs. |
Identité <T> | Renvoie un tenseur avec la même forme et le même contenu que le tenseur ou la valeur d'entrée. |
IdentitéN | Renvoie une liste de tenseurs avec les mêmes formes et contenus que l'entrée tenseurs. |
IgnoreErrorsDataset | Crée un ensemble de données qui contient les éléments de `input_dataset` en ignorant les erreurs. |
ImageProjectiveTransformV2 <T étend le nombre> | Applique la transformation donnée à chacune des images. |
ImageProjectiveTransformV3 <T étend le nombre> | Applique la transformation donnée à chacune des images. |
ImmutableConst <T> | Renvoie un tenseur immuable à partir de la région mémoire. |
InfeedDequeue <T> | Un espace réservé op pour une valeur qui sera introduite dans le calcul. |
InfeedDequeueTuple | Récupère plusieurs valeurs de l'alimentation sous la forme d'un tuple XLA. |
InfeedEnqueue | Un op qui alimente une seule valeur Tensor dans le calcul. |
InfeedEnqueuePrelinearizedBuffer | Une opération qui met en file d'attente le tampon prélinéarisé dans l'alimentation TPU. |
InfeedEnqueueTuple | Insère plusieurs valeurs Tensor dans le calcul sous la forme d'un tuple XLA. |
InitialiserTable | Initialiseur de table qui prend deux tenseurs pour les clés et les valeurs respectivement. |
InitializeTableFromDataset | |
InitializeTableFromTextFile | Initialise une table à partir d'un fichier texte. |
EnplaceAjouter <T> | Ajoute v dans les lignes spécifiées de x. |
InplaceSub <T> | Soustrait `v` dans les lignes spécifiées de `x`. |
InplaceUpdate <T> | Met à jour les lignes spécifiées 'i' avec les valeurs 'v'. |
IsBoostedTreesEnsembleInitialized | Vérifie si un ensemble d'arbres a été initialisé. |
IsBoostedTreesQuantileStreamResourceInitialized | Vérifie si un flux quantile a été initialisé. |
IsTPUEmbeddingInitialized | Indique si l'incorporation de TPU est initialisée dans un système TPU distribué. |
IsVariableInitialized | Vérifie si un tenseur a été initialisé. |
IsotonicRegression <U étend le nombre> | Résout un lot de problèmes de régression isotonique. |
IteratorGetDevice | Renvoie le nom de l'appareil sur lequel `ressource` a été placé. |
KMC2ChainInitialization | Renvoie l'index d'un point de données qui doit être ajouté à l'ensemble de départ. |
KmeansPlusPlusInitialization | Sélectionne num_to_sample lignes d'entrée à l'aide du critère KMeans++. |
KthOrderStatistic | Calcule la statistique d'ordre K d'un ensemble de données. |
Ensemble de données LMDB | Crée un jeu de données qui émet les paires clé-valeur dans un ou plusieurs fichiers LMDB. |
LSTMBlockCell <T étend le nombre> | Calcule la propagation vers l'avant de la cellule LSTM pour 1 pas de temps. |
LSTMBlockCellGrad <T étend le nombre> | Calcule la propagation vers l'arrière de la cellule LSTM pour 1 pas de temps. |
LinSpace <T étend le nombre> | Génère des valeurs dans un intervalle. |
ListDataset | Crée un ensemble de données qui émet chacun des "tenseurs" une fois. |
LoadAllTPUEmbeddingParameters | Une opération qui charge les paramètres d'optimisation dans la mémoire d'intégration. |
LoadTPUEmbeddingADAMParameters | Charger les paramètres d'intégration ADAM. |
LoadTPUEmbeddingAdadeltaParameters | Charger les paramètres d'incorporation d'Adadelta. |
LoadTPUEmbeddingAdagradMomentumParameters | Charger les paramètres d'incorporation d'Adagrad Momentum. |
LoadTPUEmbeddingAdagradParameters | Charger les paramètres d'intégration d'Adagrad. |
LoadTPUEmbeddingCenteredRMSPropParameters | Charger les paramètres d'intégration RMSProp centrés. |
LoadTPUEmbeddingFTRLParameters | Charger les paramètres d'intégration FTRL. |
LoadTPUEmbeddingFrequencyEstimatorParameters | Paramètres d'intégration de l'estimateur de fréquence de charge. |
LoadTPUEmbeddingMDLAdagradLightParameters | Charger les paramètres d'intégration MDL Adagrad Light. |
LoadTPUEmbeddingMomentumParametersLoadTPUEmbeddingMomentumParameters | Charger les paramètres d'intégration Momentum. |
LoadTPUEmbeddingProximalAdagradParameters | Charger les paramètres d'enrobage d'Adagrad proximaux. |
LoadTPUEmbeddingProximalYogiParameters | |
LoadTPUEmbeddingRMSPropParameters | Charger les paramètres d'intégration RMSProp. |
LoadTPUEmbeddingStochasticGradientDescentParameters | Charger les paramètres d'intégration SGD. |
LookupTableExport <T, U> | Affiche toutes les clés et valeurs du tableau. |
LookupTableFind <U> | Recherche les clés dans une table, affiche les valeurs correspondantes. |
LookupTableImport | Remplace le contenu de la table par les clés et les valeurs spécifiées. |
LookupTableInsert | Met à jour la table pour associer les clés aux valeurs. |
LookupTableRemove | Supprime les clés et leurs valeurs associées d'une table. |
LookupTableSize | Calcule le nombre d'éléments dans la table donnée. |
BoucleCond | Transfère l'entrée vers la sortie. |
LowerBound <U prolonge le nombre> | Applique lower_bound(sorted_search_values, values) le long de chaque ligne. |
Lu <T, U prolonge le nombre> | Calcule la décomposition LU d'une ou plusieurs matrices carrées. |
MakeUnique | Rendre tous les éléments de la dimension non Batch uniques, mais \"proche\" de leur valeur initiale. |
CarteEffacer | Op supprime tous les éléments du conteneur sous-jacent. |
MapIncompleteSize | Op renvoie le nombre d'éléments incomplets dans le conteneur sous-jacent. |
CartePeek | Op jette un coup d'œil aux valeurs de la clé spécifiée. |
Taille de la carte | Op renvoie le nombre d'éléments dans le conteneur sous-jacent. |
CarteStage | Étape (clé, valeurs) dans le conteneur sous-jacent qui se comporte comme une table de hachage. |
MapUnstage | Op supprime et renvoie les valeurs associées à la clé du conteneur sous-jacent. |
MapUnstageNoKey | Op supprime et renvoie un aléatoire (clé, valeur) du conteneur sous-jacent. |
MatrixDiagPartV2 <T> | Renvoie la partie diagonale par lots d'un tenseur par lots. |
MatrixDiagPartV3 <T> | Renvoie la partie diagonale par lots d'un tenseur par lots. |
MatrixDiagV2 <T> | Renvoie un tenseur diagonal par lots avec des valeurs diagonales par lots données. |
MatrixDiagV3 <T> | Renvoie un tenseur diagonal par lots avec des valeurs diagonales par lots données. |
MatrixSetDiagV2 <T> | Renvoie un tenseur matriciel par lots avec de nouvelles valeurs diagonales par lots. |
MatrixSetDiagV3 <T> | Renvoie un tenseur matriciel par lots avec de nouvelles valeurs diagonales par lots. |
Max <T> | Calcule le maximum d'éléments sur les dimensions d'un tenseur. |
MaxIntraOpParallelismDataset | Crée un jeu de données qui remplace le parallélisme intra-op maximal. |
Fusionner <T> | Transfère la valeur d'un tenseur disponible de `inputs` à `output`. |
MergeDedupData | Une opération fusionne des éléments de tenseurs entiers et flottants dans des données de déduplication sous forme de tuple XLA. |
Mini <T> | Calcule le minimum d'éléments sur les dimensions d'un tenseur. |
MirrorPad <T> | Remplit un tenseur avec des valeurs en miroir. |
MirrorPadGrad <T> | Opération de dégradé pour l'opération `MirrorPad`. |
MlirPassthroughOp | Enveloppe un calcul MLIR arbitraire exprimé sous la forme d'un module avec une fonction main(). |
MulNoNan <T> | Renvoie x * y élément par élément. |
MutableDenseHashTable | Crée une table de hachage vide qui utilise des tenseurs comme magasin de sauvegarde. |
MutableHashTable | Crée une table de hachage vide. |
MutableHashTableOfTensors | Crée une table de hachage vide. |
mutex | Crée une ressource Mutex qui peut être verrouillée par `MutexLock`. |
MutexLock | Verrouille une ressource mutex. |
NcclAllReduce <T étend le nombre> | Génère un tenseur contenant la réduction sur tous les tenseurs d'entrée. |
NcclBroadcast <T étend le nombre> | Envoie `input` à tous les appareils connectés à la sortie. |
NcclReduce <T étend le nombre> | Réduit `input` de `num_devices` en utilisant `reduction` à un seul périphérique. |
Ndtri <T prolonge le nombre> | |
Voisins les plus proches | Sélectionne les k centres les plus proches pour chaque point. |
NextAfter <T étend le nombre> | Renvoie la prochaine valeur représentable de `x1` dans la direction de `x2`, élément par élément. |
Itération suivante <T> | Rend son entrée disponible pour la prochaine itération. |
Pas d'op | Ne fait rien. |
NonDeterministicInts <U> | Génère de manière non déterministe des nombres entiers. |
NonMaxSuppressionV5 <T étend le nombre> | Sélectionne avidement un sous-ensemble de boîtes englobantes dans l'ordre décroissant du score, élaguer les boîtes qui ont un chevauchement élevé d'intersection sur union (IOU) avec les boîtes précédemment sélectionnées. |
NonSerializableDataset | |
OneHot <U> | Renvoie un tenseur one-hot. |
CeuxComme <T> | Renvoie un tenseur de uns avec la même forme et le même type que x. |
OptimizeDatasetV2 | Crée un ensemble de données en appliquant les optimisations associées à `input_dataset`. |
OptionsDataset | Crée un ensemble de données en attachant tf.data.Options à `input_dataset`. |
OrderedMapEffacer | Op supprime tous les éléments du conteneur sous-jacent. |
OrderedMapIncompleteSize | Op renvoie le nombre d'éléments incomplets dans le conteneur sous-jacent. |
OrderedMapPeek | Op jette un coup d'œil aux valeurs de la clé spécifiée. |
OrderedMapSize | Op renvoie le nombre d'éléments dans le conteneur sous-jacent. |
OrderedMapStage | Étape (clé, valeurs) dans le conteneur sous-jacent qui se comporte comme un ordonné conteneur associatif. |
OrderedMapUnstage | Op supprime et renvoie les valeurs associées à la clé du conteneur sous-jacent. |
OrderedMapUnstageNoKey | Op supprime et renvoie l'élément (clé, valeur) avec le plus petit clé du conteneur sous-jacent. |
OutfeedDequeue <T> | Récupère un seul tenseur de la sortie de calcul. |
OutfeedDequeueTuple | Récupérez plusieurs valeurs à partir de la sortie de calcul. |
OutfeedDequeueTupleV2 | Récupérez plusieurs valeurs à partir de la sortie de calcul. |
OutfeedDequeueV2 <T> | Récupère un seul tenseur de la sortie de calcul. |
Mise en file d'attente de sortie | Mettez en file d'attente un tenseur sur la sortie de calcul. |
OutfeedEnqueueTuple | Mettez en file d'attente plusieurs valeurs Tensor sur la sortie de calcul. |
Tampon <T> | Rembourre un tenseur. |
ParallelBatchDatasetParallelBatchDataset | |
ParallèleConcat <T> | Concatène une liste de tenseurs "N" le long de la première dimension. |
ParallèleDynamicStitch <T> | Entrelacez les valeurs des tenseurs "data" dans un seul tenseur. |
ParseExampleDatasetV2 | Transforme `input_dataset` contenant des protos `Example` en tant que vecteurs de DT_STRING en un ensemble de données d'objets `Tensor` ou `SparseTensor` représentant les entités analysées. |
ParseExampleV2 | Transforme un vecteur de protos tf.Example (sous forme de chaînes) en tenseurs typés. |
ParseSequenceExampleV2 | Transforme un vecteur de protos tf.io.SequenceExample (sous forme de chaînes) en tenseurs typés. |
Espace réservé <T> | Un espace réservé op pour une valeur qui sera introduite dans le calcul. |
Espace réservéAvecDéfaut <T> | Un espace réservé op qui passe par `input` lorsque sa sortie n'est pas alimentée. |
Prélinéariser | Un op qui linéarise une valeur Tensor en un tenseur variant opaque. |
PrélinéariserTuple | Une opération qui linéarise plusieurs valeurs de tenseur en un tenseur variant opaque. |
PrimitiveOp | Une classe de base pour les implémentations Op qui sont soutenues par une seule Operation . |
Imprimer | Imprime un scalaire de chaîne. |
PrivateThreadPoolDatasetPrivateThreadPoolDataset | Crée un ensemble de données qui utilise un pool de threads personnalisé pour calculer `input_dataset`. |
Produit <T> | Calcule le produit des éléments sur les dimensions d'un tenseur. |
QuantizeAndDequantizeV4 <T étend le nombre> | Quantifie puis déquantifie un tenseur. |
QuantizeAndDequantizeV4Grad <T étend le nombre> | Renvoie le gradient de `QuantizeAndDequantizeV4`. |
QuantizedConcat <T> | Concatène les tenseurs quantifiés le long d'une dimension. |
QuantizedConcatV2 <T> | |
QuantizedConv2DAndRelu <V> | |
QuantizedConv2DAndReluAndRequantize <V> | |
QuantizedConv2DAndRequantize <V> | |
QuantizedConv2DPerChannel <V> | Calcule QuantizedConv2D par canal. |
QuantizedConv2DWithBias <V> | |
QuantizedConv2DWithBiasAndRelu <V> | |
QuantizedConv2DWithBiasAndReluAndRequantize <W> | |
QuantizedConv2DWithBiasAndRequantize <W> | |
QuantizedConv2DWithBiasSignedSumAndReluAndRequantize <X> | |
QuantizedConv2DWithBiasSumAndRelu <V> | |
QuantizedConv2DWithBiasSumAndReluAndRequantize <X> | |
QuantizedDepthwiseConv2D <V> | Calcule la Conv2D quantifiée en profondeur. |
QuantizedDepthwiseConv2DWithBias <V> | Calcule la Conv2D quantifiée en profondeur avec Bias. |
QuantizedDepthwiseConv2DWithBiasAndRelu <V> | Calcule la Conv2D en profondeur quantifiée avec Bias et Relu. |
QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize <W> | Calcule la Conv2D en profondeur quantifiée avec Bias, Relu et Requantize. |
QuantizedMatMulWithBias <W> | Effectue une multiplication matricielle quantifiée de 'a' par la matrice 'b' avec ajout de biais. |
QuantizedMatMulWithBiasAndDequantize <W étend le nombre> | |
QuantizedMatMulWithBiasAndRelu <V> | Effectuez une multiplication matricielle quantifiée de 'a' par la matrice 'b' avec addition de biais et fusion relu. |
QuantizedMatMulWithBiasAndReluAndRequantize <W> | Effectuez une multiplication matricielle quantifiée de 'a' par la matrice 'b' avec addition de biais et relu et requantification de la fusion. |
QuantizedMatMulWithBiasAndRequantize <W> | |
QuantizedReshape <T> | Remodèle un tenseur quantifié conformément à l'opération Remodeler. |
RaggedBincount <U étend le nombre> | Compte le nombre d'occurrences de chaque valeur dans un tableau d'entiers. |
RaggedCountSparseOutput <U étend le nombre> | Effectue un comptage de bacs de sortie clairsemé pour une entrée de tenseur irrégulière. |
RaggedCross <T, U prolonge le nombre> | Génère une croix de caractéristiques à partir d'une liste de tenseurs et la renvoie sous la forme d'un RaggedTensor. |
RaggedFillEmptyRows <T> | |
RaggedFillEmptyRowsGrad <T> | |
RaggedGather <T étend le nombre, U> | Rassemblez des tranches irrégulières à partir de l'axe `0` params` en fonction des `indices`. |
RaggedRange <U extends Number, T extends Number> | Returns a `RaggedTensor` containing the specified sequences of numbers. |
RaggedTensorFromVariant <U extends Number, T> | Decodes a `variant` Tensor into a `RaggedTensor`. |
RaggedTensorToSparse <U> | Converts a `RaggedTensor` into a `SparseTensor` with the same values. |
RaggedTensorToTensor <U> | Create a dense tensor from a ragged tensor, possibly altering its shape. |
RaggedTensorToVariant | Encodes a `RaggedTensor` into a `variant` Tensor. |
RaggedTensorToVariantGradient <U> | Helper used to compute the gradient for `RaggedTensorToVariant`. |
RandomDatasetV2 | Creates a Dataset that returns pseudorandom numbers. |
RandomIndexShuffle <T extends Number> | Outputs the position of `value` in a permutation of [0, ..., max_index]. |
Range <T extends Number> | Creates a sequence of numbers. |
Rank | Returns the rank of a tensor. |
ReadVariableOp <T> | Reads the value of a variable. |
ReadVariableXlaSplitND <T> | Splits resource variable input tensor across all dimensions. |
RebatchDataset | Creates a dataset that changes the batch size. |
RebatchDatasetV2 | Creates a dataset that changes the batch size. |
Recv <T> | Receives the named tensor from send_device on recv_device. |
RecvTPUEmbeddingActivations | An op that receives embedding activations on the TPU. |
ReduceAll | Computes the "logical and" of elements across dimensions of a tensor. |
ReduceAny | Computes the "logical or" of elements across dimensions of a tensor. |
ReduceMax <T> | Computes the maximum of elements across dimensions of a tensor. |
ReduceMin <T> | Computes the minimum of elements across dimensions of a tensor. |
ReduceProd <T> | Computes the product of elements across dimensions of a tensor. |
ReduceSum <T> | Computes the sum of elements across dimensions of a tensor. |
RefEnter <T> | Creates or finds a child frame, and makes `data` available to the child frame. |
RefExit <T> | Exits the current frame to its parent frame. |
RefIdentity <T> | Return the same ref tensor as the input ref tensor. |
RefMerge <T> | Forwards the value of an available tensor from `inputs` to `output`. |
RefNextIteration <T> | Makes its input available to the next iteration. |
RefSelect <T> | Forwards the `index`th element of `inputs` to `output`. |
RefSwitch <T> | Forwards the ref tensor `data` to the output port determined by `pred`. |
RegisterDataset | Registers a dataset with the tf.data service. |
RegisterDatasetV2 | Registers a dataset with the tf.data service. |
Relayout <T> | |
RelayoutGrad <T> | |
RequantizationRangePerChannel | Computes requantization range per channel. |
RequantizePerChannel <U> | Requantizes input with min and max values known per channel. |
Reshape <T> | Reshapes a tensor. |
ResourceAccumulatorApplyGradient | Applies a gradient to a given accumulator. |
ResourceAccumulatorNumAccumulated | Returns the number of gradients aggregated in the given accumulators. |
ResourceAccumulatorSetGlobalStep | Updates the accumulator with a new value for global_step. |
ResourceAccumulatorTakeGradient <T> | Extracts the average gradient in the given ConditionalAccumulator. |
ResourceApplyAdagradV2 | Update '*var' according to the adagrad scheme. |
ResourceApplyAdamWithAmsgrad | Update '*var' according to the Adam algorithm. |
ResourceApplyKerasMomentum | Update '*var' according to the momentum scheme. |
ResourceConditionalAccumulator | A conditional accumulator for aggregating gradients. |
ResourceCountUpTo <T extends Number> | Increments variable pointed to by 'resource' until it reaches 'limit'. |
ResourceGather <U> | Gather slices from the variable pointed to by `resource` according to `indices`. |
ResourceGatherNd <U> | |
ResourceScatterAdd | Adds sparse updates to the variable referenced by `resource`. |
ResourceScatterDiv | Divides sparse updates into the variable referenced by `resource`. |
ResourceScatterMax | Reduces sparse updates into the variable referenced by `resource` using the `max` operation. |
ResourceScatterMin | Reduces sparse updates into the variable referenced by `resource` using the `min` operation. |
ResourceScatterMul | Multiplies sparse updates into the variable referenced by `resource`. |
ResourceScatterNdAdd | Applies sparse addition to individual values or slices in a Variable. |
ResourceScatterNdMax | |
ResourceScatterNdMin | |
ResourceScatterNdSub | Applies sparse subtraction to individual values or slices in a Variable. |
ResourceScatterNdUpdate | Applies sparse `updates` to individual values or slices within a given variable according to `indices`. |
ResourceScatterSub | Subtracts sparse updates from the variable referenced by `resource`. |
ResourceScatterUpdate | Assigns sparse updates to the variable referenced by `resource`. |
ResourceSparseApplyAdagradV2 | Update relevant entries in '*var' and '*accum' according to the adagrad scheme. |
ResourceSparseApplyKerasMomentum | Update relevant entries in '*var' and '*accum' according to the momentum scheme. |
ResourceStridedSliceAssign | Assign `value` to the sliced l-value reference of `ref`. |
RetrieveAllTPUEmbeddingParameters | An op that retrieves optimization parameters from embedding to host memory. |
RetrieveTPUEmbeddingADAMParameters | Retrieve ADAM embedding parameters. |
RetrieveTPUEmbeddingAdadeltaParameters | Retrieve Adadelta embedding parameters. |
RetrieveTPUEmbeddingAdagradMomentumParameters | Retrieve Adagrad Momentum embedding parameters. |
RetrieveTPUEmbeddingAdagradParameters | Retrieve Adagrad embedding parameters. |
RetrieveTPUEmbeddingCenteredRMSPropParameters | Retrieve centered RMSProp embedding parameters. |
RetrieveTPUEmbeddingFTRLParameters | Retrieve FTRL embedding parameters. |
RetrieveTPUEmbeddingFrequencyEstimatorParameters | Retrieve frequency estimator embedding parameters. |
RetrieveTPUEmbeddingMDLAdagradLightParameters | Retrieve MDL Adagrad Light embedding parameters. |
RetrieveTPUEmbeddingMomentumParameters | Retrieve Momentum embedding parameters. |
RetrieveTPUEmbeddingProximalAdagradParameters | Retrieve proximal Adagrad embedding parameters. |
RetrieveTPUEmbeddingProximalYogiParameters | |
RetrieveTPUEmbeddingRMSPropParameters | Retrieve RMSProp embedding parameters. |
RetrieveTPUEmbeddingStochasticGradientDescentParameters | Retrieve SGD embedding parameters. |
Reverse <T> | Reverses specific dimensions of a tensor. |
ReverseSequence <T> | Reverses variable length slices. |
RewriteDataset | |
RiscAbs <T extends Number> | |
RiscAdd <T extends Number> | Returns x + y element-wise. |
RiscBinaryArithmetic <T extends Number> | |
RiscBinaryComparison | |
RiscBitcast <U> | |
RiscBroadcast <T> | |
RiscCast <U> | |
RiscCeil <T extends Number> | |
RiscCholesky <T extends Number> | |
RiscConcat <T> | |
RiscConv <T extends Number> | |
RiscCos <T extends Number> | |
RiscDiv <T extends Number> | |
RiscDot <T extends Number> | |
RiscExp <T extends Number> | |
RiscFft <T> | |
RiscFloor <T extends Number> | |
RiscGather <T> | |
RiscImag <U extends Number> | |
RiscIsFinite | |
RiscLog <T extends Number> | |
RiscLogicalAnd | |
RiscLogicalNot | |
RiscLogicalOr | |
RiscMax <T extends Number> | Returns max(x, y) element-wise. |
RiscMin <T extends Number> | |
RiscMul <T extends Number> | |
RiscNeg <T extends Number> | |
RiscPad <T extends Number> | |
RiscPool <T extends Number> | |
RiscPow <T extends Number> | |
RiscRandomUniform | |
RiscReal <U extends Number> | |
RiscReduce <T extends Number> | |
RiscRem <T extends Number> | |
RiscReshape <T extends Number> | |
RiscReverse <T extends Number> | |
RiscScatter <U extends Number> | |
RiscShape <U extends Number> | |
RiscSign <T extends Number> | |
RiscSlice <T extends Number> | |
RiscSort <T extends Number> | |
RiscSqueeze <T> | |
RiscSub <T extends Number> | |
RiscTranspose <T> | |
RiscTriangularSolve <T extends Number> | |
RiscUnary <T extends Number> | |
RngReadAndSkip | Advance the counter of a counter-based RNG. |
RngSkip | Advance the counter of a counter-based RNG. |
Roll <T> | Rolls the elements of a tensor along an axis. |
SamplingDataset | Creates a dataset that takes a Bernoulli sample of the contents of another dataset. |
ScaleAndTranslate | |
ScaleAndTranslateGrad <T extends Number> | |
ScatterAdd <T> | Adds sparse updates to a variable reference. |
ScatterDiv <T> | Divides a variable reference by sparse updates. |
ScatterMax <T extends Number> | Reduces sparse updates into a variable reference using the `max` operation. |
ScatterMin <T extends Number> | Reduces sparse updates into a variable reference using the `min` operation. |
ScatterMul <T> | Multiplies sparse updates into a variable reference. |
ScatterNd <U> | Scatters `updates` into a tensor of shape `shape` according to `indices`. |
ScatterNdAdd <T> | Applies sparse addition to individual values or slices in a Variable. |
ScatterNdMax <T> | Computes element-wise maximum. |
ScatterNdMin <T> | Computes element-wise minimum. |
ScatterNdNonAliasingAdd <T> | Applies sparse addition to `input` using individual values or slices from `updates` according to indices `indices`. |
ScatterNdSub <T> | Applies sparse subtraction to individual values or slices in a Variable. |
ScatterNdUpdate <T> | Applies sparse `updates` to individual values or slices within a given variable according to `indices`. |
ScatterSub <T> | Subtracts sparse updates to a variable reference. |
ScatterUpdate <T> | Applies sparse updates to a variable reference. |
SegmentMaxV2 <T extends Number> | Computes the maximum along segments of a tensor. |
SegmentMinV2 <T extends Number> | Computes the minimum along segments of a tensor. |
SegmentProdV2 <T> | Computes the product along segments of a tensor. |
SegmentSumV2 <T> | Computes the sum along segments of a tensor. |
SelectV2 <T> | |
Send | Sends the named tensor from send_device to recv_device. |
SendTPUEmbeddingGradients | Performs gradient updates of embedding tables. |
SetDiff1d <T, U extends Number> | Computes the difference between two lists of numbers or strings. |
SetSize | Number of unique elements along last dimension of input `set`. |
Shape <U extends Number> | Returns the shape of a tensor. |
ShapeN <U extends Number> | Returns shape of tensors. |
ShardDataset | Creates a `Dataset` that includes only 1/`num_shards` of this dataset. |
ShuffleAndRepeatDatasetV2 | |
ShuffleDatasetV2 | |
ShuffleDatasetV3 | |
ShutdownDistributedTPU | Shuts down a running distributed TPU system. |
ShutdownTPUSystem | An op that shuts down the TPU system. |
Size <U extends Number> | Returns the size of a tensor. |
Skipgram | Parses a text file and creates a batch of examples. |
SleepDataset | |
Slice <T> | Return a slice from 'input'. |
SlidingWindowDataset | Creates a dataset that passes a sliding window over `input_dataset`. |
Snapshot <T> | Returns a copy of the input tensor. |
SnapshotDataset | Creates a dataset that will write to / read from a snapshot. |
SnapshotDatasetReader | |
SnapshotNestedDatasetReader | |
SobolSample <T extends Number> | Generates points from the Sobol sequence. |
SpaceToBatchNd <T> | SpaceToBatch for ND tensors of type T. |
SparseApplyAdagradV2 <T> | Update relevant entries in '*var' and '*accum' according to the adagrad scheme. |
SparseBincount <U extends Number> | Counts the number of occurrences of each value in an integer array. |
SparseCountSparseOutput <U extends Number> | Performs sparse-output bin counting for a sparse tensor input. |
SparseCrossHashed | Generates sparse cross from a list of sparse and dense tensors. |
SparseCrossV2 | Generates sparse cross from a list of sparse and dense tensors. |
SparseMatrixAdd | Sparse addition of two CSR matrices, C = alpha * A + beta * B. |
SparseMatrixMatMul <T> | Matrix-multiplies a sparse matrix with a dense matrix. |
SparseMatrixMul | Element-wise multiplication of a sparse matrix with a dense tensor. |
SparseMatrixNNZ | Returns the number of nonzeroes of `sparse_matrix`. |
SparseMatrixOrderingAMD | Computes the Approximate Minimum Degree (AMD) ordering of `input`. |
SparseMatrixSoftmax | Calculates the softmax of a CSRSparseMatrix. |
SparseMatrixSoftmaxGrad | Calculates the gradient of the SparseMatrixSoftmax op. |
SparseMatrixSparseCholesky | Computes the sparse Cholesky decomposition of `input`. |
SparseMatrixSparseMatMul | Sparse-matrix-multiplies two CSR matrices `a` and `b`. |
SparseMatrixTranspose | Transposes the inner (matrix) dimensions of a CSRSparseMatrix. |
SparseMatrixZeros | Creates an all-zeros CSRSparseMatrix with shape `dense_shape`. |
SparseSegmentSumGrad <T extends Number> | Computes gradients for SparseSegmentSum. |
SparseTensorToCSRSparseMatrix | Converts a SparseTensor to a (possibly batched) CSRSparseMatrix. |
Spence <T extends Number> | |
Split <T> | Splits a tensor into `num_split` tensors along one dimension. |
SplitDedupData <T extends Number, U extends Number> | An op splits input deduplication data XLA tuple into integer and floating point tensors. |
SplitV <T> | Splits a tensor into `num_split` tensors along one dimension. |
Squeeze <T> | Removes dimensions of size 1 from the shape of a tensor. |
Stack <T> | Packs a list of `N` rank-`R` tensors into one rank-`(R+1)` tensor. |
Stage | Stage values similar to a lightweight Enqueue. |
StageClear | Op removes all elements in the underlying container. |
StagePeek | Op peeks at the values at the specified index. |
StageSize | Op returns the number of elements in the underlying container. |
StatefulRandomBinomial <V extends Number> | |
StatefulStandardNormal <U> | Outputs random values from a normal distribution. |
StatefulStandardNormalV2 <U> | Outputs random values from a normal distribution. |
StatefulTruncatedNormal <U> | Outputs random values from a truncated normal distribution. |
StatefulUniform <U> | Outputs random values from a uniform distribution. |
StatefulUniformFullInt <U> | Outputs random integers from a uniform distribution. |
StatefulUniformInt <U> | Outputs random integers from a uniform distribution. |
StatelessParameterizedTruncatedNormal <V extends Number> | |
StatelessRandomBinomial <W extends Number> | Outputs deterministic pseudorandom random numbers from a binomial distribution. |
StatelessRandomGammaV2 <V extends Number> | Outputs deterministic pseudorandom random numbers from a gamma distribution. |
StatelessRandomGammaV3 <U extends Number> | Outputs deterministic pseudorandom random numbers from a gamma distribution. |
StatelessRandomGetAlg | Picks the best counter-based RNG algorithm based on device. |
StatelessRandomGetKeyCounter | Scrambles seed into key and counter, using the best algorithm based on device. |
StatelessRandomGetKeyCounterAlg | Picks the best algorithm based on device, and scrambles seed into key and counter. |
StatelessRandomNormalV2 <U extends Number> | Outputs deterministic pseudorandom values from a normal distribution. |
StatelessRandomPoisson <W extends Number> | Outputs deterministic pseudorandom random numbers from a Poisson distribution. |
StatelessRandomUniformFullInt <V extends Number> | Outputs deterministic pseudorandom random integers from a uniform distribution. |
StatelessRandomUniformFullIntV2 <U extends Number> | Outputs deterministic pseudorandom random integers from a uniform distribution. |
StatelessRandomUniformIntV2 <U extends Number> | Outputs deterministic pseudorandom random integers from a uniform distribution. |
StatelessRandomUniformV2 <U extends Number> | Outputs deterministic pseudorandom random values from a uniform distribution. |
StatelessSampleDistortedBoundingBox <T extends Number> | Generate a randomly distorted bounding box for an image deterministically. |
StatelessShuffle <T> | Randomly and deterministically shuffles a tensor along its first dimension. |
StatelessTruncatedNormalV2 <U extends Number> | Outputs deterministic pseudorandom values from a truncated normal distribution. |
StatsAggregatorHandleV2 | |
StatsAggregatorSetSummaryWriter | Set a summary_writer_interface to record statistics using given stats_aggregator. |
StopGradient <T> | Stops gradient computation. |
StridedSlice <T> | Return a strided slice from `input`. |
StridedSliceAssign <T> | Assign `value` to the sliced l-value reference of `ref`. |
StridedSliceGrad <U> | Returns the gradient of `StridedSlice`. |
StringLower | Converts all uppercase characters into their respective lowercase replacements. |
StringNGrams <T extends Number> | Creates ngrams from ragged string data. |
StringUpper | Converts all lowercase characters into their respective uppercase replacements. |
Sum <T> | Computes the sum of elements across dimensions of a tensor. |
SwitchCond <T> | Forwards `data` to the output port determined by `pred`. |
SyncDevice | Synchronizes the device this op is run on. |
TPUCompilationResult | Returns the result of a TPU compilation. |
TPUCompileSucceededAssert | Asserts that compilation succeeded. |
TPUEmbeddingActivations | An op enabling differentiation of TPU Embeddings. |
TPUExecute | Op that loads and executes a TPU program on a TPU device. |
TPUExecuteAndUpdateVariables | Op that executes a program with optional in-place variable updates. |
TPUOrdinalSelector | A TPU core selector Op. |
TPUPartitionedInput <T> | An op that groups a list of partitioned inputs together. |
TPUPartitionedInputV2 <T> | An op that groups a list of partitioned inputs together. |
TPUPartitionedOutput <T> | An op that demultiplexes a tensor to be sharded by XLA to a list of partitioned outputs outside the XLA computation. |
TPUPartitionedOutputV2 <T> | An op that demultiplexes a tensor to be sharded by XLA to a list of partitioned outputs outside the XLA computation. |
TPUReplicateMetadata | Metadata indicating how the TPU computation should be replicated. |
TPUReplicatedInput <T> | Connects N inputs to an N-way replicated TPU computation. |
TPUReplicatedOutput <T> | Connects N outputs from an N-way replicated TPU computation. |
TPUReshardVariables | Op that reshards on-device TPU variables to specified state. |
TPURoundRobin | Round-robin load balancing on TPU cores. |
TemporaryVariable <T> | Returns a tensor that may be mutated, but only persists within a single step. |
TensorArray | An array of Tensors of given size. |
TensorArrayClose | Delete the TensorArray from its resource container. |
TensorArrayConcat <T> | Concat the elements from the TensorArray into value `value`. |
TensorArrayGather <T> | Gather specific elements from the TensorArray into output `value`. |
TensorArrayGrad | Creates a TensorArray for storing the gradients of values in the given handle. |
TensorArrayGradWithShape | Creates a TensorArray for storing multiple gradients of values in the given handle. |
TensorArrayPack <T> | |
TensorArrayRead <T> | Read an element from the TensorArray into output `value`. |
TensorArrayScatter | Scatter the data from the input value into specific TensorArray elements. |
TensorArraySize | Get the current size of the TensorArray. |
TensorArraySplit | Split the data from the input value into TensorArray elements. |
TensorArrayUnpack | |
TensorArrayWrite | Push an element onto the tensor_array. |
TensorListConcat <T> | Concats all tensors in the list along the 0th dimension. |
TensorListConcatLists | |
TensorListConcatV2 <U> | Concats all tensors in the list along the 0th dimension. |
TensorListElementShape <T extends Number> | The shape of the elements of the given list, as a tensor. |
TensorListFromTensor | Creates a TensorList which, when stacked, has the value of `tensor`. |
TensorListGather <T> | Creates a Tensor by indexing into the TensorList. |
TensorListGetItem <T> | |
TensorListLength | Returns the number of tensors in the input tensor list. |
TensorListPopBack <T> | Returns the last element of the input list as well as a list with all but that element. |
TensorListPushBack | Returns a list which has the passed-in `Tensor` as last element and the other elements of the given list in `input_handle`. |
TensorListPushBackBatch | |
TensorListReserve | List of the given size with empty elements. |
TensorListResize | Resizes the list. |
TensorListScatter | Creates a TensorList by indexing into a Tensor. |
TensorListScatterIntoExistingList | Scatters tensor at indices in an input list. |
TensorListScatterV2 | Creates a TensorList by indexing into a Tensor. |
TensorListSetItem | |
TensorListSplit | Splits a tensor into a list. |
TensorListStack <T> | Stacks all tensors in the list. |
TensorMapErase | Returns a tensor map with item from given key erased. |
TensorMapHasKey | Returns whether the given key exists in the map. |
TensorMapInsert | Returns a map that is the 'input_handle' with the given key-value pair inserted. |
TensorMapLookup <U> | Returns the value from a given key in a tensor map. |
TensorMapSize | Returns the number of tensors in the input tensor map. |
TensorMapStackKeys <T> | Returns a Tensor stack of all keys in a tensor map. |
TensorScatterAdd <T> | Adds sparse `updates` to an existing tensor according to `indices`. |
TensorScatterMax <T> | Apply a sparse update to a tensor taking the element-wise maximum. |
TensorScatterMin <T> | |
TensorScatterSub <T> | Subtracts sparse `updates` from an existing tensor according to `indices`. |
TensorScatterUpdate <T> | Scatter `updates` into an existing tensor according to `indices`. |
TensorStridedSliceUpdate <T> | Assign `value` to the sliced l-value reference of `input`. |
ThreadPoolDataset | Creates a dataset that uses a custom thread pool to compute `input_dataset`. |
ThreadPoolHandle | Creates a dataset that uses a custom thread pool to compute `input_dataset`. |
Tile <T> | Constructs a tensor by tiling a given tensor. |
Timestamp | Provides the time since epoch in seconds. |
ToBool | Converts a tensor to a scalar predicate. |
TopKUnique | Returns the TopK unique values in the array in sorted order. |
TopKWithUnique | Returns the TopK values in the array in sorted order. |
TpuHandleToProtoKey | Converts XRT's uid handles to TensorFlow-friendly input format. |
TridiagonalMatMul <T> | Calculate product with tridiagonal matrix. |
TridiagonalSolve <T> | Solves tridiagonal systems of equations. |
Unbatch <T> | Reverses the operation of Batch for a single output Tensor. |
UnbatchGrad <T> | Gradient of Unbatch. |
UncompressElement | Uncompresses a compressed dataset element. |
UnicodeDecode <T extends Number> | Decodes each string in `input` into a sequence of Unicode code points. |
UnicodeEncode | Encode a tensor of ints into unicode strings. |
UniformDequantize <U extends Number> | Perform dequantization on the quantized Tensor `input`. |
UniformQuantize <U> | Perform quantization on Tensor `input`. |
UniformQuantizedAdd <T> | Perform quantized add of quantized Tensor `lhs` and quantized Tensor `rhs` to make quantized `output`. |
UniformQuantizedClipByValue <T> | Perform clip by value on the quantized Tensor `operand`. |
UniformQuantizedConvolution <U> | Perform quantized convolution of quantized Tensor `lhs` and quantized Tensor `rhs`. |
UniformQuantizedConvolutionHybrid <V extends Number> | Perform hybrid quantized convolution of float Tensor `lhs` and quantized Tensor `rhs`. |
UniformQuantizedDot <U> | Perform quantized dot of quantized Tensor `lhs` and quantized Tensor `rhs` to make quantized `output`. |
UniformQuantizedDotHybrid <V extends Number> | Perform hybrid quantized dot of float Tensor `lhs` and quantized Tensor `rhs`. |
UniformRequantize <U> | Given quantized tensor `input`, requantize it with new quantization parameters. |
Unique <T, V extends Number> | Finds unique elements along an axis of a tensor. |
UniqueDataset | Creates a dataset that contains the unique elements of `input_dataset`. |
UniqueWithCounts <T, V extends Number> | Finds unique elements along an axis of a tensor. |
UnravelIndex <T extends Number> | Converts an array of flat indices into a tuple of coordinate arrays. |
UnsortedSegmentJoin | |
Unstack <T> | Unpacks a given dimension of a rank-`R` tensor into `num` rank-`(R-1)` tensors. |
Unstage | Op is similar to a lightweight Dequeue. |
UnwrapDatasetVariant | |
UpperBound <U extends Number> | Applies upper_bound(sorted_search_values, values) along each row. |
VarHandleOp | Creates a handle to a Variable resource. |
VarIsInitializedOp | Checks whether a resource handle-based variable has been initialized. |
Variable <T> | Holds state in the form of a tensor that persists across steps. |
VariableShape <T extends Number> | Returns the shape of the variable pointed to by `resource`. |
Where | Returns locations of nonzero / true values in a tensor. |
Where3 <T> | Selects elements from `x` or `y`, depending on `condition`. |
WindowOp | |
WorkerHeartbeat | Worker heartbeat op. |
WrapDatasetVariant | |
WriteRawProtoSummary | Writes a serialized proto summary. |
XlaConcatND <T> | Concats input tensor across all dimensions. |
XlaRecvFromHost <T> | An op to receive a tensor from the host. |
XlaRecvTPUEmbeddingActivations | An op that receives embedding activations on the TPU. |
XlaRecvTPUEmbeddingDeduplicationData | Receives deduplication data (indices and weights) from the embedding core. |
XlaSendTPUEmbeddingGradients | An op that performs gradient updates of embedding tables. |
XlaSendToHost | An op to send a tensor to the host. |
XlaSplitND <T> | Splits input tensor across all dimensions. |
Xlog1py <T> | Returns 0 if x == 0, and x * log1p(y) otherwise, elementwise. |
Zeros <T> | An operator creating a constant initialized with zeros of the shape given by `dims`. |
ZerosLike <T> | Returns a tensor of zeros with the same shape and type as x. |