'mhlo' Dialecte

Opérations

mhlo.abs (mhlo::AbsOp)

Fonctionnement des abdominaux

Syntaxe:

operation ::= `mhlo.abs` $operand attr-dict
              `:` custom<SameOperandsAndResultType>(type($operand), type($result))

Effectue une opération abs élément par élément sur le tenseur operand et produit un tenseur result .

Voir : https://github.com/openxla/stablehlo/blob/main/docs/spec.md#abs

Exemple:

%result = mhlo.abs %operand : tensor<3xi32>

Traits : AlwaysSpeculatableImplTrait , Elementwise , SameOperandsAndResultShape

Interfaces : ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effets : MemoryEffects::Effect{}

Opérandes:

Résultats:

mhlo.acos (mhlo::AcosOp)

Opération Acos

Syntaxe:

operation ::= `mhlo.acos` $operand attr-dict
              `:` custom<SameOperandsAndResultType>(type($operand), type($result))

Exécute une opération acos élément par élément sur le tenseur operand et produit un tenseur result .

Exemple:

%result = mhlo.acos %operand : tensor<2x2xf32>

Traits : CompatibleOperandsAndResultType , Elementwise , SameOperandsAndResultShape

Interfaces : InferShapedTypeOpInterface , InferTypeOpInterface

Opérandes:

Résultats:

mhlo.acosh (mhlo::AcoshOp)

Opération Acosh

Syntaxe:

operation ::= `mhlo.acosh` $operand attr-dict
              `:` custom<SameOperandsAndResultType>(type($operand), type($result))

Exécute une opération acosh élément par élément sur le tenseur operand et produit un tenseur result .

Exemple:

%result = mhlo.acosh %operand : tensor<2x2xf32>

Traits : CompatibleOperandsAndResultType , Elementwise , SameOperandsAndResultShape

Interfaces : InferShapedTypeOpInterface , InferTypeOpInterface

Opérandes:

Résultats:

mhlo.add (mhlo::AddOp)

Opération d'ajout

Syntaxe:

operation ::= `mhlo.add` $lhs `,` $rhs attr-dict
              `:` custom<SameOperandsAndResultType>(type($lhs), type($rhs), type($result))

Effectue l'addition élément par élément de deux tenseurs lhs et rhs et produit un tenseur result .

Voir : https://github.com/openxla/stablehlo/blob/main/docs/spec.md#add

Exemple:

%result = mhlo.add %lhs, %rhs : tensor<2x2xi32>

Traits : AlwaysSpeculatableImplTrait , Commutative , CompatibleOperandsAndResultType , Elementwise , SameOperandsAndResultShape

Interfaces : ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effets : MemoryEffects::Effect{}

Opérandes:

Résultats:

mhlo.add_dependency (mhlo::AddDependencyOp)

Opération AddDependency

Syntaxe:

operation ::= `mhlo.add_dependency` operands attr-dict `:` functional-type(operands, results)

Cette opération est privée pour le compilateur XLA, elle n'a donc pas encore de spécification.

De manière informelle, cette opération comporte deux opérandes : un opérande de données et un jeton. Le résultat de l'opération est l'opérande de données. Utilisée avec AfterAll, cette opération permet de classer les opérations sans effets secondaires (celles qui ne produisent pas de valeurs de jeton).

Exemple:

%1 = mhlo.add_dependency %arg0, %0 : (tensor<3x4xf32>, !mhlo.token) -> tensor<3x4xf32>

Traits : AlwaysSpeculatableImplTrait

Interfaces : ConditionallySpeculatable , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effets : MemoryEffects::Effect{}

Opérandes:

Résultats:

mhlo.after_all (mhlo::AfterAllOp)

Opération AfterAll

Syntaxe:

operation ::= `mhlo.after_all` $inputs attr-dict
              `:` custom<VariadicSameOperandsAndResultType>(ref($inputs), type($inputs), type($result))

Garantit que les opérations produisant les inputs sont exécutées avant toute opération qui dépend du result .

Voir : https://github.com/openxla/stablehlo/blob/main/docs/spec.md#after_all

Exemple:

%result = mhlo.after_all %input0, %input1 : !mhlo.token

Traits : AlwaysSpeculatableImplTrait

Interfaces : ConditionallySpeculatable , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effets : MemoryEffects::Effect{}

Opérandes:

Résultats:

mhlo.all_gather (mhlo::AllGatherOp)

Opération AllGather

Au sein de chaque groupe de processus de la grille, concatène les valeurs du tenseur d'opérandes de chaque processus selon all_gather_dim et produit un tenseur de résultats. Le computation est appliqué séparément pour chaque opérande de operands , produisant un résultat par opérande.

Voir : https://github.com/openxla/stablehlo/blob/main/docs/spec.md#all_gather

Exemple:

%result = "mhlo.all_gather"(%operand) {
  all_gather_dim = 1 : i64,
  replica_groups = dense<[[0, 1]]> : tensor<1x2xi64>
  // channel_id = 0
  channel_handle = #mhlo.channel_handle<handle = 0, type = 0>,
  // use_global_device_ids = false
} : (tensor<2x2xf32>) -> tensor<2x4xf32>

Traits : SameOperandsAndResultElementType

Attributs:

Opérandes:

Résultats:

mhlo.all_reduce (mhlo::AllReduceOp)

Opération AllReduce

Au sein de chaque groupe de processus de la grille de processus, un computation de fonction de réduction est appliqué aux valeurs d'un tenseur d'opérandes de chaque processus et produit un tenseur de résultats. Le computation est appliqué séparément pour chaque opérande de operands , produisant un résultat par opérande.

Voir : https://github.com/openxla/stablehlo/blob/main/docs/spec.md#all_reduce

Exemple:

%result = "mhlo.all_reduce"(%operand) ({
  ^bb0(%arg0: tensor<f32>, %arg1: tensor<f32>):
    %0 = mhlo.add %arg1, %arg2 : tensor<f32>
    mhlo.return %0 : tensor<f32>
}) {
  replica_groups = dense<[[0, 1]]> : tensor<1x2xi64>
  // channel_id = 0
  channel_handle = #mhlo.channel_handle<handle = 0, type = 0>
  // use_global_device_ids = false
} : (tensor<4xf32>) -> tensor<4xf32>

Traits : InferTensorType , SingleBlockImplicitTerminator<ReturnOp> , SingleBlock

Interfaces : InferShapedTypeOpInterface , InferTypeOpInterface

Attributs:

Opérandes:

Résultats:

mhlo.all_to_all (mhlo::AllToAllOp)

Opération AllToAll

Dans chaque groupe de processus de la grille de processus, divise les valeurs du tenseur d' operand le long split_dimension en parties, disperse les parties divisées entre les processus, concatène les parties dispersées le long concat_dimension et produit un tenseur result .

Voir : https://github.com/openxla/stablehlo/blob/main/docs/spec.md#all_to_all

Exemple:

%result = "mhlo.all_to_all"(%operand) {
  split_dimension = 1 : i64,
  concat_dimension = 0 : i64,
  split_count = 2 : i64,
  replica_groups = dense<[[0, 1]]> : tensor<1x2xi64>
} : (tensor<2x4xf32>) -> tensor<4x2xf32>

Traits : AlwaysSpeculatableImplTrait , InferTensorType , SameOperandsElementType , SameOperandsShape , SameVariadicOperandSize

Interfaces : ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effets : MemoryEffects::Effect{}

Attributs:

Opérandes:

Résultats:

mhlo.and (mhlo::AndOp)

Et l'opération

Syntaxe:

operation ::= `mhlo.and` $lhs `,` $rhs attr-dict
              `:` custom<SameOperandsAndResultType>(type($lhs), type($rhs), type($result))

Effectue un ET élément par élément de deux tenseurs lhs et rhs et produit un tenseur result

Voir : https://github.com/openxla/stablehlo/blob/main/docs/spec.md#and

Exemple:

%result = mhlo.and %lhs, %rhs : tensor<2x2xi32>

Traits : AlwaysSpeculatableImplTrait , Commutative , CompatibleOperandsAndResultType , Elementwise , SameOperandsAndResultShape

Interfaces : ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effets : MemoryEffects::Effect{}

Opérandes:

Résultats:

mhlo.asin (mhlo::AsinOp)

Opération Asin

Syntaxe:

operation ::= `mhlo.asin` $operand attr-dict
              `:` custom<SameOperandsAndResultType>(type($operand), type($result))

Effectue une opération asin élément par élément sur le tenseur operand et produit un tenseur result .

Exemple:

%result = mhlo.asin %operand : tensor<2x2xf32>

Traits : CompatibleOperandsAndResultType , Elementwise , SameOperandsAndResultShape

Interfaces : InferShapedTypeOpInterface , InferTypeOpInterface

Opérandes:

Résultats:

mhlo.asinh (mhlo::AsinhOp)

Opération Asinh

Syntaxe:

operation ::= `mhlo.asinh` $operand attr-dict
              `:` custom<SameOperandsAndResultType>(type($operand), type($result))

Exécute une opération asinh élément par élément sur le tenseur operand et produit un tenseur result .

Exemple:

%result = mhlo.asinh %operand : tensor<2x2xf32>

Traits : CompatibleOperandsAndResultType , Elementwise , SameOperandsAndResultShape

Interfaces : InferShapedTypeOpInterface , InferTypeOpInterface

Opérandes:

Résultats:

mhlo.async_done (mhlo::AsyncDoneOp)

Opération AsyncDone

Cette opération est privée pour le compilateur XLA, elle n'a donc pas encore de spécification.

De manière informelle, cette opération se bloque jusqu'à la fin d'un calcul asynchrone. Elle renvoie le résultat final du calcul asynchrone.

Consultez la documentation d'AsyncStart pour plus d'informations.

Interfaces : InferTypeOpInterface

Opérandes:

Résultats:

mhlo.async_start (mhlo::AsyncStartOp)

Opération AsyncStart

Cette opération est privée pour le compilateur XLA, elle n'a donc pas encore de spécification.

De manière informelle, cette opération déclenche un calcul asynchrone.

Ceci est utilisé lorsque des fonctions contiennent à la fois des attentes asynchrones (comme les DMA) et des calculs sur thread. Par exemple, une fonction peut être composée d'un calcul, d'un DMA, d'un autre calcul, d'un second DMA et d'un calcul final. Ceci est représenté par un async_start suivi d'un async_update et d'un async_done. L'async_start effectue le premier calcul sur thread, puis démarre le DMA. L'async_update attend la fin du DMA s'il n'est pas encore terminé, puis exécute le second calcul de la fonction et démarre le second DMA. Enfin, l'async_done attend ce dernier DMA, puis exécute le dernier calcul à exécuter sur thread et renvoie le résultat de ce dernier calcul.

operands sont transmis directement au calcul. called_computation désigne la fonction qui sera exécutée de manière asynchrone. execution_thread désigne le nom du thread dans lequel elle sera exécutée. Le thread principal est appelé « main ». Tous les threads ont un nom.

Cela renvoie tous les états nécessaires entre les opérations asynchrones. Après l'affectation du tampon, les valeurs de retour représentent l'espace nécessaire pour stocker les entrées, les résultats et les blocs-notes nécessaires ou modifiés par l'opération asynchrone.

Attributs:

Opérandes:

Résultats:

mhlo.async_update (mhlo::AsyncUpdateOp)

Opération AsyncUpdate

Cette opération est privée pour le compilateur XLA, elle n'a donc pas encore de spécification.

De manière informelle, cette opération bloque un calcul asynchrone jusqu'à une barrière de synchronisation. Elle renvoie bundle après l'opération.

Consultez la documentation d'AsyncStart pour plus d'informations.

Interfaces : InferTypeOpInterface

Opérandes:

Résultats:

mhlo.atan2 (mhlo::Atan2Op)

Fonctionnement d'Atan2

Syntaxe:

operation ::= `mhlo.atan2` $lhs `,` $rhs attr-dict
              `:` custom<SameOperandsAndResultType>(type($lhs), type($rhs), type($result))

Exécute l'opération atan2 élément par élément sur le tenseur lhs et rhs et produit un tenseur result .

Voir : https://github.com/openxla/stablehlo/blob/main/docs/spec.md#atan2

Exemple:

%result = mhlo.atan2 %lhs, %rhs : tensor<3xf32>

Traits : AlwaysSpeculatableImplTrait , CompatibleOperandsAndResultType , Elementwise , SameOperandsAndResultShape

Interfaces : ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effets : MemoryEffects::Effect{}

Opérandes:

Résultats:

mhlo.atanh (mhlo::AtanhOp)

Opération Atanh

Syntaxe:

operation ::= `mhlo.atanh` $operand attr-dict
              `:` custom<SameOperandsAndResultType>(type($operand), type($result))

Exécute une opération atanh élément par élément sur le tenseur operand et produit un tenseur result .

Exemple:

%result = mhlo.atanh %operand : tensor<2x2xf32>

Traits : CompatibleOperandsAndResultType , Elementwise , SameOperandsAndResultShape

Interfaces : InferShapedTypeOpInterface , InferTypeOpInterface

Opérandes:

Résultats:

mhlo.batch_norm_grad (mhlo :: BatchNormGradOp)

Opération BatchNormGrad

Calcule les gradients de plusieurs entrées de BatchNormTrainingOp rétropropagées à partir de grad_output et produit les tenseurs grad_operand , grad_scale et grad_offset .

Voir : https://github.com/openxla/stablehlo/blob/main/docs/spec.md#batch_norm_grad

Exemple:

%grad_operand, %grad_scale, %grad_offset =
"mhlo.batch_norm_grad"(%operand, %scale, %mean, %variance, %grad_output) {
  epsilon = 0.0 : f32,
  feature_index = 2 : i64
} : (tensor<2x2x2xf32>, tensor<2xf32>, tensor<2xf32>, tensor<2xf32>,
    tensor<2x2x2xf32>) -> (tensor<2x2x2xf32>, tensor<2xf32>, tensor<2xf32>)

Traits : AlwaysSpeculatableImplTrait , InferTensorType

Interfaces : ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effets : MemoryEffects::Effect{}

Attributs:

Opérandes:

Résultats:

mhlo.batch_norm_inference (mhlo::BatchNormInferenceOp)

Opération BatchNormInference

Normalise le tenseur operand sur toutes les dimensions à l'exception de la dimension feature_index et produit un tenseur result .

Voir : https://github.com/openxla/stablehlo/blob/main/docs/spec.md#batch_norm_inference

Exemple:

%result = "mhlo.batch_norm_inference"(%operand, %scale, %offset, %mean, %variance) {
  epsilon = 0.0 : f32,
  feature_index = 2 : i64
} : (tensor<2x2x2xf32>, tensor<2xf32>, tensor<2xf32>, tensor<2xf32>, tensor<2xf32>) -> tensor<2x2x2xf32>

Traits : AlwaysSpeculatableImplTrait , InferTensorType

Interfaces : ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effets : MemoryEffects::Effect{}

Attributs:

Opérandes:

Résultats:

mhlo.batch_norm_training (mhlo::BatchNormTrainingOp)

Opération BatchNormTraining

Calcule la moyenne et la variance sur les dimensions de lot et spatiales et normalise le tenseur operand , pour chaque fonctionnalité de la dimension feature_index et produit des tenseurs output , batch_mean et batch_var .

Voir : https://github.com/openxla/stablehlo/blob/main/docs/spec.md#batch_norm_training

Exemple:

%output, %batch_mean, %batch_var = "mhlo.batch_norm_training"(%operand, %scale, %offset) {
  epsilon = 0.0 : f32,
  feature_index = 2 : i64
} : (tensor<2x2x2xf32>, tensor<2xf32>, tensor<2xf32>) -> (tensor<2x2x2xf32>, tensor<2xf32>, tensor<2xf32>)

Traits : AlwaysSpeculatableImplTrait , InferTensorType

Interfaces : ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effets : MemoryEffects::Effect{}

Attributs:

Opérandes:

Résultats:

mhlo.bitcast (mhlo::BitcastOp)

Opération Bitcast

Syntaxe:

operation ::= `mhlo.bitcast` operands attr-dict `:` functional-type(operands, results)

Cette opération est privée pour le compilateur XLA, elle n'a donc pas encore de spécification.

De manière informelle, cette opération modifie la forme de l’entrée de telle sorte que la disposition physique des éléments reste inchangée.

Cette opération nécessite des informations de mise en page pour donner un sens à la « disposition physique des éléments », et la prise en charge de la mise en page dans MHLO est actuellement en cours de développement.

Exemple:

%0 = mhlo.bitcast %arg0 : (tensor<3x4xf32>) -> tensor<3x4x1xf32>

Traits : AlwaysSpeculatableImplTrait

Interfaces : ConditionallySpeculatable , NoMemoryEffect (MemoryEffectOpInterface)

Effets : MemoryEffects::Effect{}

Opérandes:

Résultats:

mhlo.bitcast_convert (mhlo::BitcastConvertOp)

Opération BitcastConvert

Syntaxe:

operation ::= `mhlo.bitcast_convert` operands attr-dict `:` functional-type(operands, results)

Exécute une opération de conversion de bits sur le tenseur operand et produit un tenseur result où les bits de l'ensemble du tenseur operand sont réinterprétés à l'aide du type du tenseur result .

Voir : https://github.com/openxla/stablehlo/blob/main/docs/spec.md#bitcast_convert

Exemple:

%result = mhlo.bitcast_convert %operand : (tensor<2xf32>) -> tensor<2x4xi8>

Traits : AlwaysSpeculatableImplTrait

Interfaces : ConditionallySpeculatable , InferShapedTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effets : MemoryEffects::Effect{}

Opérandes:

Résultats:

mhlo.broadcast (mhlo::BroadcastOp)

Opération de diffusion

Cette opération est sur le point de quitter StableHLO, elle n'est donc pas incluse dans la spécification : https://github.com/openxla/stablehlo/issues/3

De manière informelle, cette opération fait la même chose que la diffusion de XLA : https://www.tensorflow.org/xla/operation_semantics#broadcast

Exemple:

%result = mhlo.broadcast %operand, sizes = [1, 2] : (tensor<3xi32>) -> tensor<1x2x3xi32>

Traits : AlwaysSpeculatableImplTrait , InferTensorType , SameOperandsAndResultElementType

Interfaces : ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effets : MemoryEffects::Effect{}

Attributs:

Opérandes:

Résultats:

mhlo.broadcast_in_dim (mhlo :: BroadcastInDimOp)

Fonctionnement BroadcastInDim

Étend les dimensions et/ou le rang d'un tenseur d'entrée en dupliquant les données dans le tenseur operand et produit un tenseur result .

