chlo MLIR Dialect Builder API

Методы конструктора

chlo::AcosOp

Создает новую операцию chlo.acos .

MlirOp Acos(MlirOp &operand);

chlo::AcoshOp

Создает новую операцию chlo.acosh .

MlirOp Acosh(MlirOp &operand);

chlo::AsinAcosKernelOp

Создает новую операцию chlo._asin_acos_kernel .

MlirOp AsinAcosKernel(MlirOp &operand);

chlo::AsinOp

Создает новую операцию chlo.asin .

MlirOp Asin(MlirOp &operand);

chlo::AsinhOp

Создает новую операцию chlo.asinh .

MlirOp Asinh(MlirOp &operand);

chlo::AtanOp

Создает новую операцию chlo.atan .

MlirOp Atan(MlirOp &operand);

chlo::AtanhOp

Создает новую операцию chlo.atanh .

MlirOp Atanh(MlirOp &operand);

chlo::BesselI1eOp

Создает новую операцию chlo.bessel_i1e .

MlirOp BesselI1e(MlirOp &operand);

chlo::BroadcastAddOp

Создает новую операцию chlo.broadcast_add .

MlirOp BroadcastAdd(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastAndOp

Создает новую операцию chlo.broadcast_and .

MlirOp BroadcastAnd(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastAtan2Op

Создает новую операцию chlo.broadcast_atan2 .

MlirOp BroadcastAtan2(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastCompareOp

Создает новую операцию chlo.broadcast_compare .

MlirOp BroadcastCompare(MlirOp &lhs, MlirOp &rhs, ::mlir::chlo::ComparisonDirection comparison_direction, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {}, /*optional*/::mlir::chlo::ComparisonTypeAttr compare_type = {});

chlo::BroadcastComplexOp

Создает новую операцию chlo.broadcast_complex .

MlirOp BroadcastComplex(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastDivOp

Создает новую операцию chlo.broadcast_divide .

MlirOp BroadcastDiv(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastMaxOp

Создает новую операцию chlo.broadcast_maximum .

MlirOp BroadcastMax(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastMinOp

Создает новую операцию chlo.broadcast_minimum .

MlirOp BroadcastMin(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastMulOp

Создает новую операцию chlo.broadcast_multiply .

MlirOp BroadcastMul(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastNextAfterOp

Создает новую операцию chlo.broadcast_next_after .

MlirOp BroadcastNextAfter(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastOrOp

Создает новую операцию chlo.broadcast_or .

MlirOp BroadcastOr(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastPolygammaOp

Создает новую операцию chlo.broadcast_polygamma .

MlirOp BroadcastPolygamma(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastPowOp

Создает новую операцию chlo.broadcast_power .

MlirOp BroadcastPow(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastRemOp

Создает новую операцию chlo.broadcast_remainder .

MlirOp BroadcastRem(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastSelectOp

Создает новую операцию chlo.broadcast_select .

MlirOp BroadcastSelect(MlirOp &pred, MlirOp &on_true, MlirOp &on_false);

chlo::BroadcastShiftLeftOp

Создает новую операцию chlo.broadcast_shift_left .

MlirOp BroadcastShiftLeft(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastShiftRightArithmeticOp

Создает новую операцию chlo.broadcast_shift_right_arithmetic .

MlirOp BroadcastShiftRightArithmetic(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastShiftRightLogicalOp

Создает новую операцию chlo.broadcast_shift_right_logical .

MlirOp BroadcastShiftRightLogical(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastSubOp

Создает новую операцию chlo.broadcast_subtract .

MlirOp BroadcastSub(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastXorOp

Создает новую операцию chlo.broadcast_xor .

MlirOp BroadcastXor(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastZetaOp

Создает новую операцию chlo.broadcast_zeta .

MlirOp BroadcastZeta(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::ConjOp

Создает новую операцию chlo.conj .

MlirOp Conj(MlirOp &operand);

chlo::ConstantOp

Создает новую операцию chlo.constant .

MlirOp Constant(MlirBuilder &builder, ::mlir::ElementsAttr value);

chlo::CoshOp

Создает новую операцию chlo.cosh .

MlirOp Cosh(MlirOp &operand);

chlo::DigammaOp

Создает новую операцию chlo.digamma .

MlirOp Digamma(MlirOp &operand);

chlo::ErfInvOp

Создает новую операцию chlo.erf_inv .

MlirOp ErfInv(MlirOp &operand);

chlo::ErfOp

Создает новую операцию chlo.erf .

MlirOp Erf(MlirOp &operand);

chlo::ErfcOp

Создает новую операцию chlo.erfc .

MlirOp Erfc(MlirOp &operand);

chlo::IsInfOp

Создает новую операцию chlo.is_inf .

MlirOp IsInf(MlirOp &operand);

chlo::IsNegInfOp

Создает новую операцию chlo.is_neg_inf .

MlirOp IsNegInf(MlirOp &operand);

chlo::IsPosInfOp

Создает новую операцию chlo.is_pos_inf .

MlirOp IsPosInf(MlirOp &operand);

chlo::LgammaOp

Создает новую операцию chlo.lgamma .

MlirOp Lgamma(MlirOp &operand);

chlo::NextAfterOp

Создает новую операцию chlo.next_after .

MlirOp NextAfter(MlirOp &x, MlirOp &y);

chlo::PolygammaOp

Создает новую операцию chlo.polygamma .

MlirOp Polygamma(MlirOp &n, MlirOp &x);

chlo::RaggedDotOp

Создает новую операцию chlo.ragged_dot .

MlirOp RaggedDot(Type resultType, MlirOp &lhs, MlirOp &rhs, MlirOp &group_sizes, ::mlir::chlo::RaggedDotDimensionNumbersAttr ragged_dot_dimension_numbers, /*optional*/::mlir::ArrayAttr precision_config = {});

chlo::SinhOp

Создает новую операцию chlo.sinh .

MlirOp Sinh(MlirOp &operand);

chlo::SquareOp

Создает новую операцию chlo.square .

MlirOp Square(MlirOp &operand);

chlo::TanOp

Создает новую операцию chlo.tan .

MlirOp Tan(MlirOp &operand);

chlo::TopKOp

Создает новую операцию chlo.top_k .

SmallVector<MlirOp, 2> TopK(MlirOp &operand, uint64_t k);

chlo::ZetaOp

Создает новую операцию chlo.zeta .

MlirOp Zeta(MlirOp &x, MlirOp &q);

Пропущенные операции

Невозможно создать конструктор для следующих операций: