عمليات
mhlo.abs
(mhlo::AbsOp)
عملية القيمة المطلقة
بناء الجملة:
operation ::= `mhlo.abs` $operand attr-dict
`:` custom<SameOperandsAndResultType>(type($operand), type($result))
ينفذ عملية القيمة المطلقة للعناصر على موتر operand
وينتج موترًا result
.
راجع: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#abs
مثال:
%result = mhlo.abs %operand : tensor<3xi32>
السمات: AlwaysSpeculatableImplTrait
، Elementwise
، SameOperandsAndResultShape
الواجهات: ConditionallySpeculatable
، InferShapedTypeOpInterface
، InferTypeOpInterface
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
المعاملات:
المعامل | وصف |
---|---|
operand | موتر مصنف لعدد صحيح بدون إشارة 4/8/16/32/64 بت أو نوع f8E4M3B11FNUZ أو نوع f8E4M3FN أو نوع f8E4M3FNUZ أو نوع f8E5M2 أو نوع f8E5M2FNUZ أو نوع float 16 بت أو نوع float 32 بت أو نوع float 64 بت أو bfloat16 أو نوع معقد يحتوي على عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو عدد صحيح موقّع كمي موحد 4/8/16/32 بت أو عدد صحيح موقّع كمي 4/8/16/32 بت لكل محور عدد صحيح موقّع أو 4/8/16/ 32 بت موحد كمي عدد صحيح غير موقع أو 4/8/16/32 بت موحد كمي لكل محور قيم عددية غير موقعة |
نتائج:
نتيجة | وصف |
---|---|
result | موتر مصنف لعدد صحيح بدون إشارة 4/8/16/32/64 بت أو نوع f8E4M3B11FNUZ أو نوع f8E4M3FN أو نوع f8E4M3FNUZ أو نوع f8E5M2 أو نوع f8E5M2FNUZ أو نوع float 16 بت أو نوع float 32 بت أو نوع float 64 بت أو bfloat16 أو 4/8/16/32 بت عدد صحيح موقّع منتظم أو 4/8/16/32 بت موحد مكمّم لكل محور عدد صحيح موقّع أو 4/8/16/32 بت عدد صحيح مكمّم غير موقّع أو 4/8/16/ 32 بت موحد كمي لكل محور قيم عددية غير موقعة |
mhlo.add
(mhlo::AddOp)
إضافة عملية
بناء الجملة:
operation ::= `mhlo.add` $lhs `,` $rhs attr-dict
`:` custom<SameOperandsAndResultType>(type($lhs), type($rhs), type($result))
ينفذ إضافة عنصرية لموترين lhs
و rhs
وينتج موترًا result
.
راجع: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#add
مثال:
%result = mhlo.add %lhs, %rhs : tensor<2x2xi32>
السمات: AlwaysSpeculatableImplTrait
، Commutative
، CompatibleOperandsAndResultType
، Elementwise
، SameOperandsAndResultShape
الواجهات: ConditionallySpeculatable
، InferShapedTypeOpInterface
، InferTypeOpInterface
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
المعاملات:
المعامل | وصف |
---|---|
lhs | موتر مصنف من النوع f8E4M3B11FNUZ أو نوع f8E4M3FN أو نوع f8E4M3FNUZ أو نوع f8E5M2 أو نوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو pred (AKA منطقي أو عدد صحيح 1 بت) أو 4/8 /16/32/64 بت عدد صحيح بدون إشارة أو 4/8/16/32/64 بت عدد صحيح بدون إشارة أو نوع معقد مع عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو 4/8/16/32 بت موحدة كمية عدد صحيح موقّع أو 4/8/16/32 بت عدد صحيح مكمّم غير موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور قيم عددية غير موقّعة |
rhs | موتر مصنف من النوع f8E4M3B11FNUZ أو نوع f8E4M3FN أو نوع f8E4M3FNUZ أو نوع f8E5M2 أو نوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو pred (AKA منطقي أو عدد صحيح 1 بت) أو 4/8 /16/32/64 بت عدد صحيح بدون إشارة أو 4/8/16/32/64 بت عدد صحيح بدون إشارة أو نوع معقد مع عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو 4/8/16/32 بت موحدة كمية عدد صحيح موقّع أو 4/8/16/32 بت عدد صحيح مكمّم غير موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور قيم عددية غير موقّعة |
نتائج:
نتيجة | وصف |
---|---|
result | موتر مصنف من النوع f8E4M3B11FNUZ أو نوع f8E4M3FN أو نوع f8E4M3FNUZ أو نوع f8E5M2 أو نوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو pred (AKA منطقي أو عدد صحيح 1 بت) أو 4/8 /16/32/64 بت عدد صحيح بدون إشارة أو 4/8/16/32/64 بت عدد صحيح بدون إشارة أو نوع معقد مع عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو 4/8/16/32 بت موحدة كمية عدد صحيح موقّع أو 4/8/16/32 بت عدد صحيح مكمّم غير موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور قيم عددية غير موقّعة |
mhlo.add_dependency
(mhlo::AddDependencyOp)
عملية إضافة التبعية
بناء الجملة:
operation ::= `mhlo.add_dependency` operands attr-dict `:` functional-type(operands, results)
هذه العملية خاصة بمترجم XLA، لذا فهي لا تحتوي على مواصفات بعد.
بشكل غير رسمي، هذه العملية معاملين: معامل البيانات والرمز المميز. إخراج العملية هو معامل البيانات. عند استخدامها مع AfterAll، تتيح هذه العملية طلب العمليات غير ذات التأثير الجانبي (تلك التي لا تنتج قيمًا مميزة).
مثال:
%1 = mhlo.add_dependency %arg0, %0 : (tensor<3x4xf32>, !mhlo.token) -> tensor<3x4xf32>
السمات: AlwaysSpeculatableImplTrait
الواجهات: ConditionallySpeculatable
، InferTypeOpInterface
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
المعاملات:
المعامل | وصف |
---|---|
operand | موتر مصنف من النوع f8E4M3B11FNUZ أو نوع f8E4M3FN أو نوع f8E4M3FNUZ أو نوع f8E5M2 أو نوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو pred (AKA منطقي أو عدد صحيح 1 بت) أو 4/8 /16/32/64 بت عدد صحيح بدون إشارة أو 4/8/16/32/64 بت عدد صحيح بدون إشارة أو نوع معقد مع عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو 4/8/16/32 بت موحدة كمية عدد صحيح موقّع أو 4/8/16/32 بت موحد قيم عددية غير موقعة أو موتر مرتب من 4/8/16/32 بت موحد مكمّم لكل محور عدد صحيح موقّع أو 4/8/16/32 بت موحد مكمّم لكل محور قيم عدد صحيح غير موقعة أو رمز مميز |
token | رمز مميز |
نتائج:
نتيجة | وصف |
---|---|
output | موتر مصنف من النوع f8E4M3B11FNUZ أو نوع f8E4M3FN أو نوع f8E4M3FNUZ أو نوع f8E5M2 أو نوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو pred (AKA منطقي أو عدد صحيح 1 بت) أو 4/8 /16/32/64 بت عدد صحيح بدون إشارة أو 4/8/16/32/64 بت عدد صحيح بدون إشارة أو نوع معقد مع عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو 4/8/16/32 بت موحدة كمية عدد صحيح موقّع أو 4/8/16/32 بت موحد قيم عددية غير موقعة أو موتر مرتب من 4/8/16/32 بت موحد مكمّم لكل محور عدد صحيح موقّع أو 4/8/16/32 بت موحد مكمّم لكل محور قيم عدد صحيح غير موقعة أو رمز مميز |
mhlo.after_all
(mhlo::AfterAllOp)
بعد كل العملية
بناء الجملة:
operation ::= `mhlo.after_all` $inputs attr-dict
`:` custom<VariadicSameOperandsAndResultType>(ref($inputs), type($inputs), type($result))
التأكد من تنفيذ العمليات المنتجة inputs
قبل أي عمليات تعتمد على result
.
راجع: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#after_all
مثال:
%result = mhlo.after_all %input0, %input1 : !mhlo.token
السمات: AlwaysSpeculatableImplTrait
الواجهات: ConditionallySpeculatable
، InferTypeOpInterface
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
المعاملات:
المعامل | وصف |
---|---|
inputs | متغير من الرمز المميز |
نتائج:
نتيجة | وصف |
---|---|
result | رمز مميز |
mhlo.all_gather
(mhlo::AllGatherOp)
عملية AllGather
داخل كل مجموعة عمليات في شبكة العمليات، يتم تسلسل قيم موتر المعامل من كل عملية على طول all_gather_dim
وينتج موتر نتيجة. يتم تطبيق computation
بشكل منفصل لكل معامل في operands
، مما ينتج نتيجة واحدة لكل معامل.
راجع: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#all_gather
مثال:
%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>
السمات: SameOperandsAndResultElementType
صفات:
يصف | نوع ملير | وصف |
---|---|---|
all_gather_dim | ::mlir::IntegerAttr | سمة عدد صحيح بدون إشارة 64 بت |
replica_groups | ::mlir::DenseIntElementsAttr | سمة العناصر الصحيحة بدون إشارة 64 بت |
channel_handle | ::mlir::mhlo::ChannelHandleAttr | عددان صحيحان 64 بت "مقبض" و"نوع" |
use_global_device_ids | ::mlir::UnitAttr | سمة الوحدة |
المعاملات:
المعامل | وصف |
---|---|
operands | متغير من الموتر المصنف من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو pred (AKA منطقي أو عدد صحيح 1 بت) أو 4 /8/16/32/64 عدد صحيح بدون إشارة أو 4/8/16/32/64 بت عدد صحيح بدون إشارة أو نوع معقد مع عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو 4/8/16/32 بت عدد صحيح موقّع موحّد أو 4/8/16/32 بت عدد صحيح مكمّم غير موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح غير موقّع قيم |
نتائج:
نتيجة | وصف |
---|---|
«غير مسمى» | متغير من الموتر المصنف من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو pred (AKA منطقي أو عدد صحيح 1 بت) أو 4 /8/16/32/64 عدد صحيح بدون إشارة أو 4/8/16/32/64 بت عدد صحيح بدون إشارة أو نوع معقد مع عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو 4/8/16/32 بت عدد صحيح موقّع موحّد أو 4/8/16/32 بت عدد صحيح مكمّم غير موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح غير موقّع قيم |
mhlo.all_reduce
(mhlo::AllReduceOp)
كل تقليل العملية
داخل كل مجموعة عمليات في شبكة العمليات، يتم تطبيق computation
دالة الاختزال على قيم موتر المعامل من كل عملية وينتج موتر نتيجة. يتم تطبيق computation
بشكل منفصل لكل معامل في operands
، مما ينتج نتيجة واحدة لكل معامل.
راجع: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#all_reduce
مثال:
%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>
السمات: InferTensorType
، SingleBlockImplicitTerminator<ReturnOp>
، SingleBlock
الواجهات: InferShapedTypeOpInterface
، InferTypeOpInterface
صفات:
يصف | نوع ملير | وصف |
---|---|---|
replica_groups | ::mlir::DenseIntElementsAttr | سمة العناصر الصحيحة بدون إشارة 64 بت |
channel_handle | ::mlir::mhlo::ChannelHandleAttr | عددان صحيحان 64 بت "مقبض" و"نوع" |
use_global_device_ids | ::mlir::UnitAttr | سمة الوحدة |
المعاملات:
المعامل | وصف |
---|---|
operands | متغير من الموتر المصنف من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو pred (AKA منطقي أو عدد صحيح 1 بت) أو 4 /8/16/32/64 عدد صحيح بدون إشارة أو 4/8/16/32/64 بت عدد صحيح بدون إشارة أو نوع معقد مع عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو 4/8/16/32 بت عدد صحيح موقّع موحّد أو 4/8/16/32 بت عدد صحيح مكمّم غير موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح غير موقّع قيم |
نتائج:
نتيجة | وصف |
---|---|
«غير مسمى» | متغير من الموتر المصنف من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو pred (AKA منطقي أو عدد صحيح 1 بت) أو 4 /8/16/32/64 عدد صحيح بدون إشارة أو 4/8/16/32/64 بت عدد صحيح بدون إشارة أو نوع معقد مع عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو 4/8/16/32 بت عدد صحيح موقّع موحّد أو 4/8/16/32 بت عدد صحيح مكمّم غير موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح غير موقّع قيم |
mhlo.all_to_all
(mhlo::AllToAllOp)
عملية الكل للكل
داخل كل مجموعة عمليات في شبكة العملية، يتم تقسيم قيم موتر operand
على طول split_dimension
إلى أجزاء، ويبعثر الأجزاء المقسمة بين العمليات، ويسلسل الأجزاء المتناثرة على طول concat_dimension
وينتج موتر result
.
راجع: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#all_to_all
مثال:
%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>
السمات: AlwaysSpeculatableImplTrait
، InferTensorType
، SameOperandsElementType
، SameOperandsShape
، SameVariadicOperandSize
الواجهات: ConditionallySpeculatable
، InferShapedTypeOpInterface
، InferTypeOpInterface
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
صفات:
يصف | نوع ملير | وصف |
---|---|---|
split_dimension | ::mlir::IntegerAttr | سمة عدد صحيح بدون إشارة 64 بت |
concat_dimension | ::mlir::IntegerAttr | سمة عدد صحيح بدون إشارة 64 بت |
split_count | ::mlir::IntegerAttr | سمة عدد صحيح بدون إشارة 64 بت |
replica_groups | ::mlir::DenseIntElementsAttr | سمة العناصر الصحيحة بدون إشارة 64 بت |
channel_handle | ::mlir::mhlo::ChannelHandleAttr | عددان صحيحان 64 بت "مقبض" و"نوع" |
المعاملات:
المعامل | وصف |
---|---|
operand | متغير من الموتر المصنف من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو pred (AKA منطقي أو عدد صحيح 1 بت) أو 4 /8/16/32/64 عدد صحيح بدون إشارة أو 4/8/16/32/64 بت عدد صحيح بدون إشارة أو نوع معقد مع عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو 4/8/16/32 بت عدد صحيح موقّع موحّد أو 4/8/16/32 بت عدد صحيح مكمّم غير موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح غير موقّع قيم |
نتائج:
نتيجة | وصف |
---|---|
«غير مسمى» | متغير من الموتر المصنف من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو pred (AKA منطقي أو عدد صحيح 1 بت) أو 4 /8/16/32/64 عدد صحيح بدون إشارة أو 4/8/16/32/64 بت عدد صحيح بدون إشارة أو نوع معقد مع عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو 4/8/16/32 بت عدد صحيح موقّع موحّد أو 4/8/16/32 بت عدد صحيح مكمّم غير موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح غير موقّع قيم |
mhlo.and
(mhlo::AndOp)
والتشغيل
بناء الجملة:
operation ::= `mhlo.and` $lhs `,` $rhs attr-dict
`:` custom<SameOperandsAndResultType>(type($lhs), type($rhs), type($result))
ينفذ عنصرًا حكيمًا وممتدين lhs
و rhs
وينتج موترًا result
راجع: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#and
مثال:
%result = mhlo.and %lhs, %rhs : tensor<2x2xi32>
السمات: AlwaysSpeculatableImplTrait
، Commutative
، CompatibleOperandsAndResultType
، Elementwise
، SameOperandsAndResultShape
الواجهات: ConditionallySpeculatable
، InferShapedTypeOpInterface
، InferTypeOpInterface
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
المعاملات:
المعامل | وصف |
---|---|
lhs | موتر مرتب لـ pred (AKA منطقي أو عدد صحيح 1 بت) أو عدد صحيح بدون إشارة 4/8/16/32/64 بت أو قيم عدد صحيح بدون إشارة 4/8/16/32/64 بت |
rhs | موتر مرتب لـ pred (AKA منطقي أو عدد صحيح 1 بت) أو عدد صحيح بدون إشارة 4/8/16/32/64 بت أو قيم عدد صحيح بدون إشارة 4/8/16/32/64 بت |
نتائج:
نتيجة | وصف |
---|---|
result | موتر مصنف من النوع f8E4M3B11FNUZ أو نوع f8E4M3FN أو نوع f8E4M3FNUZ أو نوع f8E5M2 أو نوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو pred (AKA منطقي أو عدد صحيح 1 بت) أو 4/8 /16/32/64 بت عدد صحيح بدون إشارة أو 4/8/16/32/64 بت عدد صحيح بدون إشارة أو نوع معقد مع عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو 4/8/16/32 بت موحدة كمية عدد صحيح موقّع أو 4/8/16/32 بت عدد صحيح مكمّم غير موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور قيم عددية غير موقّعة |
mhlo.async_done
(mhlo::AsyncDoneOp)
عملية غير متزامنة
هذه العملية خاصة بمترجم XLA، لذا فهي لا تحتوي على مواصفات بعد.
بشكل غير رسمي، يتم حظر هذه العملية حتى نهاية الحساب غير المتزامن. تقوم بإرجاع النتيجة النهائية للحساب غير المتزامن.
راجع وثائق AsyncStart لمزيد من المعلومات.
الواجهات: InferTypeOpInterface
صفات:
يصف | نوع ملير | وصف |
---|---|---|
called_computation | ::mlir::FlatSymbolRefAttr | السمة المرجعية للرمز المسطح |
execution_thread | ::mlir::StringAttr | سمة السلسلة |
المعاملات:
المعامل | وصف |
---|---|
bundle | async_bundle مع أي مجموعة من الموتر المصنف من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو pred (AKA منطقي أو عدد صحيح 1 بت ) أو عدد صحيح بدون إشارة 4/8/16/32/64 بت أو عدد صحيح بدون إشارة 4/8/16/32/64 بت أو نوع معقد مع عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو 4/8/16/ عدد صحيح موقّع مكمّم 32 بت أو 4/8/16/32 بت عدد صحيح مكمّم غير موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح موقّع أو 4/8/16/32 بت مكمّم موحّد لكل قيم الأعداد الصحيحة غير الموقعة أو قيم الرمز المميز |
نتائج:
نتيجة | وصف |
---|---|
«غير مسمى» | متغير من الموتر المصنف من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو pred (AKA منطقي أو عدد صحيح 1 بت) أو 4 /8/16/32/64 عدد صحيح بدون إشارة أو 4/8/16/32/64 بت عدد صحيح بدون إشارة أو نوع معقد مع عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو 4/8/16/32 بت عدد صحيح موقّع موحّد أو 4/8/16/32 بت عدد صحيح مكمّم غير موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح غير موقّع القيم أو الرمز المميز أو الصف المتداخل مع أي مجموعة من الموتر المصنف من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو التعويم 16 بت أو التعويم 32 بت أو التعويم 64 بت أو النوع bfloat16 أو pred (AKA boolean أو عدد صحيح 1 بت) أو عدد صحيح بدون إشارة 4/8/16/32/64 بت أو عدد صحيح بدون إشارة 4/8/16/32/64 بت أو نوع معقد مع عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو 4 /8/16/32 بت عدد صحيح موقّع كمي أو 4/8/16/32 بت قيم عددية صحيحة غير موقعة أو موتر مرتب من 4/8/16/32 بت موحد كمي لكل محور عدد صحيح موقّع أو 4/8 /16/32 بت موحد مكمّم لكل محور قيم عددية غير موقعة أو قيم رمزية |
mhlo.async_start
(mhlo::AsyncStartOp)
عملية AsyncStart
هذه العملية خاصة بمترجم XLA، لذا فهي لا تحتوي على مواصفات بعد.
بشكل غير رسمي، تبدأ هذه العملية عملية حسابية غير متزامنة.
يتم استخدام هذا عندما تكون هناك وظائف تحتوي على كل من عمليات الانتظار غير المتزامنة (مثل DMAs) والحساب على مؤشر الترابط. على سبيل المثال، قد تتكون الدالة من عملية حسابية، وDMA، وحساب آخر، وDMA ثاني، وحساب نهائي. سيتم تمثيل ذلك على أنه async_start متبوعًا بـ async_update وasync_done. سيقوم async_start بإجراء الحساب الأول على مؤشر الترابط ثم يبدأ DMA. سينتظر async_update حتى يكتمل DMA إذا لم يتم ذلك بعد، ثم ينفذ الحساب الثاني في الوظيفة، ويبدأ DMA الثاني. أخيرًا، سينتظر async_done في DMA الأخير هذا، ثم يقوم بتشغيل الحساب الأخير الذي يجب تشغيله على مؤشر الترابط وإرجاع نتيجة هذا الحساب النهائي.
يتم تمرير operands
إلى الحساب مباشرة called_computation
هي الوظيفة التي سيتم تشغيلها بشكل غير متزامن execution_thread
هو اسم مؤشر الترابط الذي سيتم تشغيله فيه. الخيط الرئيسي يسمى "الرئيسي". جميع المواضيع لها أسماء.
يؤدي هذا إلى إرجاع كل الحالة المطلوبة بين العمليات غير المتزامنة. بعد تعيين المخزن المؤقت، تمثل القيم المرجعة المساحة اللازمة للاحتفاظ بالمدخلات والنتائج وأي لوحات ملاحظات مطلوبة أو تم تحريرها بواسطة عملية المزامنة.
صفات:
يصف | نوع ملير | وصف |
---|---|---|
called_computation | ::mlir::FlatSymbolRefAttr | السمة المرجعية للرمز المسطح |
execution_thread | ::mlir::StringAttr | سمة السلسلة |
المعاملات:
المعامل | وصف |
---|---|
inputs | متغير من الموتر المصنف من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو pred (AKA منطقي أو عدد صحيح 1 بت) أو 4 /8/16/32/64 عدد صحيح بدون إشارة أو 4/8/16/32/64 بت عدد صحيح بدون إشارة أو نوع معقد مع عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو 4/8/16/32 بت عدد صحيح موقّع موحّد أو 4/8/16/32 بت عدد صحيح مكمّم غير موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح غير موقّع القيم أو الرمز المميز أو الصف المتداخل مع أي مجموعة من الموتر المصنف من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو التعويم 16 بت أو التعويم 32 بت أو التعويم 64 بت أو النوع bfloat16 أو pred (AKA boolean أو عدد صحيح 1 بت) أو عدد صحيح بدون إشارة 4/8/16/32/64 بت أو عدد صحيح بدون إشارة 4/8/16/32/64 بت أو نوع معقد مع عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو 4 /8/16/32 بت عدد صحيح موقّع كمي أو 4/8/16/32 بت قيم عددية صحيحة غير موقعة أو موتر مرتب من 4/8/16/32 بت موحد كمي لكل محور عدد صحيح موقّع أو 4/8 /16/32 بت موحد مكمّم لكل محور قيم عددية غير موقعة أو قيم رمزية |
نتائج:
نتيجة | وصف |
---|---|
«غير مسمى» | async_bundle مع أي مجموعة من الموتر المصنف من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو pred (AKA منطقي أو عدد صحيح 1 بت ) أو عدد صحيح بدون إشارة 4/8/16/32/64 بت أو عدد صحيح بدون إشارة 4/8/16/32/64 بت أو نوع معقد مع عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو 4/8/16/ عدد صحيح موقّع مكمّم 32 بت أو 4/8/16/32 بت عدد صحيح مكمّم غير موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح موقّع أو 4/8/16/32 بت مكمّم موحّد لكل قيم الأعداد الصحيحة غير الموقعة أو قيم الرمز المميز |
mhlo.async_update
(mhlo::AsyncUpdateOp)
عملية AsyncUpdate
هذه العملية خاصة بمترجم XLA، لذا فهي لا تحتوي على مواصفات بعد.
بشكل غير رسمي، تقوم هذه العملية بحظر الحسابات غير المتزامنة حتى حاجز المزامنة. يؤدي هذا إلى إرجاع bundle
بعد التشغيل عليها.
راجع وثائق AsyncStart لمزيد من المعلومات.
الواجهات: InferTypeOpInterface
صفات:
يصف | نوع ملير | وصف |
---|---|---|
called_computation | ::mlir::FlatSymbolRefAttr | السمة المرجعية للرمز المسطح |
execution_thread | ::mlir::StringAttr | سمة السلسلة |
المعاملات:
المعامل | وصف |
---|---|
bundle | async_bundle مع أي مجموعة من الموتر المصنف من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو pred (AKA منطقي أو عدد صحيح 1 بت ) أو عدد صحيح بدون إشارة 4/8/16/32/64 بت أو عدد صحيح بدون إشارة 4/8/16/32/64 بت أو نوع معقد مع عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو 4/8/16/ عدد صحيح موقّع مكمّم 32 بت أو 4/8/16/32 بت عدد صحيح مكمّم غير موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح موقّع أو 4/8/16/32 بت مكمّم موحّد لكل قيم الأعداد الصحيحة غير الموقعة أو قيم الرمز المميز |
نتائج:
نتيجة | وصف |
---|---|
«غير مسمى» | async_bundle مع أي مجموعة من الموتر المصنف من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو pred (AKA منطقي أو عدد صحيح 1 بت ) أو عدد صحيح بدون إشارة 4/8/16/32/64 بت أو عدد صحيح بدون إشارة 4/8/16/32/64 بت أو نوع معقد مع عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو 4/8/16/ عدد صحيح موقّع مكمّم 32 بت أو 4/8/16/32 بت عدد صحيح مكمّم غير موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح موقّع أو 4/8/16/32 بت مكمّم موحّد لكل قيم الأعداد الصحيحة غير الموقعة أو قيم الرمز المميز |
mhlo.atan2
(mhlo::Atan2Op)
عملية أتان2
بناء الجملة:
operation ::= `mhlo.atan2` $lhs `,` $rhs attr-dict
`:` custom<SameOperandsAndResultType>(type($lhs), type($rhs), type($result))
ينفذ عملية atan2 من حيث العناصر على موتر lhs
و rhs
وينتج موترًا result
.
انظر: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#atan2
مثال:
%result = mhlo.atan2 %lhs, %rhs : tensor<3xf32>
السمات: AlwaysSpeculatableImplTrait
، CompatibleOperandsAndResultType
، Elementwise
، SameOperandsAndResultShape
الواجهات: ConditionallySpeculatable
، InferShapedTypeOpInterface
، InferTypeOpInterface
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
المعاملات:
المعامل | وصف |
---|---|
lhs | موتر مصنف من النوع f8E4M3B11FNUZ أو نوع f8E4M3FN أو نوع f8E4M3FNUZ أو نوع f8E5M2 أو نوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو نوع معقد مع عناصر تعويم 32 بت أو عناصر تعويم 64 بت أو 4/8/16/32 بت عدد صحيح موقّع وموحد أو 4/8/16/32 بت قيم عدد صحيح مكمّم غير موقّع |
rhs | موتر مصنف من النوع f8E4M3B11FNUZ أو نوع f8E4M3FN أو نوع f8E4M3FNUZ أو نوع f8E5M2 أو نوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو نوع معقد مع عناصر تعويم 32 بت أو عناصر تعويم 64 بت أو 4/8/16/32 بت عدد صحيح موقّع وموحد أو 4/8/16/32 بت قيم عدد صحيح مكمّم غير موقّع |
نتائج:
نتيجة | وصف |
---|---|
result | موتر مصنف من النوع f8E4M3B11FNUZ أو نوع f8E4M3FN أو نوع f8E4M3FNUZ أو نوع f8E5M2 أو نوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو نوع معقد مع عناصر تعويم 32 بت أو عناصر تعويم 64 بت أو 4/8/16/32 بت عدد صحيح موقّع وموحد أو 4/8/16/32 بت قيم عدد صحيح مكمّم غير موقّع |
mhlo.batch_norm_grad
(mhlo::BatchNormGradOp)
عملية BatchNormGrad
يحسب التدرجات لعدة مدخلات من BatchNormTrainingOp backpropagating من grad_output
وينتج grad_operand
و grad_scale
و grad_offset
Tensors.
راجع: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#batch_norm_grad
مثال:
%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>)
السمات: AlwaysSpeculatableImplTrait
، InferTensorType
الواجهات: ConditionallySpeculatable
، InferShapedTypeOpInterface
، InferTypeOpInterface
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
صفات:
يصف | نوع ملير | وصف |
---|---|---|
epsilon | ::mlir::FloatAttr | سمة تعويم 32 بت |
feature_index | ::mlir::IntegerAttr | سمة عدد صحيح بدون إشارة 64 بت |
المعاملات:
المعامل | وصف |
---|---|
operand | الموتر المصنف من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو قيم النوع float 16 بت أو تعويم 32 بت أو قيم نوع bfloat16 64 بت |
scale | موتر 1D من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو قيم النوع float 16 بت أو تعويم 32 بت أو تعويم 64 بت أو bfloat16 |
mean | موتر 1D من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو قيم النوع float 16 بت أو تعويم 32 بت أو تعويم 64 بت أو bfloat16 |
variance | موتر 1D من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو قيم النوع float 16 بت أو تعويم 32 بت أو تعويم 64 بت أو bfloat16 |
grad_output | الموتر المصنف من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو قيم النوع float 16 بت أو تعويم 32 بت أو قيم نوع bfloat16 64 بت |
نتائج:
نتيجة | وصف |
---|---|
grad_operand | الموتر المصنف من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو قيم النوع float 16 بت أو تعويم 32 بت أو قيم نوع bfloat16 64 بت |
grad_scale | موتر 1D من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو قيم النوع float 16 بت أو تعويم 32 بت أو قيم النوع bfloat16 64 بت |
grad_offset | موتر 1D من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو قيم النوع float 16 بت أو تعويم 32 بت أو قيم النوع bfloat16 64 بت |
mhlo.batch_norm_inference
(mhlo::BatchNormInferenceOp)
عملية BatchNormInference
يقوم بتطبيع موتر operand
عبر جميع الأبعاد باستثناء بُعد feature_index
وينتج موتر result
.
راجع: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#batch_norm_inference
مثال:
%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>
السمات: AlwaysSpeculatableImplTrait
، InferTensorType
الواجهات: ConditionallySpeculatable
، InferShapedTypeOpInterface
، InferTypeOpInterface
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
صفات:
يصف | نوع ملير | وصف |
---|---|---|
epsilon | ::mlir::FloatAttr | سمة تعويم 32 بت |
feature_index | ::mlir::IntegerAttr | سمة عدد صحيح بدون إشارة 64 بت |
المعاملات:
المعامل | وصف |
---|---|
operand | الموتر المصنف من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو قيم النوع float 16 بت أو تعويم 32 بت أو قيم نوع bfloat16 64 بت |
scale | موتر 1D من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو قيم النوع float 16 بت أو تعويم 32 بت أو تعويم 64 بت أو bfloat16 |
offset | موتر 1D من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو قيم النوع float 16 بت أو تعويم 32 بت أو تعويم 64 بت أو bfloat16 |
mean | موتر 1D من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو قيم النوع float 16 بت أو تعويم 32 بت أو تعويم 64 بت أو bfloat16 |
variance | موتر 1D من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو قيم النوع float 16 بت أو تعويم 32 بت أو تعويم 64 بت أو bfloat16 |
نتائج:
نتيجة | وصف |
---|---|
result | الموتر المصنف من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو قيم النوع float 16 بت أو تعويم 32 بت أو قيم نوع bfloat16 64 بت |
mhlo.batch_norm_training
(mhlo::BatchNormTrainingOp)
عملية BatchNormTraining
يحسب المتوسط والتباين عبر الأبعاد الدفعية والمكانية ويقوم بتطبيع موتر operand
، لكل ميزة في بُعد feature_index
وينتج output
وموترات batch_mean
و batch_var
.
راجع: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#batch_norm_training
مثال:
%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>)
السمات: AlwaysSpeculatableImplTrait
، InferTensorType
الواجهات: ConditionallySpeculatable
، InferShapedTypeOpInterface
، InferTypeOpInterface
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
صفات:
يصف | نوع ملير | وصف |
---|---|---|
epsilon | ::mlir::FloatAttr | سمة تعويم 32 بت |
feature_index | ::mlir::IntegerAttr | سمة عدد صحيح بدون إشارة 64 بت |
المعاملات:
المعامل | وصف |
---|---|
operand | الموتر المصنف من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو قيم النوع float 16 بت أو تعويم 32 بت أو قيم نوع bfloat16 64 بت |
scale | موتر 1D من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو قيم النوع float 16 بت أو تعويم 32 بت أو تعويم 64 بت أو bfloat16 |
offset | موتر 1D من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو قيم النوع float 16 بت أو تعويم 32 بت أو تعويم 64 بت أو bfloat16 |
نتائج:
نتيجة | وصف |
---|---|
output | الموتر المصنف من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو قيم النوع float 16 بت أو تعويم 32 بت أو قيم نوع bfloat16 64 بت |
batch_mean | موتر 1D من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو قيم النوع float 16 بت أو تعويم 32 بت أو قيم النوع bfloat16 64 بت |
batch_var | موتر 1D من النوع f8E4M3B11FNUZ أو النوع f8E4M3FN أو النوع f8E4M3FNUZ أو النوع f8E5M2 أو النوع f8E5M2FNUZ أو قيم النوع float 16 بت أو تعويم 32 بت أو تعويم 64 بت أو bfloat16 |
mhlo.bitcast
(mhlo::BitcastOp)
عملية البث الثنائي
بناء الجملة:
operation ::= `mhlo.bitcast` operands attr-dict `:` functional-type(operands, results)
هذه العملية خاصة بمترجم XLA، لذا فهي لا تحتوي على مواصفات بعد.
بشكل غير رسمي، تعمل هذه العملية على تغيير شكل المدخلات بالطريقة التي لا يتغير بها الترتيب المادي للعناصر.
تحتاج هذه العملية إلى معلومات التخطيط لفهم "الترتيب المادي للعناصر"، ودعم التخطيط في MHLO قيد التنفيذ حاليًا.
مثال:
%0 = mhlo.bitcast %arg0 : (tensor<3x4xf32>) -> tensor<3x4x1xf32>
السمات: AlwaysSpeculatableImplTrait
الواجهات: ConditionallySpeculatable
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
المعاملات:
المعامل | وصف |
---|---|
operand | موتر مصنف من النوع f8E4M3B11FNUZ أو نوع f8E4M3FN أو نوع f8E4M3FNUZ أو نوع f8E5M2 أو نوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو pred (AKA منطقي أو عدد صحيح 1 بت) أو 4/8 /16/32/64 بت عدد صحيح بدون إشارة أو 4/8/16/32/64 بت عدد صحيح بدون إشارة أو نوع معقد مع عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو 4/8/16/32 بت موحدة كمية عدد صحيح موقّع أو 4/8/16/32 بت عدد صحيح مكمّم غير موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور قيم عددية غير موقّعة |
نتائج:
نتيجة | وصف |
---|---|
«غير مسمى» | موتر مصنف من النوع f8E4M3B11FNUZ أو نوع f8E4M3FN أو نوع f8E4M3FNUZ أو نوع f8E5M2 أو نوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو pred (AKA منطقي أو عدد صحيح 1 بت) أو 4/8 /16/32/64 بت عدد صحيح بدون إشارة أو 4/8/16/32/64 بت عدد صحيح بدون إشارة أو نوع معقد مع عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو 4/8/16/32 بت موحدة كمية عدد صحيح موقّع أو 4/8/16/32 بت عدد صحيح مكمّم غير موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور قيم عددية غير موقّعة |
mhlo.bitcast_convert
(mhlo::BitcastConvertOp)
عملية BitcastConvert
بناء الجملة:
operation ::= `mhlo.bitcast_convert` operands attr-dict `:` functional-type(operands, results)
ينفذ عملية البث البتي على operand
وينتج موتر result
حيث تتم إعادة تفسير بتات موتر operand
بأكمله باستخدام نوع موتر result
.
راجع: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#bitcast_convert
مثال:
%result = mhlo.bitcast_convert %operand : (tensor<2xf32>) -> tensor<2x4xi8>
السمات: AlwaysSpeculatableImplTrait
الواجهات: ConditionallySpeculatable
، InferShapedTypeOpInterface
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
المعاملات:
المعامل | وصف |
---|---|
operand | موتر مصنف من النوع f8E4M3B11FNUZ أو نوع f8E4M3FN أو نوع f8E4M3FNUZ أو نوع f8E5M2 أو نوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو pred (AKA منطقي أو عدد صحيح 1 بت) أو 4/8 /16/32/64 بت عدد صحيح بدون إشارة أو 4/8/16/32/64 بت عدد صحيح بدون إشارة أو نوع معقد مع عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو 4/8/16/32 بت موحدة كمية عدد صحيح موقّع أو 4/8/16/32 بت عدد صحيح مكمّم غير موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور عدد صحيح موقّع أو 4/8/16/32 بت مكمّم موحّد لكل محور قيم عددية غير موقّعة |
نتائج:
نتيجة | وصف |
---|---|
«غير مسمى» | موتر مصنف من النوع f8E4M3B11FNUZ أو نوع f8E4M3FN أو نوع f8E4M3FNUZ أو نوع f8E5M2 أو نوع f8E5M2FNUZ أو تعويم 16 بت أو تعويم 32 بت أو تعويم 64 بت أو نوع bfloat16 أو pred (AKA منطقي أو عدد صحيح 1 بت) أو 4/8 /16/32/64 بت عدد صحيح بدون إشارة أو 4/8/16/32/64 بت عدد صحيح بدون إشارة أو نوع معقد مع عناصر عائمة 32 بت أو عناصر عائمة 64 بت أو 4/8/16/32 بت موحدة كمية عدد صحيح موقّع أو 4/8/16/16/32 بت عدد صحيح غير موقّع أو 4/8/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/16/32 بت كمية موحدة لكل محور غير موقّع |
mhlo.broadcast
(Mhlo :: Broadcastop)
عملية البث
هذه العملية في طريقها للخروج من مستقر ، لذلك لا يتم تضمينها في المواصفات: https://github.com/openxla/stablehlo/issues/3
بشكل غير رسمي ، هذه العملية تفعل نفس الشيء مثل إذاعة XLA: https://www.tensorflow.org/xla/operation_semantics#broadcast
مثال:
%result = mhlo.broadcast %operand, sizes = [1, 2] : (tensor<3xi32>) -> tensor<1x2x3xi32>
السمات: AlwaysSpeculatableImplTrait
، InferTensorType
، SameOperandsAndResultElementType
واجهات: ConditionallySpeculatable
، InferShapedTypeOpInterface
، InferTypeOpInterface
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
صفات:
يصف | نوع mlir | وصف |
---|---|---|
broadcast_sizes | :: mlir :: denseintelementattr | سمة 64 بت من عناصر عدد صحيح غير مألوف |
المعاملات:
المعامل | وصف |
---|---|
operand | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz /16/32/64 بت عدد صحيح غير محدد أو 4/8/16/32/64 بت غير موقّع أو نوع معقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت موحدة موحدة عدد صحيح موقّع أو 4/8/16/16/32 بت عدد صحيح غير موقّع أو 4/8/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/16/32 بت كمية موحدة لكل محور غير موقّع |
نتائج:
نتيجة | وصف |
---|---|
«لم يكشف عن اسمه» | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz /16/32/64 بت عدد صحيح غير محدد أو 4/8/16/32/64 بت غير موقّع أو نوع معقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت موحدة موحدة عدد صحيح موقّع أو 4/8/16/16/32 بت عدد صحيح غير موقّع أو 4/8/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/16/32 بت كمية موحدة لكل محور غير موقّع |
mhlo.broadcast_in_dim
(mhlo :: broadcastindimop)
عملية البث
يوسع أبعاد و/أو رتبة موتر إدخال عن طريق تكرار البيانات في الموتر operand
وينتج عن موتر result
.
انظر: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#broadcast_in_dim
مثال:
%result = mhlo.broadcast_in_dim %operand, dims = [2, 1] : (tensor<1x3xi32>) -> tensor<2x3x2xi32>
السمات: AlwaysSpeculatableImplTrait
، HLO_CompatibleOperandsAndResultElementType
واجهات: ConditionallySpeculatable
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
صفات:
يصف | نوع mlir | وصف |
---|---|---|
broadcast_dimensions | :: mlir :: denseintelementattr | سمة 64 بت من عناصر عدد صحيح غير مألوف |
المعاملات:
المعامل | وصف |
---|---|
operand | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz /16/32/64 بت عدد صحيح غير محدد أو 4/8/16/32/64 بت غير موقّع أو نوع معقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت موحدة موحدة عدد صحيح موقّع أو 4/8/16/16/32 بت عدد صحيح غير موقّع أو 4/8/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/16/32 بت كمية موحدة لكل محور غير موقّع |
نتائج:
نتيجة | وصف |
---|---|
«لم يكشف عن اسمه» | نوع موتر من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz 8/16/16/32/64 بت عدد صحيح غير محدد أو 4/8/16/32/64 بت عدد صحيح غير موقّع أو نوع معقد مع عناصر عائمة 32 بت أو 64 بت أو 4/8/16/32 بت موحدة عدد صحيح موقّع كميًا أو عدد صحيح موحد 4/8/16/32 بت كميات غير موقعة أو 4/8/6/16/32 بت كمية موحدة لكل محور صحيح أو 4/8/16/32 بت كمية موحدة لكل محور غير موقّع |
mhlo.case
(MHLO :: Caseop)
عملية القضية
ينتج الإخراج من تنفيذ function
واحدة بالضبط من branches
اعتمادًا على قيمة index
.
انظر: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#case
مثال:
%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>)
السمات: RecursiveMemoryEffects
، SingleBlockImplicitTerminator<ReturnOp>
، SingleBlock
واجهات: InferTypeOpInterface
المعاملات:
المعامل | وصف |
---|---|
index | توتر من قيم عدد صحيح 32 بت غير مألوف |
نتائج:
نتيجة | وصف |
---|---|
«لم يكشف عن اسمه» | variadic من tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5mm2fnuz /8/6/16/32/64 بتات غير مميزة أو 4/8/16/32/64 بت النوع غير موقّع أو النوع المعقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت عدد صحيح موقّع موحدًا أو 4/8/16/16/32 بت قيم عدد صحيح غير موقّع موحدة أو موحد من الموحد 4/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/16/32 بت كمية موحدة لكل محور قيم عدد صحيح غير موقعة أو رمز الرمز |
mhlo.cbrt
(mhlo :: cbrtop)
عملية CBRT
بناء الجملة:
operation ::= `mhlo.cbrt` $operand attr-dict
`:` custom<SameOperandsAndResultType>(type($operand), type($result))
يؤدي تشغيل الجذر المكعب العناصر على الموتر operand
وينتج عن توتر result
.
انظر: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#cbrt
مثال:
%result = mhlo.cbrt %operand : tensor<4xf32>
السمات: AlwaysSpeculatableImplTrait
، CompatibleOperandsAndResultType
، Elementwise
، SameOperandsAndResultShape
واجهات: ConditionallySpeculatable
، InferShapedTypeOpInterface
، InferTypeOpInterface
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
المعاملات:
المعامل | وصف |
---|---|
operand | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz أو نوع التعويم أو العوامات 32 بت أو 64 بت 4/8/16/16/32 بت عدد صحيح موقّع كميًا أو 4/8/16/16/32 بت قيم عدد صحيح غير موقّع موحدة |
نتائج:
نتيجة | وصف |
---|---|
result | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz أو نوع التعويم أو العوامات 32 بت أو 64 بت 4/8/16/16/32 بت عدد صحيح موقّع كميًا أو 4/8/16/16/32 بت قيم عدد صحيح غير موقّع موحدة |
mhlo.ceil
(mhlo :: ceilop)
عملية السقف
بناء الجملة:
operation ::= `mhlo.ceil` $operand attr-dict
`:` custom<SameOperandsAndResultType>(type($operand), type($result))
يؤدي سقف العنصر من الموتر operand
وينتج عن توتر result
.
انظر: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#ceil
مثال:
%result = mhlo.ceil %operand : tensor<5xf32>
السمات: AlwaysSpeculatableImplTrait
، CompatibleOperandsAndResultType
، Elementwise
، SameOperandsAndResultShape
واجهات: ConditionallySpeculatable
، InferShapedTypeOpInterface
، InferTypeOpInterface
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
المعاملات:
المعامل | وصف |
---|---|
operand | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz أو عائم 16 بت أو تعويم 32 بت أو 64 بت عطف أو bfloat16 |
نتائج:
نتيجة | وصف |
---|---|
result | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz أو عائم 16 بت أو تعويم 32 بت أو 64 بت عطف أو bfloat16 |
mhlo.cholesky
(Mhlo :: choleskyop)
عملية تشولسكي
يحسب تحلل Cholesky لمجموعة من المصفوفات.
انظر: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#cholesky
مثال:
%result = mhlo.cholesky %a, lower = true : tensor<3x3xf32>
السمات: AlwaysSpeculatableImplTrait
، InferTensorType
، SameOperandsAndResultElementType
واجهات: ConditionallySpeculatable
، InferShapedTypeOpInterface
، InferTypeOpInterface
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
صفات:
يصف | نوع mlir | وصف |
---|---|---|
lower | :: mlir :: boolattr | سمة Bool |
المعاملات:
المعامل | وصف |
---|---|
a | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz أو نوع التعويم أو العوفى 32 بت أو 64 بت. |
نتائج:
نتيجة | وصف |
---|---|
«لم يكشف عن اسمه» | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz أو نوع التعويم أو العوفى 32 بت أو 64 بت. |
mhlo.clamp
(mhlo :: clampop)
عملية المشبك
بناء الجملة:
operation ::= `mhlo.clamp` $min `,` $operand `,` $max attr-dict
`:` custom<SameOperandsAndResultType>(type($min), type($operand), type($max), type($result))
يشمل كل عنصر من عناصر الموتر operand
بين الحد الأدنى والحد الأقصى لقيمة وينتج عن موتر result
.
انظر: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#clamp
مثال:
%result = mhlo.clamp %min, %operand, %max : tensor<3xi32>
السمات: AlwaysSpeculatableImplTrait
، HLO_BroadcastingElementwise
، InferTensorType
، SameOperandsAndResultElementType
واجهات: ConditionallySpeculatable
، InferShapedTypeOpInterface
، InferTypeOpInterface
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
المعاملات:
المعامل | وصف |
---|---|
min | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz /16/32/64 بت عدد صحيح غير محدد أو 4/8/16/32/64 بت غير موقّع أو نوع معقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت موحدة موحدة عدد صحيح موقّع أو 4/8/16/16/32 بت عدد صحيح غير موقّع أو 4/8/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/16/32 بت كمية موحدة لكل محور غير موقّع |
operand | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz /16/32/64 بت عدد صحيح غير محدد أو 4/8/16/32/64 بت غير موقّع أو نوع معقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت موحدة موحدة عدد صحيح موقّع أو 4/8/16/16/32 بت عدد صحيح غير موقّع أو 4/8/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/16/32 بت كمية موحدة لكل محور غير موقّع |
max | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz /16/32/64 بت عدد صحيح غير محدد أو 4/8/16/32/64 بت غير موقّع أو نوع معقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت موحدة موحدة عدد صحيح موقّع أو 4/8/16/16/32 بت عدد صحيح غير موقّع أو 4/8/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/16/32 بت كمية موحدة لكل محور غير موقّع |
نتائج:
نتيجة | وصف |
---|---|
result | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz /16/32/64 بت عدد صحيح غير محدد أو 4/8/16/32/64 بت غير موقّع أو نوع معقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت موحدة موحدة عدد صحيح موقّع أو 4/8/16/16/32 بت عدد صحيح غير موقّع أو 4/8/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/16/32 بت كمية موحدة لكل محور غير موقّع |
mhlo.collective_broadcast
(mhlo :: collectivebroadcastop)
عملية البث الجماعي
ضمن كل مجموعة عملية في شبكة العملية ، أرسل قيمة operand
المعامل من عملية المصدر إلى العمليات المستهدفة وإنتاج موتر result
.
انظر: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#collective_broadcast
مثال:
%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>
السمات: CompatibleOperandsAndResultType
واجهات InferTypeOpInterface
InferShapedTypeOpInterface
صفات:
يصف | نوع mlir | وصف |
---|---|---|
replica_groups | :: mlir :: denseintelementattr | سمة 64 بت من عناصر عدد صحيح غير مألوف |
channel_handle | :: mlir :: mhlo :: channelhandleattr | اثنين من أعداد صحيحة 64 بت "مقبض" و "نوع" |
المعاملات:
المعامل | وصف |
---|---|
operand | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz /16/32/64 بت عدد صحيح غير محدد أو 4/8/16/32/64 بت غير موقّع أو نوع معقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت موحدة موحدة عدد صحيح موقّع أو 4/8/16/16/32 بت عدد صحيح غير موقّع أو 4/8/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/16/32 بت كمية موحدة لكل محور غير موقّع |
نتائج:
نتيجة | وصف |
---|---|
«لم يكشف عن اسمه» | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz /16/32/64 بت عدد صحيح غير محدد أو 4/8/16/32/64 بت غير موقّع أو نوع معقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت موحدة موحدة عدد صحيح موقّع أو 4/8/16/16/32 بت عدد صحيح غير موقّع أو 4/8/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/16/32 بت كمية موحدة لكل محور غير موقّع |
mhlo.collective_permute
(mhlo :: collectivepermuteop)
collectivepermute العملية
ضمن كل مجموعة عملية في شبكة العملية ، يرسل قيمة operand
المعامل من عملية المصدر إلى العملية المستهدفة وينتج عن موتر result
.
انظر: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#collective_permute
مثال:
%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>
السمات: AlwaysSpeculatableImplTrait
، CompatibleOperandsAndResultType
واجهات: ConditionallySpeculatable
، InferShapedTypeOpInterface
، InferTypeOpInterface
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
صفات:
يصف | نوع mlir | وصف |
---|---|---|
source_target_pairs | :: mlir :: denseintelementattr | سمة 64 بت من عناصر عدد صحيح غير مألوف |
channel_handle | :: mlir :: mhlo :: channelhandleattr | اثنين من أعداد صحيحة 64 بت "مقبض" و "نوع" |
المعاملات:
المعامل | وصف |
---|---|
operand | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz /16/32/64 بت عدد صحيح غير محدد أو 4/8/16/32/64 بت غير موقّع أو نوع معقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت موحدة موحدة عدد صحيح موقّع أو 4/8/16/16/32 بت عدد صحيح غير موقّع أو 4/8/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/16/32 بت كمية موحدة لكل محور غير موقّع |
نتائج:
نتيجة | وصف |
---|---|
«لم يكشف عن اسمه» | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz /16/32/64 بت عدد صحيح غير محدد أو 4/8/16/32/64 بت غير موقّع أو نوع معقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت موحدة موحدة عدد صحيح موقّع أو 4/8/16/16/32 بت عدد صحيح غير موقّع أو 4/8/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/16/32 بت كمية موحدة لكل محور غير موقّع |
mhlo.compare
(mhlo :: compareop)
قارن العملية
بناء الجملة:
operation ::= `mhlo.compare` $comparison_direction `,` $lhs `,` $rhs (`,` $compare_type^)?
attr-dict `:` functional-type(operands, results)
يؤدي مقارنة العناصر من Tensors lhs
و rhs
وفقًا لـ comparison_direction
و compare_type
، وينتج عن موتر result
.
انظر: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#compare
مثال:
%result = mhlo.compare LT, %lhs, %rhs, FLOAT : (tensor<2xf32>, tensor<2xf32>) -> tensor<2xi1>
السمات: AlwaysSpeculatableImplTrait
، Elementwise
، InferTensorType
، SameOperandsAndResultShape
، SameOperandsElementType
واجهات: ConditionallySpeculatable
، InferShapedTypeOpInterface
، InferTypeOpInterface
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
صفات:
يصف | نوع mlir | وصف |
---|---|---|
comparison_direction | :: mlir :: mhlo :: المقارن | أي عملية مقارنة لأداء. |
compare_type | :: mlir :: mhlo :: comparisontypeattr | أي نوع مقارنة للاستخدام. |
المعاملات:
المعامل | وصف |
---|---|
lhs | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz /16/32/64 بت عدد صحيح غير محدد أو 4/8/16/32/64 بت غير موقّع أو نوع معقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت موحدة موحدة عدد صحيح موقّع أو 4/8/16/16/32 بت عدد صحيح غير موقّع أو 4/8/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/16/32 بت كمية موحدة لكل محور غير موقّع |
rhs | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz /16/32/64 بت عدد صحيح غير محدد أو 4/8/16/32/64 بت غير موقّع أو نوع معقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت موحدة موحدة عدد صحيح موقّع أو 4/8/16/16/32 بت عدد صحيح غير موقّع أو 4/8/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/16/32 بت كمية موحدة لكل محور غير موقّع |
نتائج:
نتيجة | وصف |
---|---|
«لم يكشف عن اسمه» | موتر مرتبة من قيم pred (ويعرف أيضا باسم المنطقية أو عدد صحيح 1 بت) |
mhlo.complex
(mhlo :: complexop)
عملية معقدة
بناء الجملة:
operation ::= `mhlo.complex` operands attr-dict
`:` custom<ComplexOpType>(type($lhs), type($rhs), type($result))
يؤدي تحويل العناصر إلى قيمة معقدة من زوج من القيم الحقيقية والخيالية ، lhs
و rhs
، وينتج عن موتر result
.
انظر: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#complex
مثال:
%result = mhlo.complex %lhs, %rhs : tensor<2xcomplex<f32>>
السمات: AlwaysSpeculatableImplTrait
، Elementwise
، SameOperandsAndResultShape
، SameOperandsElementType
واجهات: ConditionallySpeculatable
، InferShapedTypeOpInterface
، InferTypeOpInterface
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
المعاملات:
المعامل | وصف |
---|---|
lhs | الموتر المرتبة من قيم تعويم 32 بت أو 64 بت قيم تعويم |
rhs | الموتر المرتبة من قيم تعويم 32 بت أو 64 بت قيم تعويم |
نتائج:
نتيجة | وصف |
---|---|
result | موتر مرتبة من النوع المعقد مع قيم عناصر عائمة 32 بت أو 64 بت قيم |
mhlo.composite
(mhlo :: compositeop)
تشغيل مركب
بناء الجملة:
operation ::= `mhlo.composite` $name $inputs attr-dict `:` functional-type(operands, results)
يتغلف عملية مكونة (مكونة) من عمليات مستقرة أخرى ، وأخذ inputs
و composite_attributes
وإنتاج results
. يتم تنفيذ دلالات OP بواسطة سمة decomposition
. يمكن استبدال OP composite
بالتحلل دون تغيير دلالات البرنامج. في الحالات التي لا يوفر فيها تحديد التحلل نفس دلالات OP ، تفضل استخدام custom_call
.
يتم استخدام حقل version
(الإعدادات الافتراضية إلى 0
) للدلالة على موعد تغيير دلالات المركب.
انظر: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#composite
مثال:
%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>
واجهات: SymbolUserOpInterface
صفات:
يصف | نوع mlir | وصف |
---|---|---|
name | :: mlir :: stringattr | سمة السلسلة |
composite_attributes | :: mlir :: القاموس | قاموس قيم السمة المسماة |
decomposition | :: mlir :: flatsymbolrefattr | سمة مرجع الرمز المسطح |
version | :: mlir :: integerattr | سمة عدد صحيح 32 بت غير مألوف |
المعاملات:
المعامل | وصف |
---|---|
inputs | variadic من tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5mm2fnuz /8/6/16/32/64 بتات غير مميزة أو 4/8/16/32/64 بت النوع غير موقّع أو النوع المعقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت عدد صحيح موقّع موحد أو 4/8/16/16/32 بت من عدد صحيح غير موقّع أو 32 بت أو 4/8/16/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/6/16/ 32 بت كمية موحدة لكل محور غير مُقدس القيم أو الرمز المميز أو التداخل المتداخل مع أي مجموعة من الموتر المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz أو عدد صحيح 1 بت) أو عدد صحيح 4/8/16/32/64 بت غير محدد أو 4/8/16/32/64 بت نوع غير موقّع أو نوع معقد مع تعويم 32 بت أو 64 بت العائم أو 4 /8/16/16/32 بت عدد صحيح موقّع كمياً أو 4/8/16/16/32 بت قيم عدد صحيح غير موقّع أو موحد أو موحد من 4/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8 /16/32 بت موحدة كمية لكل قيم عدد صحيح غير موقعة أو قيم رمزية |
نتائج:
نتيجة | وصف |
---|---|
«لم يكشف عن اسمه» | variadic من tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5mm2fnuz /8/6/16/32/64 بتات غير مميزة أو 4/8/16/32/64 بت النوع غير موقّع أو النوع المعقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت عدد صحيح موقّع موحد أو 4/8/16/16/32 بت من عدد صحيح غير موقّع أو 32 بت أو 4/8/16/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/6/16/ 32 بت كمية موحدة لكل محور غير مُقدس القيم أو الرمز المميز أو التداخل المتداخل مع أي مجموعة من الموتر المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz أو عدد صحيح 1 بت) أو عدد صحيح 4/8/16/32/64 بت غير محدد أو 4/8/16/32/64 بت نوع غير موقّع أو نوع معقد مع تعويم 32 بت أو 64 بت العائم أو 4 /8/16/16/32 بت عدد صحيح موقّع كمياً أو 4/8/16/16/32 بت قيم عدد صحيح غير موقّع أو موحد أو موحد من 4/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8 /16/32 بت موحدة كمية لكل قيم عدد صحيح غير موقعة أو قيم رمزية |
mhlo.compute_reshape_shape
(mhlo :: computereshapeshapeop)
تشغيل ComputereshapeShape
بناء الجملة:
operation ::= `mhlo.compute_reshape_shape` operands attr-dict `:` functional-type(operands, results)
هذه العملية هي عمل مستمر ، لذلك لم يتم تضمينه بعد في المواصفات: https://github.com/openxla/stablehlo/issues/8
بشكل غير رسمي ، تحسب هذه العملية dynamic_shape
لـ DynamicReshapeop من عدد العناصر num_elements
على سبيل المثال ، بالنسبة لـ num_elements = 12
و dynamic_shape = [2, -1]
، ستكون result
[2, 6]
. إذا كانت المعاملات غير صالحة (على سبيل المثال ، إذا لم تقسم الأبعاد بالتساوي عدد العناصر ، أو إذا كانت هناك قيم متعددة -1 في أبعاد) ، فهذا يؤدي إلى سلوك غير محدد.
مثال:
%result = mhlo.compute_reshape_shape %num_elements, %dynamic_shape
: (index, tensor<2xi32>) -> tensor<2xi32>
السمات: AlwaysSpeculatableImplTrait
واجهات: ConditionallySpeculatable
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
المعاملات:
المعامل | وصف |
---|---|
num_elements | فِهرِس |
dynamic_shape | 1D Tensor من قيم عدد صحيح أو فهرس |
نتائج:
نتيجة | وصف |
---|---|
result | 1D Tensor من قيم عدد صحيح أو فهرس |
mhlo.concatenate
(mhlo :: concatenateop)
عملية التسلسل
يسلط عدد المتنوع من الموترات في inputs
على طول dimension
في نفس الترتيب مثل الوسائط المحددة وينتج عن موتر result
.
انظر: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#concatenate
مثال:
%result = mhlo.concatenate %input0, %input1, dim = 0 : (tensor<3x2xi64>, tensor<1x2xi64>) -> tensor<4x2xi64>
السمات: AlwaysSpeculatableImplTrait
، SameOperandsAndResultElementType
واجهات: ConditionallySpeculatable
، InferShapedTypeOpInterface
، InferTypeOpInterface
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
صفات:
يصف | نوع mlir | وصف |
---|---|---|
dimension | :: mlir :: integerattr | سمة عدد صحيح 64 بت غير مألوف |
المعاملات:
المعامل | وصف |
---|---|
val | variadic من tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5mm2fnuz /8/6/16/32/64 بتات غير مميزة أو 4/8/16/32/64 بت النوع غير موقّع أو النوع المعقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت عدد صحيح موقّع موحد أو 4/8/16/16/32 بت من عدد صحيح غير موقّع أو 32 بت أو 4/8/16/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/6/16/ 32 بت كمية موحدة لكل محور غير مُقدس قيم |
نتائج:
نتيجة | وصف |
---|---|
«لم يكشف عن اسمه» | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz /16/32/64 بت عدد صحيح غير محدد أو 4/8/16/32/64 بت غير موقّع أو نوع معقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت موحدة موحدة عدد صحيح موقّع أو 4/8/16/16/32 بت عدد صحيح غير موقّع أو 4/8/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/16/32 بت كمية موحدة لكل محور غير موقّع |
mhlo.constant
(mhlo :: constantop)
عملية مستمرة
ينتج موتر output
من value
ثابتة.
انظر: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#constant
مثال:
%output = mhlo.constant dense<[[0.0, 1.0], [2.0, 3.0]]> : tensor<2x2xf32>
السمات: AlwaysSpeculatableImplTrait
، ConstantLike
واجهات: ConditionallySpeculatable
، InferTypeOpInterface
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
صفات:
يصف | نوع mlir | وصف |
---|---|---|
value | :: mlir :: elementsattr | سمة ناقلات/موتر ثابتة |
نتائج:
نتيجة | وصف |
---|---|
output | نوع موتر من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz 8/16/16/32/64 بت عدد صحيح غير محدد أو 4/8/16/32/64 بت عدد صحيح غير موقّع أو نوع معقد مع عناصر عائمة 32 بت أو 64 بت أو 4/8/16/32 بت موحدة عدد صحيح موقّع كميًا أو عدد صحيح موحد 4/8/16/32 بت كميات غير موقعة أو 4/8/6/16/32 بت كمية موحدة لكل محور صحيح أو 4/8/16/32 بت كمية موحدة لكل محور غير موقّع |
mhlo.convert
(mhlo :: convertop)
تحويل العملية
بناء الجملة:
operation ::= `mhlo.convert` $operand attr-dict
`:` custom<SameOperandsAndResultType>(type($operand), type($result))
يقوم بتحويل عناصر من نوع عنصر إلى آخر على موتر operand
وينتج عن موتر result
.
انظر: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#convert
مثال:
%result = mhlo.convert %operand : (tensor<3xi32>) -> tensor<3xcomplex<f32>>
السمات: AlwaysSpeculatableImplTrait
، Elementwise
، SameOperandsAndResultShape
واجهات: ConditionallySpeculatable
، InferShapedTypeOpInterface
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
المعاملات:
المعامل | وصف |
---|---|
operand | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz /16/32/64 بت عدد صحيح غير محدد أو 4/8/16/32/64 بت غير موقّع أو نوع معقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت موحدة موحدة عدد صحيح موقّع أو 4/8/16/16/32 بت عدد صحيح غير موقّع أو 4/8/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/16/32 بت كمية موحدة لكل محور غير موقّع |
نتائج:
نتيجة | وصف |
---|---|
result | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz /16/32/64 بت عدد صحيح غير محدد أو 4/8/16/32/64 بت غير موقّع أو نوع معقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت موحدة موحدة عدد صحيح موقّع أو 4/8/16/16/32 بت عدد صحيح غير موقّع أو 4/8/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/16/32 بت كمية موحدة لكل محور غير موقّع |
mhlo.convolution
(mhlo :: connertolutionop)
عملية الالتفاف
بناء الجملة:
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)
يحسب منتجات DOT بين Windows of lhs
وشرائح rhs
وتنتج result
.
انظر: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#convolution
مثال:
%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>
السمات: AlwaysSpeculatableImplTrait
واجهات: ConditionallySpeculatable
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
صفات:
يصف | نوع mlir | وصف |
---|---|---|
window_strides | :: mlir :: denseintelementattr | سمة 64 بت من عناصر عدد صحيح غير مألوف |
padding | :: mlir :: denseintelementattr | سمة 64 بت من عناصر عدد صحيح غير مألوف |
lhs_dilation | :: mlir :: denseintelementattr | سمة 64 بت من عناصر عدد صحيح غير مألوف |
rhs_dilation | :: mlir :: denseintelementattr | سمة 64 بت من عناصر عدد صحيح غير مألوف |
window_reversal | :: mlir :: DenseElementAttr | سمة المتجه/الموتر الثابت |
dimension_numbers | :: mlir :: mhlo :: convdimensionnumbersattr | هيكل معلومات البعد لـ Conv Op |
feature_group_count | :: mlir :: integerattr | سمة عدد صحيح 64 بت غير مألوف |
batch_group_count | :: mlir :: integerattr | سمة عدد صحيح 64 بت غير مألوف |
precision_config | :: mlir :: arrayattr | سمة التكوين الدقيق |
المعاملات:
المعامل | وصف |
---|---|
lhs | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz /16/32/64 بت عدد صحيح غير محدد أو 4/8/16/32/64 بت غير موقّع أو نوع معقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت موحدة موحدة عدد صحيح موقّع أو 4/8/16/16/32 بت عدد صحيح غير موقّع أو 4/8/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/16/32 بت كمية موحدة لكل محور غير موقّع |
rhs | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz /16/32/64 بت عدد صحيح غير محدد أو 4/8/16/32/64 بت غير موقّع أو نوع معقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت موحدة موحدة عدد صحيح موقّع أو 4/8/16/16/32 بت عدد صحيح غير موقّع أو 4/8/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/16/32 بت كمية موحدة لكل محور غير موقّع |
نتائج:
نتيجة | وصف |
---|---|
«لم يكشف عن اسمه» | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz /16/32/64 بت عدد صحيح غير محدد أو 4/8/16/32/64 بت غير موقّع أو نوع معقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت موحدة موحدة عدد صحيح موقّع أو 4/8/16/16/32 بت عدد صحيح غير موقّع أو 4/8/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/16/32 بت كمية موحدة لكل محور غير موقّع |
mhlo.copy
(mhlo :: copyop)
نسخ العملية
بناء الجملة:
operation ::= `mhlo.copy` operands attr-dict
`:` custom<SameOperandsAndResultType>(type($operand), type($result))
هذه العملية خاصة لمرجم XLA ، لذلك ليس لديها مواصفات بعد.
بشكل غير رسمي ، هذه العملية نسخة من operand
. اعتمادًا على البيانات الوصفية المرتبطة بالعملية ، يمكن أن تتصرف بشكل مختلف تمامًا عن عدم وجود OP.
مثال:
%0 = mhlo.copy %arg0 : tensor<f32>
السمات: AlwaysSpeculatableImplTrait
، CompatibleOperandsAndResultType
، Elementwise
واجهات: ConditionallySpeculatable
، InferShapedTypeOpInterface
، InferTypeOpInterface
، NoMemoryEffect (MemoryEffectOpInterface)
التأثيرات: MemoryEffects::Effect{}
صفات:
يصف | نوع mlir | وصف |
---|---|---|
cross_program_prefetch_index | :: mlir :: integerattr | سمة عدد صحيح 32 بت غير مألوف |
المعاملات:
المعامل | وصف |
---|---|
operand | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz /16/32/64 بت عدد صحيح غير محدد أو 4/8/16/32/64 بت غير موقّع أو نوع معقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت موحدة موحدة عدد صحيح موقّع أو 4/8/16/16/32 بت عدد صحيح غير موقّع أو 4/8/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/16/32 بت كمية موحدة لكل محور غير موقّع. رمز الرمز المميز أو المتداخل مع أي مزيج من الموتر المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz أو 16 بت عطف أو 32-bit -عدد صحيح-أو 4/8/16/32/64 بتات غير محددة أو عدد صحيح 4/8/16/32/64 بت غير موقّع أو نوع معقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8 /16/32 بت عدد صحيح موقّع موحد أو 4/8/6/16/32 بت قيم عدد صحيح غير موقّع أو موحد أو موحد من 4/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/16 /32 بت موحدة كمية لكل محور قيم عدد صحيح غير موقعة أو قيم رمزية |
نتائج:
نتيجة | وصف |
---|---|
result | Tensor المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz /16/32/64 بت عدد صحيح غير محدد أو 4/8/16/32/64 بت غير موقّع أو نوع معقد مع عناصر تعويم 32 بت أو 64 بت أو 4/8/16/32 بت موحدة موحدة عدد صحيح موقّع أو 4/8/16/16/32 بت عدد صحيح غير موقّع أو 4/8/8/16/32 بت كمية موحدة لكل محور موقّع أو 4/8/16/32 بت كمية موحدة لكل محور غير موقّع. رمز الرمز المميز أو المتداخل مع أي مزيج من الموتر المرتبة من نوع f8e4m3b11fnuz أو نوع f8e4m3fn أو نوع f8e4m3fnuz أو نوع f8e5m2 أو نوع f8e5m2fnuz أو 16 بت عطف أو 32-bit -bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8 /16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values or ranked tensor of 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16 /32-bit uniform quantized per axis unsigned integer values or token values |
mhlo.cosine
(mhlo::CosineOp)
Cosine operation
بناء الجملة:
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
مثال:
%result = mhlo.cosine %operand : tensor<2xf32>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values |
mhlo.count_leading_zeros
(mhlo::ClzOp)
Clz operation
بناء الجملة:
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
مثال:
%result = mhlo.count_leading_zeros %operand : tensor<2x2xi8>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
mhlo.create_token
(mhlo::CreateTokenOp)
CreateToken operation
بناء الجملة:
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
مثال:
%output = mhlo.create_token : !mhlo.token
Traits: AlwaysSpeculatableImplTrait
Interfaces: ConditionallySpeculatable
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
نتائج:
نتيجة | وصف |
---|---|
output | token |
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
مثال:
%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{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
replica_groups | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.cstr_reshapable
(mhlo::CstrReshapableOp)
CstrReshapable operation
بناء الجملة:
operation ::= `mhlo.cstr_reshapable` operands attr-dict `:` functional-type(operands, results)
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 creates a witness on the constraint that ComputeReshapeShape would succeed with the provided operands.
مثال:
%result = mhlo.cstr_reshapable %num_elements, %dynamic_shape
: (index, tensor<3xi32>) -> !shape.witness
Traits: AlwaysSpeculatableImplTrait
Interfaces: ConditionallySpeculatable
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
num_elements | فِهرِس |
dynamic_shape | 1D tensor of integer or index values |
نتائج:
نتيجة | وصف |
---|---|
result |
mhlo.custom_call
(mhlo::CustomCallOp)
CustomCall operation
بناء الجملة:
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
مثال:
%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
صفات:
يصف | MLIR Type | وصف |
---|---|---|
call_target_name | ::mlir::StringAttr | string attribute |
has_side_effect | ::mlir::BoolAttr | bool attribute |
backend_config | ::mlir::Attribute | string attribute or dictionary of named attribute values |
api_version | ::mlir::mhlo::CustomCallApiVersionAttr | Custom call API version |
called_computations | ::mlir::ArrayAttr | flat symbol ref array attribute |
custom_call_schedule | ::mlir::mhlo::CustomCallScheduleAttr | Specifies the desired schedule for the custom-call. |
operand_layouts | ::mlir::ArrayAttr | Array of layout (1D tensor of index type) attributes |
result_layouts | ::mlir::ArrayAttr | Array of layout (1D tensor of index type) attributes |
output_operand_aliases | ::mlir::ArrayAttr | Aliasing attribute for outputs and operands of CustomCall |
Operands:
المعامل | وصف |
---|---|
inputs | variadic of tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values or token or nested tuple with any combination of tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values or token values |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | variadic of tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values or token or nested tuple with any combination of tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values or token values |
mhlo.divide
(mhlo::DivOp)
Div operation
بناء الجملة:
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
مثال:
%result = mhlo.divide %lhs, %rhs : tensor<4xf32>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
lhs | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
rhs | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
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{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
kind | ::mlir::mhlo::DomainKindAttr | Kind of domain metatdata attached to an HLO domain. |
entry_metadata | ::mlir::StringAttr | string attribute |
exit_metadata | ::mlir::StringAttr | string attribute |
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values or ranked tensor of 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values or token |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values or ranked tensor of 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values or token |
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
مثال:
%0 = mhlo.dot %arg0, %arg1 : (tensor<1x2xi32>, tensor<2x1xi32>) -> tensor<1x1xi32>
Traits: AlwaysSpeculatableImplTrait
Interfaces: ConditionallySpeculatable
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
precision_config | ::mlir::ArrayAttr | Precision Config attribute |
Operands:
المعامل | وصف |
---|---|
lhs | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
rhs | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
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
مثال:
%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{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
dot_dimension_numbers | ::mlir::mhlo::DotDimensionNumbersAttr | Attribute that models the dimension information for dot. |
precision_config | ::mlir::ArrayAttr | Precision Config attribute |
Operands:
المعامل | وصف |
---|---|
lhs | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
rhs | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.dynamic_broadcast_in_dim
(mhlo::DynamicBroadcastInDimOp)
DynamicBroadcastInDim 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 BroadcastInDimOp except that the result shape is specified dynamically via output_dimensions
: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#broadcast_in_dim
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.
Traits: AlwaysSpeculatableImplTrait
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
broadcast_dimensions | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
known_expanding_dimensions | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
known_nonexpanding_dimensions | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
output_dimensions | 1D tensor of index or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
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
مثال:
%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{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
window_strides | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
padding | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
lhs_dilation | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
rhs_dilation | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
window_reversal | ::mlir::DenseElementsAttr | constant boolean vector/tensor attribute |
dimension_numbers | ::mlir::mhlo::ConvDimensionNumbersAttr | Structure of dimension information for conv op |
feature_group_count | ::mlir::IntegerAttr | 64-bit signless integer attribute |
batch_group_count | ::mlir::IntegerAttr | 64-bit signless integer attribute |
precision_config | ::mlir::ArrayAttr | Precision Config attribute |
Operands:
المعامل | وصف |
---|---|
lhs | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
rhs | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
d_padding | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
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
مثال:
%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{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
dimension_numbers | ::mlir::mhlo::GatherDimensionNumbersAttr | Attribute that models the dimension information for gather |
indices_are_sorted | ::mlir::BoolAttr | bool attribute |
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
start_indices | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
slice_sizes | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.dynamic_iota
(mhlo::DynamicIotaOp)
DynamicIota 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 IotaOp except that the result shape is specified dynamically via output_shape
: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#iota
مثال:
%0 = mhlo.dynamic_iota %arg0, dim = 0 : (tensor<1xindex>) -> tensor<4xi32>
Traits: AlwaysSpeculatableImplTrait
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
iota_dimension | ::mlir::IntegerAttr | 64-bit signless integer attribute |
Operands:
المعامل | وصف |
---|---|
output_shape | 1D tensor of index or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.dynamic_pad
(mhlo::DynamicPadOp)
DynamicPad operation
بناء الجملة:
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:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
padding_value | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
edge_padding_low | 1D tensor of index or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
edge_padding_high | 1D tensor of index or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
interior_padding | 1D tensor of index or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.dynamic_reshape
(mhlo::DynamicReshapeOp)
DynamicReshape operation
بناء الجملة:
operation ::= `mhlo.dynamic_reshape` operands attr-dict `:` functional-type(operands, results)
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 ReshapeOp except that the result shape is specified dynamically via output_shape
: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#reshape
مثال:
%0 = mhlo.dynamic_reshape %arg0, %shape : (tensor<?xf32>, tensor<2xindex>) -> tensor<?x?xf32>
Traits: AlwaysSpeculatableImplTrait
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
output_shape | 1D tensor of index or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
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
مثال:
%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{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
slice_sizes | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
start_indices | variadic of 0D tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.dynamic_update_slice
(mhlo::DynamicUpdateSliceOp)
DynamicUpdateSlice operation
بناء الجملة:
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
مثال:
%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:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
update | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
start_indices | variadic of 0D tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
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
مثال:
%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{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
einsum_config | ::mlir::StringAttr | string attribute |
Operands:
المعامل | وصف |
---|---|
lhs | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
rhs | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.erf
(mhlo::ErfOp)
Erf operation
بناء الجملة:
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.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#erf
مثال:
%result = mhlo.erf %operand : tensor<2x2xf32>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type values |
mhlo.exponential
(mhlo::ExpOp)
Exp operation
بناء الجملة:
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
مثال:
%result = mhlo.exponential %operand : tensor<2x2xf64>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values |
mhlo.exponential_minus_one
(mhlo::Expm1Op)
Expm1 operation
بناء الجملة:
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
مثال:
%result = mhlo.exponential_minus_one %operand : tensor<2xf32>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values |
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
مثال:
%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{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
fft_type | ::mlir::mhlo::FftTypeAttr | XLA fast fourier transform type. |
fft_length | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.floor
(mhlo::FloorOp)
Floor operation
بناء الجملة:
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
مثال:
%result = mhlo.floor %operand : tensor<2xf32>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type values |
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.
صفات:
يصف | MLIR Type | وصف |
---|---|---|
fusion_kind | ::mlir::mhlo::FusionKindAttr | fusion kind |
output_operand_aliases | ::mlir::ArrayAttr | Aliasing attribute for outputs and operands of Fusion |
Operands:
المعامل | وصف |
---|---|
inputs | variadic of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values or ranked tensor of 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values or token |
نتائج:
نتيجة | وصف |
---|---|
results | variadic of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values or nested tuple with any combination of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values or ranked tensor of 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values or token values |
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
مثال:
%result = "mhlo.gather"(%operand, %start_indices) {
dimension_numbers = #mhlo.gather<
offset_dims = [2, 3],
collapsed_slice_dims = [0],
start_index_map = [0, 2],
index_vector_dim = 2>,
slice_sizes = dense<[0, 2, 2]> : tensor<3xi64>,
indices_are_sorted = false
} : (tensor<3x4x2xi32>, tensor<2x3x2xi64>) -> tensor<2x3x2x2xi32>
Traits: AlwaysSpeculatableImplTrait
, InferTensorType
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
dimension_numbers | ::mlir::mhlo::GatherDimensionNumbersAttr | Attribute that models the dimension information for gather |
slice_sizes | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
indices_are_sorted | ::mlir::BoolAttr | bool attribute |
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
start_indices | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
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
مثال:
%result = mhlo.get_dimension_size %operand, dim = 1 : (tensor<2x3xf32>) -> tensor<i32>
Traits: AlwaysSpeculatableImplTrait
, InferTensorType
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
dimension | ::mlir::IntegerAttr | 64-bit signless integer attribute |
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | tensor of 32-bit signless integer values |
mhlo.get_tuple_element
(mhlo::GetTupleElementOp)
GetTupleElement operation
بناء الجملة:
operation ::= `mhlo.get_tuple_element` $operand `[` $index `]` attr-dict `:` functional-type(operands, results)
Extracts element at index
position of the operand
tuple and produces a result
.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#get_tuple_element
مثال:
%result = mhlo.get_tuple_element %operand[0] : (tuple<tensor<2xf32>, tuple<tensor<i32>>>) -> tensor<2xf32>
Traits: AlwaysSpeculatableImplTrait
Interfaces: ConditionallySpeculatable
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
index | ::mlir::IntegerAttr | 32-bit signless integer attribute |
Operands:
المعامل | وصف |
---|---|
operand | nested tuple with any combination of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values or ranked tensor of 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values or token values |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values or token or nested tuple with any combination of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values or ranked tensor of 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values or token values |
mhlo.if
(mhlo::IfOp)
If operation
Produces the output from executing exactly one branch from true_branch
or false_branch
depending on the value of pred
.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#if
Example: %result = "mhlo.if"(%pred) ({ "mhlo.return"(%result_true_branch) : (tensor
Traits: RecursiveMemoryEffects
, SingleBlockImplicitTerminator<ReturnOp>
, SingleBlock
Interfaces: InferTypeOpInterface
Operands:
المعامل | وصف |
---|---|
pred | ranked tensor of pred (AKA boolean or 1-bit integer) values |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | variadic of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values or ranked tensor of 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values or token |
mhlo.imag
(mhlo::ImagOp)
Imag operation
بناء الجملة:
operation ::= `mhlo.imag` $operand attr-dict
`:` custom<SameOperandsAndResultType>(type($operand), type($result))
Extracts the imaginary part, element-wise, from the operand
and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#imag
مثال:
%result = mhlo.imag %operand : (tensor<2xcomplex<f32>>) -> tensor<2xf32>
Traits: AlwaysSpeculatableImplTrait
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type values |
mhlo.infeed
(mhlo::InfeedOp)
Infeed operation
Reads data from the infeed and produces results
.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#infeed
مثال:
%results:2 = "mhlo.infeed"(%token) {
infeed_config = ""
} : (!mhlo.token) -> (tensor<3x3x3xi32>, !mhlo.token)
صفات:
يصف | MLIR Type | وصف |
---|---|---|
infeed_config | ::mlir::StringAttr | string attribute |
layout | ::mlir::ArrayAttr | array attribute |
Operands:
المعامل | وصف |
---|---|
token | token |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | variadic of statically shaped tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values or statically shaped tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values or token |
mhlo.iota
(mhlo::IotaOp)
Iota operation
Fills an output
tensor with values in increasing order starting from zero along the iota_dimension
dimension.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#iota
مثال:
%output = mhlo.iota dim = 0 : tensor<4x5xi32>
Traits: AlwaysSpeculatableImplTrait
Interfaces: ConditionallySpeculatable
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
iota_dimension | ::mlir::IntegerAttr | 64-bit signless integer attribute |
نتائج:
نتيجة | وصف |
---|---|
output | statically shaped tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements values |
mhlo.is_finite
(mhlo::IsFiniteOp)
IsFinite operation
بناء الجملة:
operation ::= `mhlo.is_finite` $x attr-dict `:` functional-type(operands, results)
Performs element-wise check whether the value in x
is finite (ie is neither +Inf, -Inf, nor NaN) and produces a y
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#is_finite
مثال:
%y = mhlo.is_finite %x : (tensor<7xf32>) -> tensor<7xi1>
Traits: AlwaysSpeculatableImplTrait
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
x | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type values |
نتائج:
نتيجة | وصف |
---|---|
y | ranked tensor of pred (AKA boolean or 1-bit integer) values |
mhlo.log
(mhlo::LogOp)
Log operation
بناء الجملة:
operation ::= `mhlo.log` $operand attr-dict
`:` custom<SameOperandsAndResultType>(type($operand), type($result))
Performs element-wise logarithm operation on operand
tensor and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#log
مثال:
%result = mhlo.log %operand : tensor<2x2xf64>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values |
mhlo.log_plus_one
(mhlo::Log1pOp)
Log1p operation
بناء الجملة:
operation ::= `mhlo.log_plus_one` $operand attr-dict
`:` custom<SameOperandsAndResultType>(type($operand), type($result))
Performs element-wise logarithm plus one operation on operand
tensor and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#log_plus_one
مثال:
%result = mhlo.log_plus_one %operand : tensor<6xf32>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values |
mhlo.logistic
(mhlo::LogisticOp)
Logistic operation
بناء الجملة:
operation ::= `mhlo.logistic` $operand attr-dict
`:` custom<SameOperandsAndResultType>(type($operand), type($result))
Performs element-wise logistic operation on operand
tensor and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#logistic
مثال:
%result = mhlo.logistic %operand : tensor<2x2xf32>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values |
mhlo.map
(mhlo::MapOp)
Map operation
Applies a map function computation
to inputs
along the dimensions
and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#map
مثال:
%result = "mhlo.map"(%input0, %input1) ({
^bb0(%arg0: tensor<i32>, %arg1: tensor<i32>):
%0 = mhlo.multiply %arg0, %arg1 : tensor<i32>
mhlo.return %0 : tensor<i32>
}) {
dimensions = dense<[0, 1]> : tensor<2xi64>
} : (tensor<2x2xi32>, tensor<2x2xi32>) -> tensor<2x2xi32>
Traits: InferTensorType
, RecursiveMemoryEffects
, SameOperandsAndResultShape
, SingleBlockImplicitTerminator<ReturnOp>
, SingleBlock
Interfaces: InferShapedTypeOpInterface
, InferTypeOpInterface
صفات:
يصف | MLIR Type | وصف |
---|---|---|
dimensions | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
Operands:
المعامل | وصف |
---|---|
inputs | variadic of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4 /8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer قيم |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.maximum
(mhlo::MaxOp)
Max operation
بناء الجملة:
operation ::= `mhlo.maximum` $lhs `,` $rhs attr-dict
`:` custom<SameOperandsAndResultType>(type($lhs), type($rhs), type($result))
Performs element-wise max operation on tensors lhs
and rhs
and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#maximum
مثال:
%result = mhlo.maximum %lhs, %rhs : tensor<4xf32>
Traits: AlwaysSpeculatableImplTrait
, Commutative
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
lhs | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
rhs | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.minimum
(mhlo::MinOp)
Min operation
بناء الجملة:
operation ::= `mhlo.minimum` $lhs `,` $rhs attr-dict
`:` custom<SameOperandsAndResultType>(type($lhs), type($rhs), type($result))
Performs element-wise min operation on tensors lhs
and rhs
and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#minimum
مثال:
%result = mhlo.minimum %lhs, %rhs : tensor<4xf32>
Traits: AlwaysSpeculatableImplTrait
, Commutative
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
lhs | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
rhs | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.multiply
(mhlo::MulOp)
Mul operation
بناء الجملة:
operation ::= `mhlo.multiply` $lhs `,` $rhs attr-dict
`:` custom<SameOperandsAndResultType>(type($lhs), type($rhs), type($result))
Performs element-wise product of two tensors lhs
and rhs
and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#multiply
مثال:
%result = mhlo.multiply %lhs, %rhs : tensor<2xi32>
Traits: AlwaysSpeculatableImplTrait
, Commutative
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
lhs | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
rhs | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.negate
(mhlo::NegOp)
Neg operation
بناء الجملة:
operation ::= `mhlo.negate` $operand attr-dict
`:` custom<SameOperandsAndResultType>(type($operand), type($result))
Performs element-wise negation of operand
tensor and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#negate
مثال:
%result = mhlo.negate %operand : tensor<2x3xi32>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.not
(mhlo::NotOp)
Not operation
بناء الجملة:
operation ::= `mhlo.not` $operand attr-dict
`:` custom<SameOperandsAndResultType>(type($operand), type($result))
Performs element-wise NOT of tensor operand
of type integer and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#not
مثال:
%result = mhlo.not %operand : tensor<5x3x1xi1>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
mhlo.optimization_barrier
(mhlo::OptimizationBarrierOp)
OptimizationBarrier operation
بناء الجملة:
operation ::= `mhlo.optimization_barrier` attr-dict ($operand^ `:` custom<PairwiseOpType>(type($operand), type($result))):(`(` `)`)?
Ensures that the operations that produce the operand
are executed before any operations that depend on the result
and prevents compiler transformations from moving operations across the barrier. Other than that, the operation is an identity, ie result
= operand
.
See https://github.com/openxla/stablehlo/blob/main/docs/spec.md#optimization_barrier
مثال:
%result0, %result1 = mhlo.optimization_barrier %operand0, %operand1 : tensor<f32>, tensor<f32>
Traits: AlwaysSpeculatableImplTrait
, HLO_PairwiseSameOperandAndResultType
Interfaces: ConditionallySpeculatable
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | variadic of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values or ranked tensor of 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values or token |
نتائج:
نتيجة | وصف |
---|---|
result | variadic of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values or ranked tensor of 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values or token |
mhlo.or
(mhlo::OrOp)
Or operation
بناء الجملة:
operation ::= `mhlo.or` $lhs `,` $rhs attr-dict
`:` custom<SameOperandsAndResultType>(type($lhs), type($rhs), type($result))
Performs element-wise OR of two tensors lhs
and rhs
and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#or
مثال:
%result = mhlo.or %lhs, %rhs : tensor<2xi1>
Traits: AlwaysSpeculatableImplTrait
, Commutative
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
lhs | ranked tensor of pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
rhs | ranked tensor of pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.outfeed
(mhlo::OutfeedOp)
Outfeed operation
Writes inputs
to the outfeed and produces a result
token.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#outfeed
مثال:
%result = "mhlo.outfeed"(%input0, %token) {
outfeed_config = ""
} : (tensor<3x3x3xi32>, !mhlo.token) -> !mhlo.token
Interfaces: InferTypeOpInterface
صفات:
يصف | MLIR Type | وصف |
---|---|---|
outfeed_config | ::mlir::StringAttr | string attribute |
Operands:
المعامل | وصف |
---|---|
inputs | variadic of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4 /8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer قيم |
token | token |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | token |
mhlo.pad
(mhlo::PadOp)
Pad operation
Expands operand
by padding around the tensor as well as between the elements of the tensor with the given padding_value
.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#pad
مثال:
%0 = mhlo.pad %arg0, %arg1, low = [0, 1], high = [2, 1], interior = [1, 2]
: (tensor<2x3xi32>, tensor<i32>) -> tensor<5x9xi32>
Traits: AlwaysSpeculatableImplTrait
, SameOperandsAndResultElementType
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
edge_padding_low | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
edge_padding_high | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
interior_padding | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
padding_value | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.partition_id
(mhlo::PartitionIdOp)
PartitionId operation
بناء الجملة:
operation ::= `mhlo.partition_id` attr-dict `:` type(results)
Produces partition_id
of the current process.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#partition_id
مثال:
%result = mhlo.partition_id : tensor<ui32>
Interfaces: InferTypeOpInterface
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | ranked tensor of 32-bit unsigned integer values |
mhlo.popcnt
(mhlo::PopulationCountOp)
PopulationCount operation
بناء الجملة:
operation ::= `mhlo.popcnt` $operand attr-dict
`:` custom<SameOperandsAndResultType>(type($operand), type($result))
Performs element-wise count of the number of bits set in the operand
tensor and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#popcnt
مثال:
%result = mhlo.popcnt %operand : tensor<4xi8>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
mhlo.power
(mhlo::PowOp)
Pow operation
بناء الجملة:
operation ::= `mhlo.power` $lhs `,` $rhs attr-dict
`:` custom<SameOperandsAndResultType>(type($lhs), type($rhs), type($result))
Performs element-wise exponentiation of lhs
tensor by rhs
tensor and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#power
مثال:
%result = mhlo.power %lhs, %rhs : tensor<6xf32>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
lhs | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
rhs | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.real
(mhlo::RealOp)
Real operation
بناء الجملة:
operation ::= `mhlo.real` $operand attr-dict
`:` custom<SameOperandsAndResultType>(type($operand), type($result))
Extracts the real part, element-wise, from the operand
and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#real
مثال:
%result = mhlo.real %operand : (tensor<2xcomplex<f32>>) -> tensor<2xf32>
Traits: AlwaysSpeculatableImplTrait
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type values |
mhlo.real_dynamic_slice
(mhlo::RealDynamicSliceOp)
RealDynamicSlice operation
بناء الجملة:
operation ::= `mhlo.real_dynamic_slice` operands attr-dict `:` functional-type(operands, results)
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 SliceOp except that start_indices
, limit_indices
and strides
are specified dynamically: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#slice
مثال:
%result = mhlo.real_dynamic_slice %operand,
%start_indices, %limit_indices, %strides
: (tensor<256x?xf32>, tensor<2xindex>, tensor<2xindex>, tensor<2xindex>) -> tensor<256x?xf32>
Traits: AlwaysSpeculatableImplTrait
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
start_indices | 1D tensor of index or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
limit_indices | 1D tensor of index or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
strides | 1D tensor of index or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.recv
(mhlo::RecvOp)
Recv operation
Receives data from a channel with channel_id
and produces results
.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#recv
مثال:
%results:2 = "mhlo.recv"(%token) {
// channel_id = 5 : i64,
// channel_type = #stablehlo<channel_type HOST_TO_DEVICE>,
channel_handle = #mhlo.channel_handle<handle = 5, type = 3>,
is_host_transfer = true
} : (!mhlo.token) -> (tensor<3x4xi32>, !mhlo.token)
صفات:
يصف | MLIR Type | وصف |
---|---|---|
channel_handle | ::mlir::mhlo::ChannelHandleAttr | two 64-bit integers 'handle' and 'type' |
is_host_transfer | ::mlir::BoolAttr | bool attribute |
Operands:
المعامل | وصف |
---|---|
token | token |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | variadic of statically shaped tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values or statically shaped tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values or token |
mhlo.reduce
(mhlo::ReduceOp)
Reduce operation
Applies a reduction function body
to inputs
and init_values
along the dimensions
and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#reduce
مثال:
%result = "mhlo.reduce"(%input, %init_value) ({
^bb0(%arg0: tensor<i32>, %arg1: tensor<i32>):
%0 = "mhlo.add"(%arg0, %arg1) : (tensor<i32>, tensor<i32>) -> tensor<i32>
"mhlo.return"(%0) : (tensor<i32>) -> ()
}) {
dimensions = dense<1> : tensor<1xi64>
} : (tensor<1x6xi32>, tensor<i32>) -> tensor<1xi32>
Traits: InferTensorType
, RecursiveMemoryEffects
, SameVariadicOperandSize
, SingleBlockImplicitTerminator<ReturnOp>
, SingleBlock
Interfaces: InferShapedTypeOpInterface
, InferTypeOpInterface
صفات:
يصف | MLIR Type | وصف |
---|---|---|
dimensions | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
Operands:
المعامل | وصف |
---|---|
inputs | variadic of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4 /8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer قيم |
init_values | variadic of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4 /8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer قيم |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | variadic of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4 /8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer قيم |
mhlo.reduce_precision
(mhlo::ReducePrecisionOp)
ReducePrecision operation
بناء الجملة:
operation ::= `mhlo.reduce_precision` $operand `,` `format` `=` custom<ExponentMantissa>($exponent_bits, $mantissa_bits)
attr-dict `:` custom<SameOperandsAndResultType>(type($operand), type($output))
Performs element-wise conversion of operand
to another floating-point type that uses exponent_bits
and mantissa_bits
and back to the original floating-point type and produces an output
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#reduce_precision
مثال:
%output = mhlo.reduce_precision %operand, format = e5m2 : tensor<6xf32>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
exponent_bits | ::mlir::IntegerAttr | 32-bit signless integer attribute |
mantissa_bits | ::mlir::IntegerAttr | 32-bit signless integer attribute |
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type values |
نتائج:
نتيجة | وصف |
---|---|
output | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type values |
mhlo.reduce_scatter
(mhlo::ReduceScatterOp)
ReduceScatter operation
Within each process group in the process grid, performs reduction, using computations
, over the values of the operand
tensor from each process, splits the reduction result along scatter_dimension
into parts, and scatters the split parts between the processes to produce the result
.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#reduce_scatter
مثال:
%result = "mhlo.reduce_scatter"(%operand) ({
^bb0(%arg0: tensor<f32>, %arg1: tensor<f32>):
%0 = mhlo.add %arg0, %arg1 : tensor<f32>
mhlo.return %0 : tensor<f32>
}) {
scatter_dimension = 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<2x4xf32>) -> tensor<2x2xf32>
صفات:
يصف | MLIR Type | وصف |
---|---|---|
scatter_dimension | ::mlir::IntegerAttr | 64-bit signless integer attribute |
replica_groups | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
channel_handle | ::mlir::mhlo::ChannelHandleAttr | two 64-bit integers 'handle' and 'type' |
use_global_device_ids | ::mlir::UnitAttr | unit attribute |
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.reduce_window
(mhlo::ReduceWindowOp)
ReduceWindow operation
Applies a reduction function body
to windows of inputs
and init_values
and produces results
.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#reduce_window
مثال:
%result = "mhlo.reduce_window"(%input, %init_value) ({
^bb0(%arg0: tensor<i32>, %arg1: tensor<i32>):
%0 = mhlo.add %arg0, %arg1 : tensor<i32>
mhlo.return %0 : tensor<i32>
}) {
window_dimensions = dense<[2, 1]> : tensor<2xi64>,
window_strides = dense<[4, 1]> : tensor<2xi64>,
base_dilations = dense<[2, 1]> : tensor<2xi64>,
window_dilations = dense<[3, 1]> : tensor<2xi64>,
padding = dense<[[2, 1], [0, 0]]> : tensor<2x2xi64>
} : (tensor<3x2xi32>, tensor<i32>) -> tensor<2x2xi32>
Traits: InferTensorType
, RecursiveMemoryEffects
, SameVariadicOperandSize
, SingleBlockImplicitTerminator<ReturnOp>
, SingleBlock
Interfaces: InferShapedTypeOpInterface
, InferTypeOpInterface
صفات:
يصف | MLIR Type | وصف |
---|---|---|
window_dimensions | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
window_strides | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
base_dilations | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
window_dilations | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
padding | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
Operands:
المعامل | وصف |
---|---|
inputs | variadic of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4 /8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer قيم |
init_values | variadic of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4 /8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer قيم |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | variadic of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4 /8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer قيم |
mhlo.remainder
(mhlo::RemOp)
Rem operation
بناء الجملة:
operation ::= `mhlo.remainder` $lhs `,` $rhs attr-dict
`:` custom<SameOperandsAndResultType>(type($lhs), type($rhs), type($result))
Performs element-wise remainder of dividend lhs
and divisor rhs
tensors and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#remainder
مثال:
%result = mhlo.remainder %lhs, %rhs : tensor<4xi64>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
lhs | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
rhs | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.replica_id
(mhlo::ReplicaIdOp)
ReplicaId operation
بناء الجملة:
operation ::= `mhlo.replica_id` attr-dict `:` type(results)
Produces replica_id
of the current process.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#replica_id
مثال:
%result = mhlo.replica_id : tensor<ui32>
Traits: AlwaysSpeculatableImplTrait
Interfaces: ConditionallySpeculatable
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | ranked tensor of 32-bit unsigned integer values |
mhlo.reshape
(mhlo::ReshapeOp)
Reshape operation
بناء الجملة:
operation ::= `mhlo.reshape` operands attr-dict `:` functional-type(operands, results)
Performs reshape of operand
tensor to a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#reshape
مثال:
%result = mhlo.reshape %operand : (tensor<2xf32>) -> tensor<1x2xf32>
Traits: AlwaysSpeculatableImplTrait
, HLO_CompatibleOperandsAndResultElementType
Interfaces: ConditionallySpeculatable
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | statically shaped tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.return
(mhlo::ReturnOp)
_This operation is a work in progress, so it is not yet included in the specification: https://github.com/openxla/stablehlo/issues/425
Informally, this operation serves as a terminator for regions defined by
the StableHLO ops. Non-StableHLO ops, e.g. `func.func`, have their own
terminators, e.g. `func.return`.
Example:
```mlir
%result = "mhlo.reduce"(%input, %init_value) ({
^bb0(%arg0: tensor<i32>, %arg1: tensor<i32>):
%0 = "mhlo.add"(%arg0, %arg1) : (tensor<i32>, tensor<i32>) -> tensor<i32>
"mhlo.return"(%0) : (tensor<i32>) -> ()
}) {
dimensions = dense<1> : tensor<1xi64>
} : (tensor<1x6xi32>, tensor<i32>) -> tensor<1xi32>
```_
Syntax:
```
operation ::= mhlo.return
$results attr-dict ( :
type($results)^)?
Traits: `AlwaysSpeculatableImplTrait`, `Terminator`
Interfaces: `ConditionallySpeculatable`, `NoMemoryEffect (MemoryEffectOpInterface)`
Effects: `MemoryEffects::Effect{}`
#### Operands:
| Operand | Description |
| :-----: | ----------- |
| `results` | variadic of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values or token or nested tuple with any combination of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values or ranked tensor of 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values or token values
### `mhlo.reverse` (mhlo::ReverseOp)
_Reverse operation_
Reverses the order of elements in the `operand` along the specified
`dimensions` and produces a `result` tensor.
See:
<a href="https://github.com/openxla/stablehlo/blob/main/docs/spec.md#reverse">https://github.com/openxla/stablehlo/blob/main/docs/spec.md#reverse</a>
Example:
```mlir
%result = mhlo.reverse %operand, dims = [1] : tensor<3x2xi32>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
dimensions | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.rng
(mhlo::RngOp)
Rng operation
Generates random numbers using the rng_distribution
algorithm and produces a result
tensor of a given shape shape
.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#rng
مثال:
%result = mhlo.rng %a, %b, %shape, distribution = NORMAL : (tensor<i32>, tensor<i32>, tensor<2xi64>) -> tensor<3x3xi32>
Traits: InferTensorType
Interfaces: InferShapedTypeOpInterface
, InferTypeOpInterface
صفات:
يصف | MLIR Type | وصف |
---|---|---|
rng_distribution | ::mlir::mhlo::RngDistributionAttr | XLA PRNG distribution to be used. |
Operands:
المعامل | وصف |
---|---|
a | 0D tensor of pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type values |
b | 0D tensor of pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type values |
shape | 1D tensor of index or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type values |
mhlo.rng_bit_generator
(mhlo::RngBitGeneratorOp)
RngBitGenerator operation
Returns an output
filled with uniform random data and an updated output state output_state
given an initial state initial_state
using the pseudorandom number generator algorithm rng_algorithm
.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#rng_bit_generator
مثال:
%output_state, %output = mhlo.rng_bit_generator %initial_state, algorithm = THREE_FRY : (tensor<2xui64>) -> (tensor<2xui64>, tensor<2x2xui64>)
Traits: AlwaysSpeculatableImplTrait
Interfaces: ConditionallySpeculatable
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
rng_algorithm | ::mlir::mhlo::RngAlgorithmAttr | XLA PRNG algorithm to be used. |
Operands:
المعامل | وصف |
---|---|
initial_state | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type values |
نتائج:
نتيجة | وصف |
---|---|
output_state | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type values |
output | statically shaped tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type values |
mhlo.round_nearest_afz
(mhlo::RoundOp)
Round operation
بناء الجملة:
operation ::= `mhlo.round_nearest_afz` $operand attr-dict
`:` custom<SameOperandsAndResultType>(type($operand), type($result))
Performs element-wise rounding towards the nearest integer, breaking ties away from zero, on the operand
tensor and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#round_nearest_afz
مثال:
%result = mhlo.round_nearest_afz %operand : tensor<5xf32>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type values |
mhlo.round_nearest_even
(mhlo::RoundNearestEvenOp)
RoundNearestEven operation
بناء الجملة:
operation ::= `mhlo.round_nearest_even` $operand attr-dict
`:` custom<SameOperandsAndResultType>(type($operand), type($result))
Performs element-wise rounding towards the nearest integer, breaking ties towards the even integer, on the operand
tensor and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#round_nearest_even
مثال:
%result = mhlo.round_nearest_even %operand : tensor<5xf32>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type values |
mhlo.rsqrt
(mhlo::RsqrtOp)
Rsqrt operation
بناء الجملة:
operation ::= `mhlo.rsqrt` $operand attr-dict
`:` custom<SameOperandsAndResultType>(type($operand), type($result))
Performs element-wise reciprocal square root operation on operand
tensor and produces a result
tensor, implementing the rSqrt
operation from the IEEE-754 specification.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#rsqrt
مثال:
%result = mhlo.rsqrt %operand : tensor<2x2xf32>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values |
mhlo.scatter
(mhlo::ScatterOp)
Scatter operation
Produces results
tensors which are equal to inputs
tensors except that several slices specified by scatter_indices
are updated with the values updates
using update_computation
.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#scatter
مثال:
%result = "mhlo.scatter"(%input, %scatter_indices, %update) ({
^bb0(%arg0: tensor<i32>, %arg1: tensor<i32>):
%0 = mhlo.add %arg0, %arg1 : tensor<i32>
mhlo.return %0 : tensor<i32>
}) {
scatter_dimension_numbers = #mhlo.scatter<
update_window_dims = [2,3],
inserted_window_dims = [0],
scatter_dims_to_operand_dims = [1, 0],
index_vector_dim = 2>,
indices_are_sorted = false,
unique_indices = false
} : (tensor<3x4x2xi32>, tensor<2x3x2xi64>, tensor<2x3x2x2xi32>) -> tensor<3x4x2xi32>
Traits: RecursiveMemoryEffects
, SameVariadicOperandSize
Interfaces: InferTypeOpInterface
صفات:
يصف | MLIR Type | وصف |
---|---|---|
scatter_dimension_numbers | ::mlir::mhlo::ScatterDimensionNumbersAttr | Attribute that models the dimension information for scatter |
indices_are_sorted | ::mlir::BoolAttr | bool attribute |
unique_indices | ::mlir::BoolAttr | bool attribute |
Operands:
المعامل | وصف |
---|---|
inputs | variadic of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4 /8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer قيم |
scatter_indices | ranked tensor of integer or index values |
updates | variadic of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4 /8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer قيم |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | variadic of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4 /8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer قيم |
mhlo.select
(mhlo::SelectOp)
Select operation
بناء الجملة:
operation ::= `mhlo.select` operands attr-dict `:`
custom<SelectOpType>(type($pred), type($on_true), type($on_false), type($result))
Produces a result
tensor where each element is selected from on_true
or on_false
tensor based on the value of the corresponding element of pred
.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#select
مثال:
%result = mhlo.select %pred, %on_true, %on_false : tensor<2x2xi1>, tensor<2x2xi32>
Traits: AlwaysSpeculatableImplTrait
, HLO_BroadcastingElementwise
, InferTensorType
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
pred | ranked tensor of pred (AKA boolean or 1-bit integer) values |
on_true | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
on_false | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.select_and_scatter
(mhlo::SelectAndScatterOp)
SelectAndScatter operation
Scatters the values from the source
tensor using scatter
based on the outcome of reduce_window
of the input
tensor using select
and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#select_and_scatter
مثال:
%result = "mhlo.select_and_scatter"(%operand, %source, %init_value) ({
^bb0(%arg0: tensor<i32>, %arg1: tensor<i32>):
%0 = "mhlo.compare"(%arg0, %arg1) {
comparison_direction = #stablehlo<comparison_direction GE>
} : (tensor<i32>, tensor<i32>) -> tensor<i1>
"mhlo.return"(%0) : (tensor<i1>) -> ()
}, {
^bb0(%arg0: tensor<i32>, %arg1: tensor<i32>):
%0 = "mhlo.add"(%arg0, %arg1) : (tensor<i32>, tensor<i32>) -> tensor<i32>
"mhlo.return"(%0) : (tensor<i32>) -> ()
}) {
window_dimensions = dense<[3, 1]> : tensor<2xi64>,
window_strides = dense<[2, 1]> : tensor<2xi64>,
padding = dense<[[0, 1], [0, 0]]> : tensor<2x2xi64>
} : (tensor<4x2xi32>, tensor<2x2xi32>, tensor<i32>) -> tensor<4x2xi32>
Traits: RecursiveMemoryEffects
Interfaces: InferTypeOpInterface
صفات:
يصف | MLIR Type | وصف |
---|---|---|
window_dimensions | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
window_strides | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
padding | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
source | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
init_value | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.send
(mhlo::SendOp)
Send operation
Sends inputs
to a channel channel_id
and produces a result
token.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#send
مثال:
%result = "mhlo.send"(%operand, %token) {
// channel_id = 5 : i64,
// channel_type = #stablehlo<channel_type DEVICE_TO_HOST>,
channel_handle = #mhlo.channel_handle<handle = 5, type = 2>,
is_host_transfer = true
} : (tensor<3x4xi32>, !mhlo.token) -> !mhlo.token
Interfaces: InferTypeOpInterface
صفات:
يصف | MLIR Type | وصف |
---|---|---|
channel_handle | ::mlir::mhlo::ChannelHandleAttr | two 64-bit integers 'handle' and 'type' |
is_host_transfer | ::mlir::BoolAttr | bool attribute |
Operands:
المعامل | وصف |
---|---|
inputs | variadic of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4 /8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer قيم |
token | token |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | token |
mhlo.set_dimension_size
(mhlo::SetDimensionSizeOp)
SetDimensionSize 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 XLA's SetDimensionSize: https://www.tensorflow.org/xla/operation_semantics#setdimensionsize
مثال:
%0 = mhlo.set_dimension_size %arg0, %arg1, dim = 1 : (tensor<4x2xf32>, tensor<i32>) -> tensor<4x2xf32>
Traits: AlwaysSpeculatableImplTrait
, InferTensorType
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
dimension | ::mlir::IntegerAttr | 64-bit signless integer attribute |
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
size | tensor of 32-bit signless integer values |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.shift_left
(mhlo::ShiftLeftOp)
ShiftLeft operation
بناء الجملة:
operation ::= `mhlo.shift_left` $lhs `,` $rhs attr-dict
`:` custom<SameOperandsAndResultType>(type($lhs), type($rhs), type($result))
Performs element-wise left-shift operation on the lhs
tensor by rhs
number of bits and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#shift_left
مثال:
%result = mhlo.shift_left %lhs, %rhs : tensor<6xi8>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
lhs | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
rhs | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
mhlo.shift_right_arithmetic
(mhlo::ShiftRightArithmeticOp)
ShiftRightArithmetic operation
بناء الجملة:
operation ::= `mhlo.shift_right_arithmetic` $lhs `,` $rhs attr-dict
`:` custom<SameOperandsAndResultType>(type($lhs), type($rhs), type($result))
Performs element-wise arithmetic right-shift operation on the lhs
tensor by rhs
number of bits and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#shift_right_arithmetic
مثال:
%result = mhlo.shift_right_arithmetic %lhs, %rhs : tensor<6xi8>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
lhs | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
rhs | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
mhlo.shift_right_logical
(mhlo::ShiftRightLogicalOp)
ShiftRightLogical operation
بناء الجملة:
operation ::= `mhlo.shift_right_logical` $lhs `,` $rhs attr-dict
`:` custom<SameOperandsAndResultType>(type($lhs), type($rhs), type($result))
Performs element-wise logical right-shift operation on the lhs
tensor by rhs
number of bits and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#shift_right_logical
مثال:
%result = mhlo.shift_right_logical %lhs, %rhs : tensor<6xi8>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
lhs | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
rhs | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
mhlo.sign
(mhlo::SignOp)
Sign operation
بناء الجملة:
operation ::= `mhlo.sign` $operand attr-dict
`:` custom<SameOperandsAndResultType>(type($operand), type($result))
Returns the sign of the operand
element-wise and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#sign
مثال:
%result = mhlo.sign %operand : tensor<7xf32>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of 4/8/16/32/64-bit signless integer or f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of 4/8/16/32/64-bit signless integer or f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values |
mhlo.sine
(mhlo::SineOp)
Sine operation
بناء الجملة:
operation ::= `mhlo.sine` $operand attr-dict
`:` custom<SameOperandsAndResultType>(type($operand), type($result))
Performs element-wise sine operation on operand
tensor and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#sine
مثال:
%result = mhlo.sine %operand : tensor<2xf32>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values |
mhlo.slice
(mhlo::SliceOp)
Slice operation
Extracts a slice from the operand
using statically-computed starting indices and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#slice
مثال:
%result = "mhlo.slice" (%operand) {
start_indices = dense<[1, 2]> : tensor<2xi64>,
limit_indices = dense<[3, 4]> : tensor<2xi64>,
strides = dense<1> : tensor<2xi64>
} : (tensor<3x4xi64>) -> tensor<2x2xi64>
Traits: AlwaysSpeculatableImplTrait
, SameOperandsAndResultElementType
Interfaces: ConditionallySpeculatable
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
start_indices | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
limit_indices | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
strides | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.sort
(mhlo::SortOp)
Sort operation
Sorts a variadic number of tensors in inputs
together, according to a custom comparator
, along the given dimension
and produces a variadic number of tensors as results
.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#sort
مثال:
%result0, %result1 = "mhlo.sort"(%input0, %input1) ({
^bb0(%arg0: tensor<i32>, %arg1: tensor<i32>, %arg2: tensor<i32>, %arg3: tensor<i32>):
%predicate = "mhlo.compare"(%arg0, %arg1) {
comparison_direction = #stablehlo<comparison_direction GT>
} : (tensor<i32>, tensor<i32>) -> tensor<i1>
"mhlo.return"(%predicate) : (tensor<i1>) -> ()
}) {
dimension = 0 : i64,
is_stable = true
} : (tensor<2x3xi32>, tensor<2x3xi32>) -> (tensor<2x3xi32>, tensor<2x3xi32>)
Traits: InferTensorType
, RecursiveMemoryEffects
, SameOperandsAndResultShape
Interfaces: InferShapedTypeOpInterface
, InferTypeOpInterface
صفات:
يصف | MLIR Type | وصف |
---|---|---|
dimension | ::mlir::IntegerAttr | 64-bit signless integer attribute |
is_stable | ::mlir::BoolAttr | bool attribute |
Operands:
المعامل | وصف |
---|---|
inputs | variadic of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4 /8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer قيم |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | variadic of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4 /8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer قيم |
mhlo.sparse_dot
(mhlo::SparseDotOp)
Sparse dot operation
Similar to dot_general
operation, with one or both of the operands being sparse. An additional argument provides sparsity meta information. Disclaimer: this op is experimental / a work in progress.
Traits: AlwaysSpeculatableImplTrait
Interfaces: ConditionallySpeculatable
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
lhs_sparsity | ::mlir::mhlo::SparsityDescriptorAttr | Describes structured (N:M) sparsity configuration |
rhs_sparsity | ::mlir::mhlo::SparsityDescriptorAttr | Describes structured (N:M) sparsity configuration |
dot_dimension_numbers | ::mlir::mhlo::DotDimensionNumbersAttr | Attribute that models the dimension information for dot. |
precision_config | ::mlir::ArrayAttr | Precision Config attribute |
Operands:
المعامل | وصف |
---|---|
lhs | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
rhs | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
meta | variadic of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4 /8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer قيم |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.sqrt
(mhlo::SqrtOp)
Sqrt operation
بناء الجملة:
operation ::= `mhlo.sqrt` $operand attr-dict
`:` custom<SameOperandsAndResultType>(type($operand), type($result))
Performs element-wise square root operation on operand
tensor and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#sqrt
مثال:
%result = mhlo.sqrt %operand : tensor<2x2xf32>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values |
mhlo.stochastic_convert
(mhlo::StochasticConvertOp)
StochasticConvert operation
This operation is a work in progress, so it is not yet included in the specification: https://github.com/openxla/stablehlo/issues/295
Informally, this operation performs element-wise conversion of values from a bigger type to a smaller one with stochastic rounding using the random number passed in.
Traits: AlwaysSpeculatableImplTrait
, Elementwise
Interfaces: ConditionallySpeculatable
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type values |
random | ranked tensor of 4/8/16/32/64-bit unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.subtract
(mhlo::SubtractOp)
Subtract operation
بناء الجملة:
operation ::= `mhlo.subtract` $lhs `,` $rhs attr-dict
`:` custom<SameOperandsAndResultType>(type($lhs), type($rhs), type($result))
Performs element-wise subtraction of two tensors lhs
and rhs
and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#subtract
مثال:
%result = mhlo.subtract %lhs, %rhs : tensor<2xi32>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
lhs | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
rhs | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.tan
(mhlo::TanOp)
Tan operation
بناء الجملة:
operation ::= `mhlo.tan` $operand attr-dict
`:` custom<SameOperandsAndResultType>(type($operand), type($result))
This operation is a work in progress, so it is not yet included in the specification: https://github.com/openxla/stablehlo/issues/954
Informally, this operation returns Tan(operand)
element-wise.
مثال:
%0 = mhlo.tan %arg0 : tensor<2xf32>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements values |
mhlo.tanh
(mhlo::TanhOp)
Tanh operation
بناء الجملة:
operation ::= `mhlo.tanh` $operand attr-dict
`:` custom<SameOperandsAndResultType>(type($operand), type($result))
Performs element-wise hyperbolic tangent operation on operand
tensor and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#tanh
مثال:
%result = mhlo.tanh %operand : tensor<2xf32>
Traits: AlwaysSpeculatableImplTrait
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values |
mhlo.topk
(mhlo::TopKOp)
TopK operation
بناء الجملة:
operation ::= `mhlo.topk` `(`$operand `,` `k` `=` $k (`,` `largest` `=` $largest^)? `)` attr-dict `:`
type($operand) `->` `(`type($values)`,` type($indices)`)`
Returns top k
values and their indices, along the last dimension of the operand if largest=true
or the bottom k
values if largest=false
.
See: https://www.tensorflow.org/xla/operation_semantics#top-k
مثال:
%values, %indices = mhlo.topk(%operand, k=5, largest=true)
: tensor<100xf32> -> (tensor<5xf32>, tensor<5xi32>)
Traits: InferTensorType
, RecursiveMemoryEffects
Interfaces: InferShapedTypeOpInterface
, InferTypeOpInterface
صفات:
يصف | MLIR Type | وصف |
---|---|---|
k | ::mlir::IntegerAttr | 64-bit signless integer attribute |
largest | ::mlir::BoolAttr | bool attribute |
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
values | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
indices | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.torch_index_select
(mhlo::TorchIndexSelectOp)
TorchIndexSelect 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 PyTorch's index_select, augmented with support for batch dimensions: https://pytorch.org/docs/stable/generated/torch.index_select.html
The batch_dims
attribute specifies the number of major batch dimensions (0 or more) that act like a multidimensional loop over both the operand and the index.
مثال:
%result = "mhlo.torch_index_select"(%operand, %index) {
dim = 2 : i64,
batch_dims = 1 : i64
} : (tensor<8x128x3072x64xf32>, tensor<8x16x1024xi32>) -> tensor<8x128x16x1024x64xf32>
Traits: AlwaysSpeculatableImplTrait
Interfaces: ConditionallySpeculatable
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
dim | ::mlir::IntegerAttr | 64-bit signless integer attribute |
batch_dims | ::mlir::IntegerAttr | 64-bit signless integer attribute |
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
index | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.trace
(mhlo::TraceOp)
Trace operation
بناء الجملة:
operation ::= `mhlo.trace` $operand `,` $tag attr-dict `:` type($operand)
This operation is on its way out of StableHLO, so it is not included in the specification: https://github.com/openxla/stablehlo/issues/604
It is not used by JAX, PyTorch or TensorFlow, so it looks like we should've classified it as "Private to XLA" and not included it in StableHLO in the first place. With that in mind, its semantics will not be documented here.
مثال:
mhlo.trace %arg0, "In test code." : tensor<5x1x5xi32>
صفات:
يصف | MLIR Type | وصف |
---|---|---|
tag | ::mlir::StringAttr | string attribute |
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.transpose
(mhlo::TransposeOp)
Transpose operation
Permutes the dimensions of operand
tensor using permutation
and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#transpose
مثال:
%0 = mhlo.transpose %arg0, dims = [2, 1, 0] : (tensor<1x2x3xi32>) -> tensor<3x2x1xi32>
Traits: AlwaysSpeculatableImplTrait
, HLO_CompatibleOperandsAndResultElementType
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
permutation | ::mlir::DenseIntElementsAttr | 64-bit signless integer elements attribute |
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.triangular_solve
(mhlo::TriangularSolveOp)
TriangularSolve operation
Solves batches of systems of linear equations with lower or upper triangular coefficient matrices.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#triangular_solve
مثال:
%result = "mhlo.triangular_solve"(%a, %b) {
left_side = true,
lower = true,
unit_diagonal = false,
transpose_a = #stablehlo<transpose NO_TRANSPOSE>
} : (tensor<3x3xf32>, tensor<3x3xf32>) -> tensor<3x3xf32>
Traits: AlwaysSpeculatableImplTrait
, InferTensorType
, SameOperandsAndResultElementType
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
left_side | ::mlir::BoolAttr | bool attribute |
lower | ::mlir::BoolAttr | bool attribute |
unit_diagonal | ::mlir::BoolAttr | bool attribute |
transpose_a | ::mlir::mhlo::TransposeAttr | Transpose options |
Operands:
المعامل | وصف |
---|---|
a | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements values |
b | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements values |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or complex type with 32-bit float or 64-bit float elements values |
mhlo.tuple
(mhlo::TupleOp)
Tuple operation
بناء الجملة:
operation ::= `mhlo.tuple` $val attr-dict `:` custom<TupleOpType>(type($val), type($result))
Produces a result
tuple from values val
.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#tuple
مثال:
%result = mhlo.tuple %val0, %val1 : tuple<tensor<2xf32>, tuple<tensor<i32>>>
Traits: AlwaysSpeculatableImplTrait
Interfaces: ConditionallySpeculatable
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
val | variadic of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values or token or nested tuple with any combination of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values or ranked tensor of 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values or token values |
نتائج:
نتيجة | وصف |
---|---|
result | nested tuple with any combination of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values or ranked tensor of 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values or token values |
mhlo.unary_einsum
(mhlo::UnaryEinsumOp)
UnaryEinsum 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
مثال:
%result = "mhlo.unary_einsum"(%operand) {
einsum_config = "ab->a"
} : (tensor<4x16xf32>) -> tensor<4xf32>
Traits: AlwaysSpeculatableImplTrait
Interfaces: ConditionallySpeculatable
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
صفات:
يصف | MLIR Type | وصف |
---|---|---|
einsum_config | ::mlir::StringAttr | string attribute |
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.uniform_dequantize
(mhlo::UniformDequantizeOp)
UniformDequantize operation
بناء الجملة:
operation ::= `mhlo.uniform_dequantize` $operand attr-dict
`:` custom<SameOperandsAndResultType>(type($operand), type($result))
Performs element-wise conversion of quantized tensor operand
to a floating-point tensor result
according to the quantization parameters defined by the operand
type.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#uniform_dequantize
مثال:
%result = mhlo.uniform_dequantize %operand : (tensor<16x16x!quant.uniform<i8:f32, 34.0:16>>) -> tensor<16x16xf32>
Traits: AlwaysSpeculatableImplTrait
, Elementwise
, InferTensorType
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type values |
mhlo.uniform_quantize
(mhlo::UniformQuantizeOp)
UniformQuantize operation
بناء الجملة:
operation ::= `mhlo.uniform_quantize` $operand attr-dict
`:` custom<SameOperandsAndResultType>(type($operand), type($result))
Performs element-wise conversion of floating-point tensor or quantized tensor operand
to a quantized tensor result
according to the quantization parameters defined by the result
type.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#uniform_quantize
مثال:
%result = mhlo.uniform_quantize %operand : (tensor<16x16xf32>) -> tensor<16x16x!quant.uniform<ui8:f32, 34.0:16>>
Traits: AlwaysSpeculatableImplTrait
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
operand | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
mhlo.while
(mhlo::WhileOp)
While operation
Produces the output from executing body
function 0 or more times while the cond
function outputs true
.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#while
مثال:
%results0, %results1 = "mhlo.while"(%operand0, %operand1) ({
^bb0(%arg0: tensor<i32>, %arg1: tensor<i32>):
%0 = "mhlo.compare"(%arg0, %arg1) {
comparison_direction = #stablehlo<comparison_direction LT>
} : (tensor<i32>, tensor<i32>) -> tensor<i1>
"mhlo.return"(%0) : (tensor<i1>) -> ()
}, {
^bb0(%arg0: tensor<i32>, %arg1: tensor<i32>):
%0 = "mhlo.add"(%arg0, %constant0) : (tensor<i32>, tensor<i32>) -> tensor<i32>
"mhlo.return"(%0, %arg1) : (tensor<i32>, tensor<i32>) -> ()
}) : (tensor<i32>, tensor<i32>) -> (tensor<i32>, tensor<i32>)
Traits: RecursiveMemoryEffects
, SingleBlockImplicitTerminator<ReturnOp>
, SingleBlock
Interfaces: InferTypeOpInterface
, OpAsmOpInterface
Operands:
المعامل | وصف |
---|---|
operand | variadic of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values or ranked tensor of 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values or token |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | variadic of ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer values or ranked tensor of 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values or token |
mhlo.xla.rng_get_and_update_state
(mhlo::XlaRngGetAndUpdateStateOp)
XlaRngGetAndUpdateState operation
بناء الجملة:
operation ::= `mhlo.xla.rng_get_and_update_state` attr-dict
This operation is private to the XLA compiler, so it is does not yet have a specification.
Informally, this operation represents the change of the global random number generator state for rng instructions. The global state is incremented by delta and the old state is returned.
The output is currently defined for a single output type. If this changes in the future to support multiple types, lowering to use of a global memref must ensure that a single memref is still used and updated appropriately.
Interfaces: InferTypeOpInterface
صفات:
يصف | MLIR Type | وصف |
---|---|---|
delta | ::mlir::IntegerAttr | 64-bit signless integer attribute |
نتائج:
نتيجة | وصف |
---|---|
«unnamed» | statically shaped tensor of 64-bit unsigned integer values |
mhlo.xor
(mhlo::XorOp)
Xor operation
بناء الجملة:
operation ::= `mhlo.xor` $lhs `,` $rhs attr-dict
`:` custom<SameOperandsAndResultType>(type($lhs), type($rhs), type($result))
Performs element-wise XOR of two tensors lhs
and rhs
and produces a result
tensor.
See: https://github.com/openxla/stablehlo/blob/main/docs/spec.md#xor
مثال:
%result = mhlo.xor %lhs, %rhs : tensor<2xi32>
Traits: AlwaysSpeculatableImplTrait
, Commutative
, CompatibleOperandsAndResultType
, Elementwise
, SameOperandsAndResultShape
Interfaces: ConditionallySpeculatable
, InferShapedTypeOpInterface
, InferTypeOpInterface
, NoMemoryEffect (MemoryEffectOpInterface)
Effects: MemoryEffects::Effect{}
Operands:
المعامل | وصف |
---|---|
lhs | ranked tensor of pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
rhs | ranked tensor of pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer values |
نتائج:
نتيجة | وصف |
---|---|
result | ranked tensor of f8E4M3B11FNUZ type or f8E4M3FN type or f8E4M3FNUZ type or f8E5M2 type or f8E5M2FNUZ type or 16-bit float or 32-bit float or 64-bit float or bfloat16 type or pred (AKA boolean or 1-bit integer) or 4/8/16/32/64-bit signless integer or 4/8/16/32/64-bit unsigned integer or complex type with 32-bit float or 64-bit float elements or 4/8/16/32-bit uniform quantized signed integer or 4/8/16/32-bit uniform quantized unsigned integer or 4/8/16/32-bit uniform quantized per axis signed integer or 4/8/16/32-bit uniform quantized per axis unsigned integer values |
صفات
ArgResultAliasAttr
Attribute that models the alias relationship of entry function argument
This attribute captures the alias relationship of an MHLO main function argument to one of the results, denoted by resultIndex
. The argTupleIndices
and resultTupleIndices
are used to index into nested tuples in operand and result respectively. If isMustAlias
is true then the operand-result pair must alias.
This is meant to be used as an attribute on a function argument in MHLO. For example, in the following code it expresses that %arg1
may alias 0-th result.
func @main(%arg0: tensor<2xf32>, %arg1: tensor<3xf32> {mhlo.result_alias =
mhlo.result_alias<result_index = [2], ...>}
) -> tensor<2xf32>, tensor<3xf32> {
// function body ...
}
حدود:
معامل | C++ type | وصف |
---|---|---|
argTupleIndices | ::llvm::ArrayRef<int64_t> | البعد |
resultIndex | int64_t | |
resultTupleIndices | ::llvm::ArrayRef<int64_t> | البعد |
isMustAlias | bool |
ChannelHandleAttr
two 64-bit integers 'handle' and 'type'
بناء الجملة:
#mhlo.channel_handle<
int64_t, # handle
int64_t # type
>
حدود:
معامل | C++ type | وصف |
---|---|---|
مقبض | int64_t | |
يكتب | int64_t |
ComparisonDirectionAttr
Which comparison operation to perform.
بناء الجملة:
#mhlo.comparison_direction<
::mlir::mhlo::ComparisonDirection # value
>
Enum cases:
- EQ (
EQ
) - NE (
NE
) - GE (
GE
) - GT (
GT
) - LE (
LE
) - LT (
LT
) #### Parameters:
معامل | C++ type | وصف |
---|---|---|
قيمة | ::mlir::mhlo::ComparisonDirection | an enum of type ComparisonDirection |
ComparisonTypeAttr
Which comparison type to use.
بناء الجملة:
#mhlo.comparison_type<
::mlir::mhlo::ComparisonType # value
>
Enum cases:
- NOTYPE (
NOTYPE
) - FLOAT (
FLOAT
) - TOTALORDER (
TOTALORDER
) - SIGNED (
SIGNED
) - UNSIGNED (
UNSIGNED
) #### Parameters:
معامل | C++ type | وصف |
---|---|---|
قيمة | ::mlir::mhlo::ComparisonType | an enum of type ComparisonType |
ConvDimensionNumbersAttr
Structure of dimension information for conv op
حدود:
معامل | C++ type | وصف |
---|---|---|
inputBatchDimension | int64_t | |
inputFeatureDimension | int64_t | |
inputSpatialDimensions | ::llvm::ArrayRef<int64_t> | البعد |
kernelInputFeatureDimension | int64_t | |
kernelOutputFeatureDimension | int64_t | |
kernelSpatialDimensions | ::llvm::ArrayRef<int64_t> | البعد |
outputBatchDimension | int64_t | |
outputFeatureDimension | int64_t | |
outputSpatialDimensions | ::llvm::ArrayRef<int64_t> | البعد |
CrossProgramPrefetchAttr
Argument that is prefetched from another program
بناء الجملة:
#mhlo.cross_program_prefetch<
int64_t, # parameter
::llvm::ArrayRef<int64_t>, # indices
std::optional<int64_t> # offset
>
This attribute captures an argument that is prefetched from another program. For a given CrossProgramPrefetchAttr
, parameter
tells us which argument of the main
function of the module is prefetched, and indices
is a shape index telling us what subshape of that argument is prefetched.
A shape has a subshape iff it is a tuple. In that case, the subshape of the tuple by indices
is the shape achieved after indexing by each element of indices
in turn. For example, the [1,0] subshape of tuple<tuple<token, token>, tuple<tensor<i32>, token>>
is tensor<i32>
.
An empty value for indices
means the whole shape is prefetched.
على سبيل المثال،
module attributes { mhlo.cross_program_prefetch = [ #mhlo.cross_program_prefetch< parameter = 0, indices = [0]> ]} {
func.func @copy(%arg0 : tuple<tensor<2x3xi32>, tensor<i32>>) -> tuple<tensor<2x3xi32>, tensor<i32>> {
%0 = "mhlo.copy"(%arg0) {is_cross_program_prefetch}
return %0 : tuple<tensor<2x3xi32>, tensor<i32>>
}
func.func @main(%arg0 : tuple<tensor<2x3xi32>, tensor<i32>>) -> tuple<tensor<2x3xi32>, tensor<i32>> {
%1 = "mhlo.async_start"(%arg0) {called_computation=@copy}
%2 = "mhlo.async_done"(%1) {called_computation=@copy}
return %2 : tuple<tensor<2x3xi32>, tensor<i32>>
}
}
The parameter = 0
tells us that the async copy of the 0
th parameter is a cross_program_prefetch
, while the index
of [0]
tells us that the 0
th element of the tuple is prefetched while the other element of the tuple is not.
حدود:
معامل | C++ type | وصف |
---|---|---|
معامل | int64_t | |
المؤشرات | ::llvm::ArrayRef<int64_t> | البعد |
عوض | std::optional<int64_t> |
CustomCallScheduleAttr
Specifies the desired schedule for the custom-call.
بناء الجملة:
#mhlo.custom_call_schedule<
::mlir::mhlo::CustomCallSchedule # value
>
Enum cases:
- ليس
NONE
) - LATEST (
LATEST
) - EARLIEST (
EARLIEST
) #### Parameters:
معامل | C++ type | وصف |
---|---|---|
قيمة | ::mlir::mhlo::CustomCallSchedule | an enum of type CustomCallSchedule |
DequantizeModeAttr
Dequantization mode. Only MIN_COMBINED is supported.
بناء الجملة:
#mhlo.dequantize_mode<
::mlir::mhlo::DequantizeMode # value
>
Enum cases:
- MIN_COMBINED (
MIN_COMBINED
) #### Parameters:
معامل | C++ type | وصف |
---|---|---|
قيمة | ::mlir::mhlo::DequantizeMode | an enum of type DequantizeMode |
DomainKindAttr
Kind of domain metatdata attached to an HLO domain.
بناء الجملة:
#mhlo.kind<
::mlir::mhlo::DomainKind # value
>
Enum cases:
- sharding (
sharding
) #### Parameters:
معامل | C++ type | وصف |
---|---|---|
قيمة | ::mlir::mhlo::DomainKind | an enum of type DomainKind |
DotDimensionNumbersAttr
Attribute that models the dimension information for dot.
حدود:
معامل | C++ type | وصف |
---|---|---|
lhsBatchingDimensions | ::llvm::ArrayRef<int64_t> | البعد |
rhsBatchingDimensions | ::llvm::ArrayRef<int64_t> | البعد |
lhsContractingDimensions | ::llvm::ArrayRef<int64_t> | البعد |
rhsContractingDimensions | ::llvm::ArrayRef<int64_t> | البعد |
FftTypeAttr
XLA fast fourier transform type.
بناء الجملة:
#mhlo.fft_type<
::mlir::mhlo::FftType # value
>
Enum cases:
- FFT (
FFT
) - IFFT (
IFFT
) - RFFT (
RFFT
) - IRFFT (
IRFFT
) #### Parameters:
معامل | C++ type | وصف |
---|---|---|
قيمة | ::mlir::mhlo::FftType | an enum of type FftType |
FusionKindAttr
fusion kind
بناء الجملة:
#mhlo.fusion_kind<
::mlir::mhlo::FusionKind # value
>
Enum cases:
- kLoop (
kLoop
) - kInput (
kInput
) - kOutput (
kOutput
) - kCustom (
kCustom
) #### Parameters:
معامل | C++ type | وصف |
---|---|---|
قيمة | ::mlir::mhlo::FusionKind | an enum of type FusionKind |
GatherDimensionNumbersAttr
Attribute that models the dimension information for gather
حدود:
معامل | C++ type | وصف |
---|---|---|
offsetDims | ::llvm::ArrayRef<int64_t> | البعد |
collapsedSliceDims | ::llvm::ArrayRef<int64_t> | البعد |
startIndexMap | ::llvm::ArrayRef<int64_t> | البعد |
indexVectorDim | int64_t |
OutputOperandAliasAttr
Attribute that models the alias relationship of output and operand of a CustomCall op
بناء الجملة:
#mhlo.output_operand_alias<
::llvm::ArrayRef<int64_t>, # outputTupleIndices
int64_t, # operandIndex
::llvm::ArrayRef<int64_t> # operandTupleIndices
>
This attribute captures the alias relationship of the output to one of the operands for a CustomCall op, denoted by operand_index
. The output_tuple_indices
and operand_tuple_indices
are used to index into output and operand types. These indices lists are empty if the corresponding types are not tuple types, and can be arbitrarily long in case of arbitrarily nested tuple types.
See https://www.tensorflow.org/xla/aliasing
Example when used as array with in mhlo.custom-call:
%0 = "mhlo.custom_call"(%arg0, %arg1) {
// other attributes
output_operand_alias = [
#mhlo.output_operand_alias<output_tuple_indices = [0],
operand_index = 0,
operand_tuple_indices = [1]>
]
} : (tuple<tensor<1x1xf32>, tensor<2x3xf32>>, tensor<5x5xf32>) -> tuple<tensor<2x3xf32>>
The output and the 0th operand are both tuples. The aliasing shows the
relationship between the 0th element in output tuple with the 1st element in
the 0th operand. And both of them are of the same type: tensor<2x3xf32>.
حدود:
معامل | C++ type | وصف |
---|---|---|
outputTupleIndices | ::llvm::ArrayRef<int64_t> | البعد |
operandIndex | int64_t | |
operandTupleIndices | ::llvm::ArrayRef<int64_t> | البعد |
PrecisionAttr
XLA precision for an operand. Has backend specific meaning.
بناء الجملة:
#mhlo.precision<
::mlir::mhlo::Precision # value
>
Enum cases:
- DEFAULT (
DEFAULT
) - عالي
HIGH
) - HIGHEST (
HIGHEST
) - PACKED_NIBBLE (
PACKED_NIBBLE
) #### Parameters:
معامل | C++ type | وصف |
---|---|---|
قيمة | ::mlir::mhlo::Precision | an enum of type Precision |
RngAlgorithmAttr
XLA PRNG algorithm to be used.
بناء الجملة:
#mhlo.rng_algorithm<
::mlir::mhlo::RngAlgorithm # value
>
Enum cases:
- DEFAULT (
DEFAULT
) - THREE_FRY (
THREE_FRY
) - PHILOX (
PHILOX
) #### Parameters:
معامل | C++ type | وصف |
---|---|---|
قيمة | ::mlir::mhlo::RngAlgorithm | an enum of type RngAlgorithm |
RngDistributionAttr
XLA PRNG distribution to be used.
بناء الجملة:
#mhlo.rng_distribution<
::mlir::mhlo::RngDistribution # value
>
Enum cases:
- UNIFORM (
UNIFORM
) - NORMAL (
NORMAL
) #### Parameters:
معامل | C++ type | وصف |
---|---|---|
قيمة | ::mlir::mhlo::RngDistribution | an enum of type RngDistribution |
ScatterDimensionNumbersAttr
Attribute that models the dimension information for scatter
حدود:
معامل | C++ type | وصف |
---|---|---|
updateWindowDims | ::llvm::ArrayRef<int64_t> | البعد |
insertedWindowDims | ::llvm::ArrayRef<int64_t> | البعد |
scatterDimsToOperandDims | ::llvm::ArrayRef<int64_t> | البعد |
indexVectorDim | int64_t |
SparsityDescriptorAttr
Describes structured (N:M) sparsity configuration
بناء الجملة:
#mhlo.sparsity<
int64_t, # dimension
int64_t, # n
int64_t # m
>
This attribute is defined for a sparse dot operation with a structured sparse input tensor. With (N=2,M=4), every 4 consecutive logical elements have exactly 2 non-zero physical elements in the input tensor.
$dimension defines the index of the contracting dimension that is sparse (it has to be the most minor dimension). The additional metadata operand in the sparse dot operation defines which logical elements are zeroed out.
حدود:
معامل | C++ type | وصف |
---|---|---|
البعد | int64_t | |
ن | int64_t | |
م | int64_t |
TransposeAttr
Transpose options
بناء الجملة:
#mhlo.transpose<
::mlir::mhlo::Transpose # value
>
Enum cases:
- TRANSPOSE_INVALID (
TRANSPOSE_INVALID
) - NO_TRANSPOSE (
NO_TRANSPOSE
) - TRANSPOSE (
TRANSPOSE
) - ADJOINT (
ADJOINT
) #### Parameters:
معامل | C++ type | وصف |
---|---|---|
قيمة | ::mlir::mhlo::Transpose | an enum of type Transpose |
TypeExtensionsAttr
Attribute that extends tensor type with MHLO type properties.
بناء الجملة:
#mhlo.type_extensions<
::llvm::ArrayRef<int64_t> # bounds
>
This attribute is used to extend MLIR tensor type with MHLO tensor specific properties. These properties aren't modeled in the MLIR type. This attribute is set in the encoding
field of the tensor type.
See HLO_BoundedAttrInterface
for documentation for bounds
.
حدود:
معامل | C++ type | وصف |
---|---|---|
الحدود | ::llvm::ArrayRef<int64_t> |
أنواع
AsyncBundleType
Opaque collection of other types
بناء الجملة:
!mhlo.async_bundle<
::llvm::ArrayRef<Type> # types
>
حدود:
معامل | C++ type | وصف |
---|---|---|
أنواع | ::llvm::ArrayRef<Type> |