Voir : https://github.com/openxla/stablehlo/blob/main/docs/spec.md#broadcast_in_dim

Exemple:

%result = mhlo.broadcast_in_dim %operand, dims = [2, 1] : (tensor<1x3xi32>) -> tensor<2x3x2xi32>

Traits : AlwaysSpeculatableImplTrait , HLO_CompatibleOperandsAndResultElementType

Interfaces : ConditionallySpeculatable , NoMemoryEffect (MemoryEffectOpInterface)

Effets : MemoryEffects::Effect{}

Attributs:

Opérandes:

Résultats:

mhlo.case (mhlo::CaseOp)

Opération de cas

Produit la sortie de l'exécution d'une seule function à partir branches en fonction de la valeur de index .

Voir : https://github.com/openxla/stablehlo/blob/main/docs/spec.md#case

Exemple:

%result0, %result1 = "mhlo.case"(%index) ({
  mhlo.return %result_branch0, %result_branch0 : tensor<2xi64>, tensor<2xi64>
}, {
  mhlo.return %result_branch1, %result_branch1 : tensor<2xi64>, tensor<2xi64>
}) : (tensor<i32>) -> (tensor<2xi64>, tensor<2xi64>)

Traits : RecursiveMemoryEffects , SingleBlockImplicitTerminator<ReturnOp> , SingleBlock

Interfaces : InferTypeOpInterface

Opérandes:

Résultats:

mhlo.cbrt (mhlo::CbrtOp)

Opération CBRT

Syntaxe:

operation ::= `mhlo.cbrt` $operand attr-dict
              `:` custom<SameOperandsAndResultType>(type($operand), type($result))

Effectue une opération de racine cubique élément par élément sur le tenseur operand et produit un tenseur result .

Voir : https://github.com/openxla/stablehlo/blob/main/docs/spec.md#cbrt

Exemple:

%result = mhlo.cbrt %operand : tensor<4xf32>

Traits : AlwaysSpeculatableImplTrait , CompatibleOperandsAndResultType , Elementwise , SameOperandsAndResultShape

Interfaces : ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effets : MemoryEffects::Effect{}

Attributs:

Opérandes:

Résultats:

mhlo.ceil (mhlo::CeilOp)

Opération de plafond

Syntaxe:

operation ::= `mhlo.ceil` $operand attr-dict
              `:` custom<SameOperandsAndResultType>(type($operand), type($result))

Effectue une cellule élément par élément du tenseur operand et produit un tenseur result .

Voir : https://github.com/openxla/stablehlo/blob/main/docs/spec.md#ceil

Exemple:

%result = mhlo.ceil %operand : tensor<5xf32>

Traits : AlwaysSpeculatableImplTrait , CompatibleOperandsAndResultType , Elementwise , SameOperandsAndResultShape

Interfaces : ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effets : MemoryEffects::Effect{}

Opérandes:

Résultats:

mhlo.cholesky (mhlo::CholeskyOp)

Opération Cholesky

Calcule la décomposition de Cholesky d'un lot de matrices.

Voir : https://github.com/openxla/stablehlo/blob/main/docs/spec.md#cholesky

Exemple:

%result = mhlo.cholesky %a, lower = true : tensor<3x3xf32>

Traits : AlwaysSpeculatableImplTrait , InferTensorType , SameOperandsAndResultElementType

Interfaces : ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effets : MemoryEffects::Effect{}

Attributs:

Opérandes:

Résultats:

mhlo.clamp (mhlo::ClampOp)

Fonctionnement de la pince

Syntaxe:

operation ::= `mhlo.clamp` $min `,` $operand `,` $max attr-dict
              `:` custom<SameOperandsAndResultType>(type($min), type($operand), type($max), type($result))

Fixe chaque élément du tenseur operand entre une valeur minimale et maximale et produit un tenseur result .

Voir : https://github.com/openxla/stablehlo/blob/main/docs/spec.md#clamp

Exemple:

%result = mhlo.clamp %min, %operand, %max : tensor<3xi32>

Traits : AlwaysSpeculatableImplTrait , HLO_BroadcastingElementwise , InferTensorType , SameOperandsAndResultElementType

Interfaces : ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effets : MemoryEffects::Effect{}

Opérandes:

Résultats:

mhlo.collective_broadcast (mhlo::CollectiveBroadcastOp)

Opération de diffusion collective

Dans chaque groupe de processus de la grille de processus, envoyez la valeur du tenseur d' operand du processus source aux processus cibles et produisez un tenseur result .

Voir : https://github.com/openxla/stablehlo/blob/main/docs/spec.md#collective_broadcast

Exemple:

%result = "mhlo.collective_broadcast"(%operand) {
  replica_groups = dense<[[0, 1]]> : tensor<1x2xi64>,
  channel_handle = #mhlo.channel_handle<handle = 0, type = 0>
} : (tensor<1x2xi64>) -> tensor<1x2xi64>

Traits : CompatibleOperandsAndResultType

Interfaces : InferShapedTypeOpInterface , InferTypeOpInterface

Attributs:

Opérandes:

Résultats:

mhlo.collective_permute (mhlo::CollectivePermuteOp)

Opération CollectivePermute

Dans chaque groupe de processus de la grille de processus, envoie la valeur du tenseur d' operand du processus source au processus cible et produit un tenseur result .

Voir : https://github.com/openxla/stablehlo/blob/main/docs/spec.md#collective_permute

Exemple:

%result = "mhlo.collective_permute"(%operand) {
  source_target_pairs = dense<[[0, 1], [1, 2]]> : tensor<2x2xi64>,
  // channel_id = 0
  channel_handle = #mhlo.channel_handle<handle = 0, type = 0>
} : (tensor<4x2xf32>) -> tensor<4x2xf32>

Traits : AlwaysSpeculatableImplTrait , CompatibleOperandsAndResultType

Interfaces : ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effets : MemoryEffects::Effect{}

Attributs:

Opérandes:

Résultats:

mhlo.compare (mhlo :: CompareOp)

Comparer les opérations

Syntaxe:

operation ::= `mhlo.compare` $comparison_direction `,` $lhs `,` $rhs (`,` $compare_type^)?
              attr-dict `:` functional-type(operands, results)

