운영
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 유형 또는 16비트 부동 소수점 또는 32비트 부동 소수점 또는 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 유형 또는 16비트 부동 소수점 또는 32비트 부동 소수점 또는 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)
AddDependency 작업
통사론:
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)
AfterAll 작업
통사론:
operation ::= `mhlo.after_all` $inputs attr-dict
`:` custom<VariadicSameOperandsAndResultType>(ref($inputs), type($inputs), type($result))
result
에 의존하는 작업보다 먼저 inputs
생성하는 작업이 실행되도록 합니다.
참조: 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
속성:
기인하다 | MLIR 유형 | 설명 |
---|---|---|
all_gather_dim | ::mlir::정수 속성 | 64비트 부호 없는 정수 속성 |
replica_groups | ::mlir::DenseIntElementsAttr | 64비트 부호 없는 정수 요소 속성 |
channel_handle | ::mlir::mhlo::ChannelHandleAttr | 두 개의 64비트 정수 'handle' 및 'type' |
use_global_device_ids | ::mlir::단위 속성 | 단위 속성 |
피연산자:
피연산자 | 설명 |
---|---|
operands | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 float 또는 32비트 float 또는 64비트 float 또는 bfloat16 유형 또는 pred(AKA 부울 또는 1비트 정수) 또는 4의 순위 텐서의 variadic /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비트 float 또는 32비트 float 또는 64비트 float 또는 bfloat16 유형 또는 pred(AKA 부울 또는 1비트 정수) 또는 4의 순위 텐서의 variadic /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)
AllReduce 작업
프로세스 그리드의 각 프로세스 그룹 내에서 각 프로세스의 피연산자 텐서 값에 축소 함수 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
속성:
기인하다 | MLIR 유형 | 설명 |
---|---|---|
replica_groups | ::mlir::DenseIntElementsAttr | 64비트 부호 없는 정수 요소 속성 |
channel_handle | ::mlir::mhlo::ChannelHandleAttr | 두 개의 64비트 정수 'handle' 및 'type' |
use_global_device_ids | ::mlir::단위 속성 | 단위 속성 |
피연산자:
피연산자 | 설명 |
---|---|
operands | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 float 또는 32비트 float 또는 64비트 float 또는 bfloat16 유형 또는 pred(AKA 부울 또는 1비트 정수) 또는 4의 순위 텐서의 variadic /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비트 float 또는 32비트 float 또는 64비트 float 또는 bfloat16 유형 또는 pred(AKA 부울 또는 1비트 정수) 또는 4의 순위 텐서의 variadic /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)
AllToAll 작업
프로세스 그리드의 각 프로세스 그룹 내에서 split_dimension
따라 operand
텐서의 값을 여러 부분으로 분할하고, 분할된 부분을 프로세스 간에 분산시키고, 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{}
속성:
기인하다 | MLIR 유형 | 설명 |
---|---|---|
split_dimension | ::mlir::정수 속성 | 64비트 부호 없는 정수 속성 |
concat_dimension | ::mlir::정수 속성 | 64비트 부호 없는 정수 속성 |
split_count | ::mlir::정수 속성 | 64비트 부호 없는 정수 속성 |
replica_groups | ::mlir::DenseIntElementsAttr | 64비트 부호 없는 정수 요소 속성 |
channel_handle | ::mlir::mhlo::ChannelHandleAttr | 두 개의 64비트 정수 'handle' 및 'type' |
피연산자:
피연산자 | 설명 |
---|---|
operand | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 float 또는 32비트 float 또는 64비트 float 또는 bfloat16 유형 또는 pred(AKA 부울 또는 1비트 정수) 또는 4의 순위 텐서의 variadic /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비트 float 또는 32비트 float 또는 64비트 float 또는 bfloat16 유형 또는 pred(AKA 부울 또는 1비트 정수) 또는 4의 순위 텐서의 variadic /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
의 요소별 AND를 수행하고 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)
AsyncDone 작업
이 작업은 XLA 컴파일러 전용이므로 아직 사양이 없습니다.
비공식적으로 이 작업은 비동기 계산이 끝날 때까지 차단됩니다. 비동기 계산의 최종 결과를 반환합니다.
자세한 내용은 AsyncStart 설명서를 참조하세요.
인터페이스: InferTypeOpInterface
속성:
기인하다 | MLIR 유형 | 설명 |
---|---|---|
called_computation | ::mlir::FlatSymbolRefAttr | 플랫 기호 참조 속성 |
execution_thread | ::mlir::문자열 속성 | 문자열 속성 |
피연산자:
피연산자 | 설명 |
---|---|
bundle | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 float 또는 32비트 float 또는 64비트 float 또는 bfloat16 유형 또는 pred(AKA 부울 또는 1비트 정수)의 순위 텐서 조합이 포함된 async_bundle ) 또는 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비트 float 또는 32비트 float 또는 64비트 float 또는 bfloat16 유형 또는 pred(AKA 부울 또는 1비트 정수) 또는 4의 순위 텐서의 variadic /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비트 float 또는 32비트 float 또는 64비트 float 또는 bfloat16 유형 또는 pred(부울이라고도 함)의 순위 텐서 조합이 포함된 값 또는 토큰 또는 중첩 튜플 또는 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 컴파일러 전용이므로 아직 사양이 없습니다.
비공식적으로 이 작업은 비동기 계산을 시작합니다.
이는 비동기 대기(예: DMA)와 스레드 내 계산을 모두 포함하는 함수가 있을 때 사용됩니다. 예를 들어 함수는 계산, DMA, 다른 계산, 두 번째 DMA 및 최종 계산으로 구성될 수 있습니다. 이는 async_start, async_update 및 async_done으로 표시됩니다. async_start는 스레드에서 첫 번째 계산을 수행한 다음 DMA를 시작합니다. async_update는 DMA가 아직 완료되지 않은 경우 완료될 때까지 기다린 다음 함수에서 두 번째 계산을 실행하고 두 번째 DMA를 시작합니다. 마지막으로 async_done은 이 마지막 DMA를 기다린 다음 스레드에서 실행해야 하는 마지막 계산을 실행하고 해당 최종 계산의 결과를 반환합니다.
operands
called_computation
계산에 직접 전달됩니다._계산은 비동기적으로 실행될 함수입니다. execution_thread
는 실행될 스레드의 이름입니다. 메인 스레드를 "main"이라고 합니다. 모든 스레드에는 이름이 있습니다.
이는 비동기 작업 간에 필요한 모든 상태를 반환합니다. 버퍼 할당 후 반환 값은 입력, 결과 및 비동기 작업에 필요하거나 편집된 스크래치패드를 보관하는 데 필요한 공간을 나타냅니다.
속성:
기인하다 | MLIR 유형 | 설명 |
---|---|---|
called_computation | ::mlir::FlatSymbolRefAttr | 플랫 기호 참조 속성 |
execution_thread | ::mlir::문자열 속성 | 문자열 속성 |
피연산자:
피연산자 | 설명 |
---|---|
inputs | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 float 또는 32비트 float 또는 64비트 float 또는 bfloat16 유형 또는 pred(AKA 부울 또는 1비트 정수) 또는 4의 순위 텐서의 variadic /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비트 float 또는 32비트 float 또는 64비트 float 또는 bfloat16 유형 또는 pred(부울이라고도 함)의 순위 텐서 조합이 포함된 값 또는 토큰 또는 중첩 튜플 또는 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비트 float 또는 32비트 float 또는 64비트 float 또는 bfloat16 유형 또는 pred(AKA 부울 또는 1비트 정수)의 순위 텐서 조합이 포함된 async_bundle ) 또는 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
속성:
기인하다 | MLIR 유형 | 설명 |
---|---|---|
called_computation | ::mlir::FlatSymbolRefAttr | 플랫 기호 참조 속성 |
execution_thread | ::mlir::문자열 속성 | 문자열 속성 |
피연산자:
피연산자 | 설명 |
---|---|
bundle | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 float 또는 32비트 float 또는 64비트 float 또는 bfloat16 유형 또는 pred(AKA 부울 또는 1비트 정수)의 순위 텐서 조합이 포함된 async_bundle ) 또는 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비트 float 또는 32비트 float 또는 64비트 float 또는 bfloat16 유형 또는 pred(AKA 부울 또는 1비트 정수)의 순위 텐서 조합이 포함된 async_bundle ) 또는 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)
Atan2 작업
통사론:
operation ::= `mhlo.atan2` $lhs `,` $rhs attr-dict
`:` custom<SameOperandsAndResultType>(type($lhs), type($rhs), type($result))
lhs
및 rhs
텐서에 대해 요소별 atan2 작업을 수행하고 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 작업
grad_output
에서 역전파되는 BatchNormTrainingOp의 여러 입력에 대한 기울기를 계산하고 grad_operand
, grad_scale
및 grad_offset
텐서를 생성합니다.
참조: 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{}
속성:
기인하다 | MLIR 유형 | 설명 |
---|---|---|
epsilon | ::mlir::FloatAttr | 32비트 부동 소수점 속성 |
feature_index | ::mlir::정수 속성 | 64비트 부호 없는 정수 속성 |
피연산자:
피연산자 | 설명 |
---|---|
operand | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 부동 소수점 또는 32비트 부동 소수점 또는 64비트 부동 소수점 또는 bfloat16 유형 값의 순위 텐서 |
scale | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 부동 또는 32비트 부동 또는 64비트 부동 또는 bfloat16 유형 값의 1D 텐서 |
mean | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 부동 또는 32비트 부동 또는 64비트 부동 또는 bfloat16 유형 값의 1D 텐서 |
variance | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 부동 또는 32비트 부동 또는 64비트 부동 또는 bfloat16 유형 값의 1D 텐서 |
grad_output | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 부동 소수점 또는 32비트 부동 소수점 또는 64비트 부동 소수점 또는 bfloat16 유형 값의 순위 텐서 |
결과:
결과 | 설명 |
---|---|
grad_operand | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 부동 소수점 또는 32비트 부동 소수점 또는 64비트 부동 소수점 또는 bfloat16 유형 값의 순위 텐서 |
grad_scale | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 부동 또는 32비트 부동 또는 64비트 부동 또는 bfloat16 유형 값의 1D 텐서 |
grad_offset | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 부동 또는 32비트 부동 또는 64비트 부동 또는 bfloat16 유형 값의 1D 텐서 |
mhlo.batch_norm_inference
(mhlo::BatchNormInferenceOp)
BatchNormInference 작업
feature_index
차원을 제외한 모든 차원에서 operand
텐서를 정규화하고 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{}
속성:
기인하다 | MLIR 유형 | 설명 |
---|---|---|
epsilon | ::mlir::FloatAttr | 32비트 부동 소수점 속성 |
feature_index | ::mlir::정수 속성 | 64비트 부호 없는 정수 속성 |
피연산자:
피연산자 | 설명 |
---|---|
operand | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 부동 소수점 또는 32비트 부동 소수점 또는 64비트 부동 소수점 또는 bfloat16 유형 값의 순위 텐서 |
scale | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 부동 또는 32비트 부동 또는 64비트 부동 또는 bfloat16 유형 값의 1D 텐서 |
offset | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 부동 또는 32비트 부동 또는 64비트 부동 또는 bfloat16 유형 값의 1D 텐서 |
mean | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 부동 또는 32비트 부동 또는 64비트 부동 또는 bfloat16 유형 값의 1D 텐서 |
variance | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 부동 또는 32비트 부동 또는 64비트 부동 또는 bfloat16 유형 값의 1D 텐서 |
결과:
결과 | 설명 |
---|---|
result | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 부동 소수점 또는 32비트 부동 소수점 또는 64비트 부동 소수점 또는 bfloat16 유형 값의 순위 텐서 |
mhlo.batch_norm_training
(mhlo::BatchNormTrainingOp)
BatchNorm훈련 작업
배치 및 공간 차원 전체에서 평균과 분산을 계산하고 feature_index
차원의 각 기능에 대해 operand
텐서를 정규화하고 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{}
속성:
기인하다 | MLIR 유형 | 설명 |
---|---|---|
epsilon | ::mlir::FloatAttr | 32비트 부동 소수점 속성 |
feature_index | ::mlir::정수 속성 | 64비트 부호 없는 정수 속성 |
피연산자:
피연산자 | 설명 |
---|---|
operand | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 부동 소수점 또는 32비트 부동 소수점 또는 64비트 부동 소수점 또는 bfloat16 유형 값의 순위 텐서 |
scale | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 부동 또는 32비트 부동 또는 64비트 부동 또는 bfloat16 유형 값의 1D 텐서 |
offset | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 부동 또는 32비트 부동 또는 64비트 부동 또는 bfloat16 유형 값의 1D 텐서 |
결과:
결과 | 설명 |
---|---|
output | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 부동 소수점 또는 32비트 부동 소수점 또는 64비트 부동 소수점 또는 bfloat16 유형 값의 순위 텐서 |
batch_mean | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 부동 또는 32비트 부동 또는 64비트 부동 또는 bfloat16 유형 값의 1D 텐서 |
batch_var | f8E4M3B11FNUZ 유형 또는 f8E4M3FN 유형 또는 f8E4M3FNUZ 유형 또는 f8E5M2 유형 또는 f8E5M2FNUZ 유형 또는 16비트 부동 또는 32비트 부동 또는 64비트 부동 또는 bfloat16 유형 값의 1D 텐서 |
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/32비트 균일 양자화 부호 없는 정수 또는 4/8/16/32비트 축당 균일 양자화 부호 있는 정수 또는 4/8/16/32비트 축당 균일 양자화 부호 없는 정수 값 |
mhlo.broadcast
(mhlo :: broadcastop)
방송 작업
이 작업은 Scablehlo에서 벗어나기 때문에 사양에 포함되지 않습니다 : 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 :: denseintelementsattr | 64 비트 부호없는 정수 요소 속성 |
피연산자 :
피연산자 | 설명 |
---|---|
operand | F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 BFLOAT16 유형 (일명 부울 또는 1- 볼트 링거) 또는 4/8 /16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4/8/16/32 비트 균일 한 균일 한 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 서명 된 정수 또는 4/8/16/32 비트 균일 한 정량화되지 않은 정수 또는 4/8/16/32 비트 균일 한 축당 정수 정수 또는 4/8/16/32 비트 축에 양자화 된 균일 한 균일 한 정수 값. |
결과:
결과 | 설명 |
---|---|
«이름이없는» | F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 BFLOAT16 유형 (일명 부울 또는 1- 볼트 링거) 또는 4/8 /16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4/8/16/32 비트 균일 한 균일 한 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 서명 된 정수 또는 4/8/16/32 비트 균일 한 정량화되지 않은 정수 또는 4/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 :: denseintelementsattr | 64 비트 부호없는 정수 요소 속성 |
피연산자 :
피연산자 | 설명 |
---|---|
operand | F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 BFLOAT16 유형 (일명 부울 또는 1- 볼트 링거) 또는 4/8 /16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4/8/16/32 비트 균일 한 균일 한 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 서명 된 정수 또는 4/8/16/32 비트 균일 한 정량화되지 않은 정수 또는 4/8/16/32 비트 균일 한 축당 정수 정수 또는 4/8/16/32 비트 축에 양자화 된 균일 한 균일 한 정수 값. |
결과:
결과 | 설명 |
---|---|
«이름이없는» | F8E4M3B11FNUZ 유형의 정적으로 형성 된 텐서 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 BFL16 유형 (대기업 부울 또는 1- 보울기) 또는 4/ 4// 4// 4/ BIT Inteor Pred. 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.case
(mhlo :: caseop)
케이스 작동
index
값에 따라 branches
에서 정확히 하나의 function
실행하여 출력을 생성합니다.
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>)
TRACITS : RecursiveMemoryEffects
, SingleBlockImplicitTerminator<ReturnOp>
, SingleBlock
인터페이스 : InferTypeOpInterface
피연산자 :
피연산자 | 설명 |
---|---|
index | 32 비트 부호없는 정수 값의 텐서 |
결과:
결과 | 설명 |
---|---|
«이름이없는» | f8e4m3b11fnuz 유형 또는 f8e4m3fn 유형 또는 f8e4m3fnuz 유형의 순위 텐서의 변수 또는 f8e5m2 유형 또는 f8e5m2fnuz 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 bfloat16 유형 (대기인 Bolean 또는 1 bit inteer inteege inteege integeer prede) /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-bit 균일 균일 한 균일 한 균일 한 축당 서명되지 않은 정수 값 또는 토큰 |
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 | F8E4M3B11FNUZ 유형의 순위 텐서 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 32 비트 플로트 또는 64- 비트 플로트 요소를 사용한 64 비트 플로트 또는 BFLOAT16 유형 또는 복잡한 유형의 텐서 4/8/16/32 비트 균일 양자화 된 서명 정수 또는 4/8/16/32 비트 양자화되지 않은 정수 값 |
결과:
결과 | 설명 |
---|---|
result | F8E4M3B11FNUZ 유형의 순위 텐서 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 32 비트 플로트 또는 64- 비트 플로트 요소를 사용한 64 비트 플로트 또는 BFLOAT16 유형 또는 복잡한 유형의 텐서 4/8/16/32 비트 균일 양자화 된 서명 정수 또는 4/8/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 | F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 BFLOAT16 유형의 순위 텐서 |
결과:
결과 | 설명 |
---|---|
result | F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 BFLOAT16 유형의 순위 텐서 |
mhlo.cholesky
(mhlo :: Choleskyop)
Cholesky Operation
매트릭스 배치의 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 | 부리 속성 |
피연산자 :
피연산자 | 설명 |
---|---|
a | F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 32 비트 플로트 또는 64- 비트 플로트 요소의 복잡한 유형의 순위 텐서 |
결과:
결과 | 설명 |
---|---|
«이름이없는» | F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 32 비트 플로트 또는 64- 비트 플로트 요소의 복잡한 유형의 순위 텐서 |
mhlo.clamp
(mhlo :: 클램프)
클램프 작동
통사론:
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 | F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 BFLOAT16 유형 (일명 부울 또는 1- 볼트 링거) 또는 4/8 /16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4/8/16/32 비트 균일 한 균일 한 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 서명 된 정수 또는 4/8/16/32 비트 균일 한 정량화되지 않은 정수 또는 4/8/16/32 비트 균일 한 축당 정수 정수 또는 4/8/16/32 비트 축에 양자화 된 균일 한 균일 한 정수 값. |
operand | F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 BFLOAT16 유형 (일명 부울 또는 1- 볼트 링거) 또는 4/8 /16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4/8/16/32 비트 균일 한 균일 한 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 서명 된 정수 또는 4/8/16/32 비트 균일 한 정량화되지 않은 정수 또는 4/8/16/32 비트 균일 한 축당 정수 정수 또는 4/8/16/32 비트 축에 양자화 된 균일 한 균일 한 정수 값. |
max | F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 BFLOAT16 유형 (일명 부울 또는 1- 볼트 링거) 또는 4/8 /16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4/8/16/32 비트 균일 한 균일 한 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 서명 된 정수 또는 4/8/16/32 비트 균일 한 정량화되지 않은 정수 또는 4/8/16/32 비트 균일 한 축당 정수 정수 또는 4/8/16/32 비트 축에 양자화 된 균일 한 균일 한 정수 값. |
결과:
결과 | 설명 |
---|---|
result | F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 BFLOAT16 유형 (일명 부울 또는 1- 볼트 링거) 또는 4/8 /16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4/8/16/32 비트 균일 한 균일 한 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 서명 된 정수 또는 4/8/16/32 비트 균일 한 정량화되지 않은 정수 또는 4/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
인터페이스 : InferShapedTypeOpInterface
, InferTypeOpInterface
속성:
기인하다 | mlir 유형 | 설명 |
---|---|---|
replica_groups | :: mlir :: denseintelementsattr | 64 비트 부호없는 정수 요소 속성 |
channel_handle | :: mlir :: mhlo :: ChannelHandleAttr | 2 개의 64 비트 정수 '핸들'과 '타입' |
피연산자 :
피연산자 | 설명 |
---|---|
operand | F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 BFLOAT16 유형 (일명 부울 또는 1- 볼트 링거) 또는 4/8 /16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4/8/16/32 비트 균일 한 균일 한 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 서명 된 정수 또는 4/8/16/32 비트 균일 한 정량화되지 않은 정수 또는 4/8/16/32 비트 균일 한 축당 정수 정수 또는 4/8/16/32 비트 축에 양자화 된 균일 한 균일 한 정수 값. |
결과:
결과 | 설명 |
---|---|
«이름이없는» | F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 BFLOAT16 유형 (일명 부울 또는 1- 볼트 링거) 또는 4/8 /16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4/8/16/32 비트 균일 한 균일 한 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 서명 된 정수 또는 4/8/16/32 비트 균일 한 정량화되지 않은 정수 또는 4/8/16/32 비트 균일 한 축당 정수 정수 또는 4/8/16/32 비트 축에 양자화 된 균일 한 균일 한 정수 값. |
mhlo.collective_permute
(mhlo :: CollectivePermuteop)
집단 관리 운영
프로세스 그리드의 각 프로세스 그룹 내에서, 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 :: denseintelementsattr | 64 비트 부호없는 정수 요소 속성 |
channel_handle | :: mlir :: mhlo :: ChannelHandleAttr | 2 개의 64 비트 정수 '핸들'과 '타입' |
피연산자 :
피연산자 | 설명 |
---|---|
operand | F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 BFLOAT16 유형 (일명 부울 또는 1- 볼트 링거) 또는 4/8 /16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4/8/16/32 비트 균일 한 균일 한 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 서명 된 정수 또는 4/8/16/32 비트 균일 한 정량화되지 않은 정수 또는 4/8/16/32 비트 균일 한 축당 정수 정수 또는 4/8/16/32 비트 축에 양자화 된 균일 한 균일 한 정수 값. |
결과:
결과 | 설명 |
---|---|
«이름이없는» | F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 BFLOAT16 유형 (일명 부울 또는 1- 볼트 링거) 또는 4/8 /16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4/8/16/32 비트 균일 한 균일 한 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 서명 된 정수 또는 4/8/16/32 비트 균일 한 정량화되지 않은 정수 또는 4/8/16/32 비트 균일 한 축당 정수 정수 또는 4/8/16/32 비트 축에 양자화 된 균일 한 균일 한 정수 값. |
mhlo.compare
(mhlo :: compraleop)
작업을 비교하십시오
통사론:
operation ::= `mhlo.compare` $comparison_direction `,` $lhs `,` $rhs (`,` $compare_type^)?
attr-dict `:` functional-type(operands, results)
comparison_direction
및 compare_type
에 따라 lhs
및 rhs
텐서의 요소 별 비교를 수행하고 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 | F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 BFLOAT16 유형 (일명 부울 또는 1- 볼트 링거) 또는 4/8 /16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4/8/16/32 비트 균일 한 균일 한 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 서명 된 정수 또는 4/8/16/32 비트 균일 한 정량화되지 않은 정수 또는 4/8/16/32 비트 균일 한 축당 정수 정수 또는 4/8/16/32 비트 축에 양자화 된 균일 한 균일 한 정수 값. |
rhs | F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 BFLOAT16 유형 (일명 부울 또는 1- 볼트 링거) 또는 4/8 /16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4/8/16/32 비트 균일 한 균일 한 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 서명 된 정수 또는 4/8/16/32 비트 균일 한 정량화되지 않은 정수 또는 4/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
속성에 의해 구현됩니다. composite
OP는 프로그램 의미를 변경하지 않고 분해로 대체 할 수 있습니다. 분해를 인라인으로 만드는 것이 동일한 OP 시맨틱을 제공하지 않는 경우 custom_call
사용하는 것이 좋습니다.
version
필드 (기본값 0에서 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 :: Dictionaryattr | 지명 된 속성 값의 사전 |
decomposition | :: mlir :: flatsymbolrefattr | 평평한 기호 참조 속성 |
version | :: mlir :: integerattr | 32 비트 부호없는 정수 속성 |
피연산자 :
피연산자 | 설명 |
---|---|
inputs | f8e4m3b11fnuz 유형 또는 f8e4m3fn 유형 또는 f8e4m3fnuz 유형의 순위 텐서의 변수 또는 f8e5m2 유형 또는 f8e5m2fnuz 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 bfloat16 유형 (대기인 Bolean 또는 1 bit inteer inteege inteege integeer prede) /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- 비트 플로트 또는 BFLOAT16 유형 (AKA BOOLEEN) 또는 1 비트 정수) 또는 4/8/16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4가있는 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 /8/16/32 비트 균일 한 양자 양자화 된 정수 또는 4/8/16/32 비트 균일 한 정렬되지 않은 정수 값 또는 4/8/16/32 비트 균일 한 균일 한 균일 한 균일 한 균일 한 균일 한 균일 한 균일 한 균일 한 균일 한 균일 한 정수 또는 4/8 /16/32 비트 균일 한 축당 정량화되지 않은 정수 값 또는 토큰 값 |
결과:
결과 | 설명 |
---|---|
«이름이없는» | f8e4m3b11fnuz 유형 또는 f8e4m3fn 유형 또는 f8e4m3fnuz 유형의 순위 텐서의 변수 또는 f8e5m2 유형 또는 f8e5m2fnuz 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 bfloat16 유형 (대기인 Bolean 또는 1 bit inteer inteege inteege integeer prede) /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- 비트 플로트 또는 BFLOAT16 유형 (AKA BOOLEEN) 또는 1 비트 정수) 또는 4/8/16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4가있는 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 /8/16/32 비트 균일 한 양자 양자화 된 정수 또는 4/8/16/32 비트 균일 한 정렬되지 않은 정수 값 또는 4/8/16/32 비트 균일 한 균일 한 균일 한 균일 한 균일 한 균일 한 균일 한 균일 한 균일 한 균일 한 균일 한 정수 또는 4/8 /16/32 비트 균일 한 축당 정량화되지 않은 정수 값 또는 토큰 값 |
mhlo.compute_reshape_shape
(mhlo :: computershapheshapeop)
컴퓨터 샤핑 운영
통사론:
operation ::= `mhlo.compute_reshape_shape` operands attr-dict `:` functional-type(operands, results)
이 작업은 진행중인 작업이므로 아직 사양에 포함되어 있지 않습니다 : https://github.com/openxla/stablehlo/issues/8
비공식적 으로이 작업은 DynamicReshapeop의 오페라에서 num_elements
수의 요소 수와 TF의 reshape에 제공되는 dynamic_shape
모양의 요소 수에 대한 output_shape를 계산합니다 : https://www.tensorflow.org/api_docs/python/tf/reshape
예를 들어, num_elements = 12
및 dynamic_shape = [2, -1]
의 경우 result
는 [2, 6]
가 될 것입니다. 피연산자가 유효하지 않은 경우 (예 : 치수가 요소 수를 고르게 나누지 않거나 차원에 여러 값이있는 경우) 정의되지 않은 동작으로 이어집니다.
예:
%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 텐서 |
결과:
결과 | 설명 |
---|---|
result | 정수 또는 인덱스 값의 1D 텐서 |
mhlo.concatenate
(mhlo :: concatenateop)
연결 작업
주어진 인수와 동일한 순서로 dimension
차원을 따라 inputs
에서 다양한 수의 텐서를 연결하고 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 | f8e4m3b11fnuz 유형 또는 f8e4m3fn 유형 또는 f8e4m3fnuz 유형의 순위 텐서의 변수 또는 f8e5m2 유형 또는 f8e5m2fnuz 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 bfloat16 유형 (대기인 Bolean 또는 1 bit inteer inteege inteege integeer prede) /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 유형 (일명 부울 또는 1- 볼트 링거) 또는 4/8 /16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4/8/16/32 비트 균일 한 균일 한 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 서명 된 정수 또는 4/8/16/32 비트 균일 한 정량화되지 않은 정수 또는 4/8/16/32 비트 균일 한 축당 정수 정수 또는 4/8/16/32 비트 축에 양자화 된 균일 한 균일 한 정수 값. |
mhlo.constant
(mhlo :: constantop)
일정한 작동
일정한 value
에서 output
텐서를 생성합니다.
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 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 BFL16 유형 (대기업 부울 또는 1- 보울기) 또는 4/ 4// 4// 4/ BIT Inteor Pred. 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.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 | F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 BFLOAT16 유형 (일명 부울 또는 1- 볼트 링거) 또는 4/8 /16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4/8/16/32 비트 균일 한 균일 한 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 서명 된 정수 또는 4/8/16/32 비트 균일 한 정량화되지 않은 정수 또는 4/8/16/32 비트 균일 한 축당 정수 정수 또는 4/8/16/32 비트 축에 양자화 된 균일 한 균일 한 정수 값. |
결과:
결과 | 설명 |
---|---|
result | F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 BFLOAT16 유형 (일명 부울 또는 1- 볼트 링거) 또는 4/8 /16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4/8/16/32 비트 균일 한 균일 한 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 서명 된 정수 또는 4/8/16/32 비트 균일 한 정량화되지 않은 정수 또는 4/8/16/32 비트 균일 한 축당 정수 정수 또는 4/8/16/32 비트 축에 양자화 된 균일 한 균일 한 정수 값. |
mhlo.convolution
(mhlo :: Convolutionop)
컨볼 루션 작업
통사론:
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)
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 :: denseintelementsattr | 64 비트 부호없는 정수 요소 속성 |
padding | :: mlir :: denseintelementsattr | 64 비트 부호없는 정수 요소 속성 |
lhs_dilation | :: mlir :: denseintelementsattr | 64 비트 부호없는 정수 요소 속성 |
rhs_dilation | :: mlir :: denseintelementsattr | 64 비트 부호없는 정수 요소 속성 |
window_reversal | :: mlir :: DenseElementsattr | 일정한 부울 벡터/텐서 속성 |
dimension_numbers | :: mlir :: mhlo :: convdimensionnumbersattr | CONV OP를위한 치수 정보의 구조 |
feature_group_count | :: mlir :: integerattr | 64 비트 부호없는 정수 속성 |
batch_group_count | :: mlir :: integerattr | 64 비트 부호없는 정수 속성 |
precision_config | :: mlir :: arrayattr | 정밀 구성 속성 |
피연산자 :
피연산자 | 설명 |
---|---|
lhs | F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 BFLOAT16 유형 (일명 부울 또는 1- 볼트 링거) 또는 4/8 /16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4/8/16/32 비트 균일 한 균일 한 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 서명 된 정수 또는 4/8/16/32 비트 균일 한 정량화되지 않은 정수 또는 4/8/16/32 비트 균일 한 축당 정수 정수 또는 4/8/16/32 비트 축에 양자화 된 균일 한 균일 한 정수 값. |
rhs | F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 BFLOAT16 유형 (일명 부울 또는 1- 볼트 링거) 또는 4/8 /16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4/8/16/32 비트 균일 한 균일 한 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 서명 된 정수 또는 4/8/16/32 비트 균일 한 정량화되지 않은 정수 또는 4/8/16/32 비트 균일 한 축당 정수 정수 또는 4/8/16/32 비트 축에 양자화 된 균일 한 균일 한 정수 값. |
결과:
결과 | 설명 |
---|---|
«이름이없는» | F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 BFLOAT16 유형 (일명 부울 또는 1- 볼트 링거) 또는 4/8 /16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4/8/16/32 비트 균일 한 균일 한 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 서명 된 정수 또는 4/8/16/32 비트 균일 한 정량화되지 않은 정수 또는 4/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
사본입니다. 작업에 부착 된 메타 데이터에 따라 NO-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 | F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 BFLOAT16 유형 (일명 부울 또는 1- 볼트 링거) 또는 4/8 /16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4/8/16/32 비트 균일 한 균일 한 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 서명 된 정수 또는 4/8/16/32 비트 균일 한 정량화되지 않은 정수 또는 4/8/16/32 비트 균일 한 축당 정수 또는 4/8/16/32 비트 균일 한 축당 정렬되지 않은 정수 값 또는 F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 64-Bit Float 16 Type (AKA BOOLE 또는 PRED)의 랭킹 텐서 조합과 함께 토큰 또는 중첩 튜플 -Bit Integer) 또는 4/8/16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4/8을 갖춘 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 /16/32 비트 균일 한 양자 양자화 된 정수 또는 4/8/16/32 비트 균일 한 양자 정렬되지 않은 정수 값 또는 4/8/16/32 비트 균일 한 균일 한 균형 잡힌 정수 또는 4/8/16 /32 비트 균일 한 축당 정량화되지 않은 정수 값 또는 토큰 값 |
결과:
결과 | 설명 |
---|---|
result | F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 32 비트 플로트 또는 64 비트 플로트 또는 BFLOAT16 유형 (일명 부울 또는 1- 볼트 링거) 또는 4/8 /16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4/8/16/32 비트 균일 한 균일 한 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 서명 된 정수 또는 4/8/16/32 비트 균일 한 정량화되지 않은 정수 또는 4/8/16/32 비트 균일 한 축당 정수 또는 4/8/16/32 비트 균일 한 축당 정렬되지 않은 정수 값 또는 F8E4M3B11FNUZ 유형 또는 F8E4M3FN 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2 유형 또는 F8E5M2FNUZ 유형 또는 F8E4M3FNUZ 유형 또는 F8E5M2FNUZ 유형 또는 16 비트 플로트 또는 64-Bit Float 16 Type (AKA BOOLE 또는 PRED)의 랭킹 텐서 조합과 함께 토큰 또는 중첩 튜플 -Bit Integer) 또는 4/8/16/32/64 비트 부호없는 정수 또는 32 비트 플로트 또는 64 비트 플로트 요소 또는 4/8을 갖춘 4/8/16/32/64 비트 부호없는 정수 또는 복잡한 유형 /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 | 토큰 |
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 | 토큰 |
결과:
결과 | 설명 |
---|---|
«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 | 토큰 |
결과:
결과 | 설명 |
---|---|
«unnamed» | 토큰 |
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 | 토큰 |
결과:
결과 | 설명 |
---|---|
«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 | 토큰 |
결과:
결과 | 설명 |
---|---|
«unnamed» | 토큰 |
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 (
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 (
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 | |
N | 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> |