Effectue une comparaison élément par élément des tenseurs lhs et rhs selon comparison_direction et compare_type , et produit un tenseur result .

Voir : https://github.com/openxla/stablehlo/blob/main/docs/spec.md#compare

Exemple:

%result = mhlo.compare LT, %lhs, %rhs, FLOAT : (tensor<2xf32>, tensor<2xf32>) -> tensor<2xi1>

Traits : AlwaysSpeculatableImplTrait , Elementwise , InferTensorType , SameOperandsAndResultShape , SameOperandsElementType

Interfaces : ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effets : MemoryEffects::Effect{}

Attributs:

Opérandes:

Résultats:

mhlo.complex (mhlo::ComplexOp)

Opération complexe

Syntaxe:

operation ::= `mhlo.complex` operands attr-dict
              `:` custom<ComplexOpType>(type($lhs), type($rhs), type($result))

Effectue une conversion élément par élément en une valeur complexe à partir d'une paire de valeurs réelles et imaginaires, lhs et rhs , et produit un tenseur result .

Voir : https://github.com/openxla/stablehlo/blob/main/docs/spec.md#complex

Exemple:

%result = mhlo.complex %lhs, %rhs : tensor<2xcomplex<f32>>

Traits : AlwaysSpeculatableImplTrait , Elementwise , SameOperandsAndResultShape , SameOperandsElementType

Interfaces : ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effets : MemoryEffects::Effect{}

Opérandes:

Résultats:

mhlo.composite (mhlo::CompositeOp)

Opération composite

Syntaxe:

operation ::= `mhlo.composite` $name $inputs attr-dict `:` functional-type(operands, results)

Encapsulates an operation made up (composed) of other StableHLO operations, taking inputs and composite_attributes and producing results . The semantics of the op are implemented by the decomposition attribute. The composite op can be replaced with its decomposition without changing program semantics. In cases where inlining the decomposition does not provide the same op semantics, prefer using custom_call .

The version field (defaults to 0 ) is used to denote when a composite's semantics change.

See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#composite

Exemple:

%results = mhlo.composite "my.op" %arg0, %arg1 {
  decomposition = @my_op,
  composite_attributes = { my_attribute = "my_value" },
  version = 1 : i32
} : (tensor<f32>, tensor<f32>) -> tensor<f32>

Interfaces: SymbolUserOpInterface

Attributs:

Operands:

Résultats:

mhlo.concatenate (mhlo::ConcatenateOp)

Concatenate operation

Concatenates a variadic number of tensors in inputs along dimension dimension in the same order as the given arguments and produces a result tensor.

See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#concatenate

Exemple:

%result = mhlo.concatenate %input0, %input1, dim = 0 : (tensor<3x2xi64>, tensor<1x2xi64>) -> tensor<4x2xi64>

Traits: AlwaysSpeculatableImplTrait , SameOperandsAndResultElementType

Interfaces: ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Attributs:

Operands:

Résultats:

mhlo.constant (mhlo::ConstantOp)

Constant operation

Produces an output tensor from a constant value .

See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#constant

Exemple:

%output = mhlo.constant dense<[[0.0, 1.0], [2.0, 3.0]]> : tensor<2x2xf32>

Traits: AlwaysSpeculatableImplTrait , ConstantLike

Interfaces: ConditionallySpeculatable , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Attributs:

Résultats:

mhlo.convert (mhlo::ConvertOp)

Convert operation

Syntaxe:

operation ::= `mhlo.convert` $operand attr-dict
              `:` custom<SameOperandsAndResultType>(type($operand), type($result))

Performs an element-wise conversion from one element type to another on operand tensor and produces a result tensor.

See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#convert

Exemple:

%result = mhlo.convert %operand : (tensor<3xi32>) -> tensor<3xcomplex<f32>>

Traits: AlwaysSpeculatableImplTrait , Elementwise , SameOperandsAndResultShape

Interfaces: ConditionallySpeculatable , InferShapedTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Operands:

Résultats:

mhlo.convolution (mhlo::ConvolutionOp)

Convolution operation

Syntaxe:

operation ::= `mhlo.convolution` `(`operands`)`
              `dim_numbers` `=` custom<ConvolutionDimensions>($dimension_numbers) `,`
              `window` `=` `{` custom<WindowAttributes>($window_strides, $padding,
              $lhs_dilation, $rhs_dilation,
              $window_reversal) `}`
              attr-dict `:` functional-type(operands, results)

Computes dot products between windows of lhs and slices of rhs and produces result .

See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#convolution

Exemple:

%result = "mhlo.convolution"(%lhs, %rhs) {
  window_strides = dense<4> : tensor<2xi64>,
  padding = dense<0> : tensor<2x2xi64>,
  lhs_dilation = dense<2> : tensor<2xi64>,
  rhs_dilation = dense<1> : tensor<2xi64>,
  window_reversal = dense<false> : tensor<2xi1>,
  dimension_numbers = #mhlo.conv<[b, 0, 1, f]x[0, 1, i, o]->[b, 0, 1, f]>,
  feature_group_count = 1 : i64,
  batch_group_count = 1 : i64,
  precision_config = [#stablehlo<precision DEFAULT>, #stablehlo<precision DEFAULT>]
} : (tensor<1x4x4x1xi32>, tensor<3x3x1x1xi32>) -> tensor<1x2x2x1xi32>

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Attributs:

Operands:

Résultats:

mhlo.copy (mhlo::CopyOp)

Copy operation

Syntaxe:

operation ::= `mhlo.copy` operands attr-dict
              `:` custom<SameOperandsAndResultType>(type($operand), type($result))

This operation is private to the XLA compiler, so it is does not yet have a specification.

Informally, this operation a copy of operand . Depending on the metadata attached to the operation, it can behave quite differently from a no-op.

Exemple:

%0 = mhlo.copy %arg0 : tensor<f32>

Traits: AlwaysSpeculatableImplTrait , CompatibleOperandsAndResultType , Elementwise

Interfaces: ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Attributs:

Operands:

Résultats:

mhlo.cosh (mhlo::CoshOp)

Cosh operation

Syntaxe:

operation ::= `mhlo.cosh` $operand attr-dict
              `:` custom<SameOperandsAndResultType>(type($operand), type($result))

Performs element-wise cosh operation on operand tensor and produces a result tensor.

Exemple:

%result = mhlo.cosh %operand : tensor<2x2xf32>

Traits: CompatibleOperandsAndResultType , Elementwise , SameOperandsAndResultShape

Interfaces: InferShapedTypeOpInterface , InferTypeOpInterface

Operands:

Résultats:

mhlo.cosine (mhlo::CosineOp)

Cosine operation

Syntaxe:

operation ::= `mhlo.cosine` $operand attr-dict
              `:` custom<SameOperandsAndResultType>(type($operand), type($result))

Performs element-wise cosine operation on operand tensor and produces a result tensor.

See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#cosine

Exemple:

%result = mhlo.cosine %operand : tensor<2xf32>

Traits: AlwaysSpeculatableImplTrait , CompatibleOperandsAndResultType , Elementwise , SameOperandsAndResultShape

Interfaces: ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Attributs:

Operands:

Résultats:

mhlo.count_leading_zeros (mhlo::ClzOp)

Clz operation

Syntaxe:

operation ::= `mhlo.count_leading_zeros` $operand attr-dict
              `:` custom<SameOperandsAndResultType>(type($operand), type($result))

Performs element-wise count of the number of leading zero bits in the operand tensor and produces a result tensor.

See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#count_leading_zeros

Exemple:

%result = mhlo.count_leading_zeros %operand : tensor<2x2xi8>

Traits: AlwaysSpeculatableImplTrait , CompatibleOperandsAndResultType , Elementwise , SameOperandsAndResultShape

Interfaces: ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Operands:

Résultats:

mhlo.create_token (mhlo::CreateTokenOp)

CreateToken operation

Syntaxe:

operation ::= `mhlo.create_token` attr-dict `:` type(results)

This operation is on its way out of StableHLO, so it is not included in the specification: https://github.com/openxla/stablehlo/issues/3

Informally, this operation does the same thing as AfterAllOp with 0 inputs: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#after_all

Exemple:

%output = mhlo.create_token : !mhlo.token

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Résultats:

mhlo.cross-replica-sum (mhlo::CrossReplicaSumOp)

CrossReplicaSum operation

This operation is on its way out of StableHLO, so it is not included in the specification: https://github.com/openxla/stablehlo/issues/3

Informally, this operation does the same thing as AllReduceOp with channel_id = 0 , use_global_device_ids = false and computation implementing addition: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#all_reduce

Exemple:

%result = "mhlo.cross-replica-sum"(%operand) {
  replica_groups = dense<[[0, 1]]> : tensor<1x2xi64>
} : (tensor<4xf32>) -> tensor<4xf32>

Traits: AlwaysSpeculatableImplTrait , CompatibleOperandsAndResultType

Interfaces: ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Attributs:

Operands:

Résultats:

mhlo.custom_call (mhlo::CustomCallOp)

CustomCall operation

Syntaxe:

operation ::= `mhlo.custom_call` custom<CustomCallTarget>($call_target_name) `(` $inputs `)`
              attr-dict `:` functional-type(operands, results)

Encapsulates an implementation-defined operation call_target_name that takes inputs and called_computations and produces results .

See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#custom_call

Exemple:

%results = "mhlo.custom_call"(%input0) {
  call_target_name = "foo",
  has_side_effect = false,
  backend_config = "bar",
  api_version = 1 : i32,
  called_computations = [@foo]
} : (tensor<f32>) -> tensor<f32>

A custom call invokes code external to XLA. The `inputs` are passed to the
external code, and the external code is expected to produce a result of the
given type. The exact mechanism is backend-specific. For example, in the CPU
backend, a call instruction is emitted which targets a symbol with the name
`call_target_name`.

If XLA runtime is enabled for a backend, then custom calls use the runtime
custom call calling convention to call into the external functions. This
calling convention defines an ABI for encoding arguments, attributes and
results.

Depending on the API version there are two ways to pass extra bits of static
information to the external function:

1. For `API_VERSION_TYPED_FFI` custom calls `backend_config` must be a
   dictionary attribute, that will be encoded according to the custom call
   calling convention and passed to the external function as the attributes
   argument. External code is expected to use declarative bindings (see
   `xla/runtime/custom_call.h`) to decode them at run time. These custom
   calls are only supported if XLA uses XLA runtime.

2. For previous API versions it is the user responsibility to encode extra
   bits of static information as a string `backend_config` attribute, and
   decode it at run time.

Interfaces: MemoryEffectOpInterface

Attributs:

Operands:

Résultats:

mhlo.divide (mhlo::DivOp)

Div operation

Syntaxe:

operation ::= `mhlo.divide` $lhs `,` $rhs attr-dict
              `:` custom<SameOperandsAndResultType>(type($lhs), type($rhs), type($result))

Performs element-wise division of dividend lhs and divisor rhs tensors and produces a result tensor.

See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#divide

Exemple:

%result = mhlo.divide %lhs, %rhs : tensor<4xf32>

Traits: AlwaysSpeculatableImplTrait , CompatibleOperandsAndResultType , Elementwise , SameOperandsAndResultShape

Interfaces: ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Operands:

Résultats:

mhlo.domain (mhlo::DomainOp)

Domain operation

This operation is private to the XLA compiler, so it is does not yet have a specification.

Informally, these operations are used to group instructions with the same DomainMetadata property. ShardingMetadata is the main use case today to group instructions on the same device. Domain instructions provide two major benefits:

• Prevent unintentionally optimizing instructions across domains.
• Automatically assign the metadata of the instructions created in the domain. Without domain instructions, each HLO optimization pass would have to check and propagate the metadata, which would be easy to miss and also adds complexity to the compiler. Since domain instructions connect two different domains, each domain instruction is associated with two DomainMetadata -- one on the operand side and one on the user side of the domain.

Traits: AlwaysSpeculatableImplTrait , CompatibleOperandsAndResultType

Interfaces: ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Attributs:

Operands:

Résultats:

mhlo.dot (mhlo::DotOp)

Dot operation

This operation is on its way out of StableHLO, so it is not included in the specification: https://github.com/openxla/stablehlo/issues/3

Informally, this operation does the same thing as XLA's Dot: https://www.tensorflow.org/xla/operation_semantics#dot

Exemple:

%0 = mhlo.dot %arg0, %arg1 : (tensor<1x2xi32>, tensor<2x1xi32>) -> tensor<1x1xi32>

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Attributs:

Operands:

Résultats:

mhlo.dot_general (mhlo::DotGeneralOp)

DotGeneral operation

Computes dot products between slices of lhs and slices of rhs and produces a result tensor.

See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#dot_general

Exemple:

%result = "mhlo.dot_general"(%lhs, %rhs) {
  dot_dimension_numbers = #mhlo.dot<
    lhs_batching_dimensions = [0],
    rhs_batching_dimensions = [0],
    lhs_contracting_dimensions = [2],
    rhs_contracting_dimensions = [1]
  >,
  precision_config = [#stablehlo<precision DEFAULT>, #stablehlo<precision DEFAULT>]
} : (tensor<2x2x2xi32>, tensor<2x2x2xi32>) -> tensor<2x2x2xi32>

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable , InferShapedTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Attributs:

Operands:

Résultats:

mhlo.dynamic_broadcast_in_dim (mhlo::DynamicBroadcastInDimOp)

DynamicBroadcastInDim operation

This operation is functionally identical to broadcast_in_dim op, but the result shape is specified dynamically via output_dimensions .

It also accepts optional attributes to express static knowledge about the expanding behavior of dimensions. If not specified, all dimensions are assumed to be possibly expanding. The sets of dimensions that are known to be expanding and the set of dimensions that are known to be non-expanding must be disjoint and they must be a subset of the operand's dimensions.

See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#dynamic_broadcast_in_dim

Exemple:

%operand = mhlo.constant dense<[[1, 2, 3]]> : tensor<1x3xi64>
%output_dimensions = mhlo.constant dense<[2, 3, 2]> : tensor<3xi64>
%result = "mhlo.dynamic_broadcast_in_dim"(%operand, %output_dimensions) {
  broadcast_dimensions = array<i64: 2, 1>,
  known_expanding_dimensions = array<i64: 0>,
  known_nonexpanding_dimensions = array<i64: 1>
} : (tensor<1x3xi64>, tensor<3xi64>) -> tensor<2x3x2xi64>

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable , InferShapedTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Attributs:

Operands:

Résultats:

mhlo.dynamic_conv (mhlo::DynamicConvOp)

DynamicConv operation

This operation is a work in progress, so it is not yet included in the specification: https://github.com/openxla/stablehlo/issues/8

Informally, this operation does the same thing as ConvolutionOp except that padding is specified dynamically via d_padding : https://github.com/openxla/stablehlo/blob/main/docs/spec.md#convolution

Exemple:

%result = "mhlo.dynamic_conv"(%lhs, %rhs, %d_padding) {
  window_strides = dense<4> : tensor<2xi64>,
  lhs_dilation = dense<2> : tensor<2xi64>,
  rhs_dilation = dense<1> : tensor<2xi64>,
  window_reversal = dense<false> : tensor<2xi1>,
  dimension_numbers = #mhlo.conv<[b, 0, 1, f]x[0, 1, i, o]->[b, 0, 1, f]>,
  feature_group_count = 1 : i64,
  batch_group_count = 1 : i64,
  precision_config = [#stablehlo<precision DEFAULT>, #stablehlo<precision DEFAULT>]
} : (tensor<1x4x4x1xi32>, tensor<3x3x1x1xi32>, tensor<2x2xi64>) -> tensor<1x2x2x1xi32>

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Attributs:

Operands:

Résultats:

mhlo.dynamic_gather (mhlo::DynamicGatherOp)

DynamicGather operation

This operation is a work in progress, so it is not yet included in the specification: https://github.com/openxla/stablehlo/issues/8

Informally, this operation does the same thing as GatherOp except that slice_sizes are specified dynamically: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#gather

Exemple:

%result = "mhlo.dynamic_gather"(%operand, %start_indices, %slice_sizes) {
  dimension_numbers = #mhlo.gather<
    offset_dims = [2, 3],
    collapsed_slice_dims = [0],
    start_index_map = [0, 2],
    index_vector_dim = 2>,
  indices_are_sorted = false
} : (tensor<3x4x2xi32>, tensor<2x3x2xi64>, tensor<3xi64>) -> tensor<2x3x2x2xi32>

Traits: AlwaysSpeculatableImplTrait , InferTensorType

Interfaces: ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Attributs:

Operands:

Résultats:

mhlo.dynamic_iota (mhlo::DynamicIotaOp)

DynamicIota operation

This operation is functionally identical to iota op, but the result shape is specified dynamically via output_shape .

See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#dynamic_iota

Exemple:

%0 = mhlo.dynamic_iota %arg0, dim = 0 : (tensor<1xindex>) -> tensor<4xi32>

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable , InferShapedTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Attributs:

Operands:

Résultats:

mhlo.dynamic_pad (mhlo::DynamicPadOp)

DynamicPad operation

Syntaxe:

operation ::= `mhlo.dynamic_pad` operands attr-dict `:` functional-type(operands, results)

Dynamically Pads the operand , with amount of padding added at low-end/high-end/interior is passed through input tensors.

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable , InferShapedTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Operands:

Résultats:

mhlo.dynamic_reshape (mhlo::DynamicReshapeOp)

DynamicReshape operation

Syntaxe:

operation ::= `mhlo.dynamic_reshape` operands attr-dict `:` functional-type(operands, results)

This operation is functionally identical to reshape op, but the result shape is specified dynamically via output_shape .

See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#dynamic_reshape

Exemple:

%output_shape = mhlo.constant dense<[3, 2]> : tensor<2xi64>
%result = mhlo.dynamic_reshape %operand, %output_shape : (tensor<2x3xi64>, tensor<2xi64>) -> tensor<3x2xi64>

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable , InferShapedTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Operands:

Résultats:

mhlo.dynamic_slice (mhlo::DynamicSliceOp)

DynamicSlice operation

Extracts a slice from the operand using dynamically-computed starting indices and produces a result tensor.

See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#dynamic_slice

Exemple:

%result = mhlo.dynamic_slice %operand, %start_indices0, %start_indices1, sizes = [2, 2]
  : (tensor<4x4xi32>, tensor<i64>, tensor<i64>) -> tensor<2x2xi32>

Traits: AlwaysSpeculatableImplTrait , InferTensorType

Interfaces: ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Attributs:

Operands:

Résultats:

mhlo.dynamic_update_slice (mhlo::DynamicUpdateSliceOp)

DynamicUpdateSlice operation

Syntaxe:

operation ::= `mhlo.dynamic_update_slice` operands attr-dict `:` functional-type(operands, results)

Produces a result tensor which is equal to the operand tensor except that the slice starting at start_indices is updated with the values in update .

See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#dynamic_update_slice

Exemple:

%result = mhlo.dynamic_update_slice %operand, %update, %start_indices0, %start_indices1
  : (tensor<4x4xi32>, tensor<2x2xi32>, tensor<i64>, tensor<i64>) -> tensor<4x4xi32>

Traits: AlwaysSpeculatableImplTrait , InferTensorType

Interfaces: ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Operands:

Résultats:

mhlo.einsum (mhlo::EinsumOp)

Einsum operation

This operation is on its way out of StableHLO, so it is not included in the specification: https://github.com/openxla/stablehlo/issues/3

Informally, this operation does the same thing as TF's einsum: https://www.tensorflow.org/api_docs/python/tf/einsum

Exemple:

%result = "mhlo.einsum"(%lhs, %rhs) {
  einsum_config = "ab,bc->ac"
} : (tensor<4x16xf32>, tensor<16x4xf32>) -> tensor<4x4xf32>

Traits: AlwaysSpeculatableImplTrait

Interfaces: ConditionallySpeculatable , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Attributs:

Operands:

Résultats:

mhlo.erf (mhlo::ErfOp)

Erf operation

Syntaxe:

operation ::= `mhlo.erf` $operand attr-dict
              `:` custom<SameOperandsAndResultType>(type($operand), type($result))

Performs element-wise erf operation on operand tensor and produces a result tensor.

Exemple:

%result = mhlo.erf %operand : tensor<2x2xf32>

Traits: AlwaysSpeculatableImplTrait , CompatibleOperandsAndResultType , Elementwise , SameOperandsAndResultShape

Interfaces: ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Operands:

Résultats:

mhlo.exponential (mhlo::ExpOp)

Exp operation

Syntaxe:

operation ::= `mhlo.exponential` $operand attr-dict
              `:` custom<SameOperandsAndResultType>(type($operand), type($result))

Performs element-wise exponential operation on operand tensor and produces a result tensor.

See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#exponential

Exemple:

%result = mhlo.exponential %operand : tensor<2x2xf64>

Traits: AlwaysSpeculatableImplTrait , CompatibleOperandsAndResultType , Elementwise , SameOperandsAndResultShape

Interfaces: ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Attributs:

Operands:

Résultats:

mhlo.exponential_minus_one (mhlo::Expm1Op)

Expm1 operation

Syntaxe:

operation ::= `mhlo.exponential_minus_one` $operand attr-dict
              `:` custom<SameOperandsAndResultType>(type($operand), type($result))

Performs element-wise exponential minus one operation on operand tensor and produces a result tensor.

See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#exponential_minus_one

Exemple:

%result = mhlo.exponential_minus_one %operand : tensor<2xf32>

Traits: AlwaysSpeculatableImplTrait , CompatibleOperandsAndResultType , Elementwise , SameOperandsAndResultShape

Interfaces: ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Attributs:

Operands:

Résultats:

mhlo.fft (mhlo::FftOp)

Fft operation

Performs the forward and inverse Fourier transforms for real and complex inputs/outputs.

See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#fft

Exemple:

%result = mhlo.fft %operand, type = FFT, length = [4] : (tensor<4xcomplex<f32>>) -> tensor<4xcomplex<f32>>

Traits: AlwaysSpeculatableImplTrait , InferTensorType

Interfaces: ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Attributs:

Operands:

Résultats:

mhlo.floor (mhlo::FloorOp)

Floor operation

Syntaxe:

operation ::= `mhlo.floor` $operand attr-dict
              `:` custom<SameOperandsAndResultType>(type($operand), type($result))

Performs element-wise floor of operand tensor and produces a result tensor.

See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#floor

Exemple:

%result = mhlo.floor %operand : tensor<2xf32>

Traits: AlwaysSpeculatableImplTrait , CompatibleOperandsAndResultType , Elementwise , SameOperandsAndResultShape

Interfaces: ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Operands:

Résultats:

mhlo.fusion (mhlo::FusionOp)

Fusion operation

This operation is private to the XLA compiler, so it is does not yet have a specification.

Informally, this operation consists of a group of basic ops (represented as a region attached to it). It serves as a hint to the backend that it is beneficial to emit the contained ops into a single loop nest or kernel.

Attributs:

Operands:

Résultats:

mhlo.gather (mhlo::GatherOp)

Gather operation

Gathers slices from operand tensor from offsets specified in start_indices and produces a result tensor.

See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#gather

Exemple:

%result = "mhlo.gather"(%operand, %start_indices) {
  dimension_numbers = #stablehlo.gather<
    offset_dims = [3, 4],
    collapsed_slice_dims = [1],
    operand_batching_dims = [0],
    start_indices_batching_dims = [1],
    start_index_map = [2, 1],
    index_vector_dim = 3>,
  slice_sizes = dense<[0, 2, 2]> : tensor<3xi64>,
  indices_are_sorted = false
} : (tensor<2x3x4x2xi64>, tensor<2x2x3x2xi64>) -> tensor<2x2x3x2x2xi64>

Traits: AlwaysSpeculatableImplTrait , InferTensorType

Interfaces: ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Attributs:

Operands:

Résultats:

mhlo.get_dimension_size (mhlo::GetDimensionSizeOp)

GetDimensionSize operation

Produces the size of the given dimension of the operand .

See https://github.com/openxla/stablehlo/blob/main/docs/spec.md#get_dimension_size

Exemple:

%result = mhlo.get_dimension_size %operand, dim = 1 : (tensor<2x3xf32>) -> tensor<i32>

Traits: AlwaysSpeculatableImplTrait , InferTensorType

Interfaces: ConditionallySpeculatable , InferShapedTypeOpInterface , InferTypeOpInterface , NoMemoryEffect (MemoryEffectOpInterface)

Effects: MemoryEffects::Effect{}

Attributs:

Operands:

Résultats:

mhlo.get_tuple_element (mhlo::GetTupleElementOp)

GetTupleElement operation

Syntaxe:

operation ::= `mhlo.get_tuple_element` $operand `[` $index `]` attr-dict `:` functional-type(operands, results)