Builder Methods
chlo::AcosOp
Creates a new chlo.acos
operation.
MlirOp Acos(MlirOp &operand);
chlo::AcoshOp
Creates a new chlo.acosh
operation.
MlirOp Acosh(MlirOp &operand);
chlo::AsinAcosKernelOp
Creates a new chlo._asin_acos_kernel
operation.
MlirOp AsinAcosKernel(MlirOp &operand);
chlo::AsinOp
Creates a new chlo.asin
operation.
MlirOp Asin(MlirOp &operand);
chlo::AsinhOp
Creates a new chlo.asinh
operation.
MlirOp Asinh(MlirOp &operand);
chlo::AtanOp
Creates a new chlo.atan
operation.
MlirOp Atan(MlirOp &operand);
chlo::AtanhOp
Creates a new chlo.atanh
operation.
MlirOp Atanh(MlirOp &operand);
chlo::BesselI1eOp
Creates a new chlo.bessel_i1e
operation.
MlirOp BesselI1e(MlirOp &operand);
chlo::BroadcastAddOp
Creates a new chlo.broadcast_add
operation.
MlirOp BroadcastAdd(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});
chlo::BroadcastAndOp
Creates a new chlo.broadcast_and
operation.
MlirOp BroadcastAnd(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});
chlo::BroadcastAtan2Op
Creates a new chlo.broadcast_atan2
operation.
MlirOp BroadcastAtan2(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});
chlo::BroadcastCompareOp
Creates a new chlo.broadcast_compare
operation.
MlirOp BroadcastCompare(MlirOp &lhs, MlirOp &rhs, ::mlir::chlo::ComparisonDirection comparison_direction, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {}, /*optional*/::mlir::chlo::ComparisonTypeAttr compare_type = {});
chlo::BroadcastComplexOp
Creates a new chlo.broadcast_complex
operation.
MlirOp BroadcastComplex(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});
chlo::BroadcastDivOp
Creates a new chlo.broadcast_divide
operation.
MlirOp BroadcastDiv(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});
chlo::BroadcastMaxOp
Creates a new chlo.broadcast_maximum
operation.
MlirOp BroadcastMax(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});
chlo::BroadcastMinOp
Creates a new chlo.broadcast_minimum
operation.
MlirOp BroadcastMin(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});
chlo::BroadcastMulOp
Creates a new chlo.broadcast_multiply
operation.
MlirOp BroadcastMul(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});
chlo::BroadcastNextAfterOp
Creates a new chlo.broadcast_next_after
operation.
MlirOp BroadcastNextAfter(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});
chlo::BroadcastOrOp
Creates a new chlo.broadcast_or
operation.
MlirOp BroadcastOr(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});
chlo::BroadcastPolygammaOp
Creates a new chlo.broadcast_polygamma
operation.
MlirOp BroadcastPolygamma(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});
chlo::BroadcastPowOp
Creates a new chlo.broadcast_power
operation.
MlirOp BroadcastPow(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});
chlo::BroadcastRemOp
Creates a new chlo.broadcast_remainder
operation.
MlirOp BroadcastRem(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});
chlo::BroadcastSelectOp
Creates a new chlo.broadcast_select
operation.
MlirOp BroadcastSelect(MlirOp &pred, MlirOp &on_true, MlirOp &on_false);
chlo::BroadcastShiftLeftOp
Creates a new chlo.broadcast_shift_left
operation.
MlirOp BroadcastShiftLeft(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});
chlo::BroadcastShiftRightArithmeticOp
Creates a new chlo.broadcast_shift_right_arithmetic
operation.
MlirOp BroadcastShiftRightArithmetic(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});
chlo::BroadcastShiftRightLogicalOp
Creates a new chlo.broadcast_shift_right_logical
operation.
MlirOp BroadcastShiftRightLogical(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});
chlo::BroadcastSubOp
Creates a new chlo.broadcast_subtract
operation.
MlirOp BroadcastSub(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});
chlo::BroadcastXorOp
Creates a new chlo.broadcast_xor
operation.
MlirOp BroadcastXor(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});
chlo::BroadcastZetaOp
Creates a new chlo.broadcast_zeta
operation.
MlirOp BroadcastZeta(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});
chlo::ConjOp
Creates a new chlo.conj
operation.
MlirOp Conj(MlirOp &operand);
chlo::ConstantOp
Creates a new chlo.constant
operation.
MlirOp Constant(MlirBuilder &builder, ::mlir::ElementsAttr value);
chlo::CoshOp
Creates a new chlo.cosh
operation.
MlirOp Cosh(MlirOp &operand);
chlo::DigammaOp
Creates a new chlo.digamma
operation.
MlirOp Digamma(MlirOp &operand);
chlo::ErfInvOp
Creates a new chlo.erf_inv
operation.
MlirOp ErfInv(MlirOp &operand);
chlo::ErfOp
Creates a new chlo.erf
operation.
MlirOp Erf(MlirOp &operand);
chlo::ErfcOp
Creates a new chlo.erfc
operation.
MlirOp Erfc(MlirOp &operand);
chlo::IsInfOp
Creates a new chlo.is_inf
operation.
MlirOp IsInf(MlirOp &operand);
chlo::IsNegInfOp
Creates a new chlo.is_neg_inf
operation.
MlirOp IsNegInf(MlirOp &operand);
chlo::IsPosInfOp
Creates a new chlo.is_pos_inf
operation.
MlirOp IsPosInf(MlirOp &operand);
chlo::LgammaOp
Creates a new chlo.lgamma
operation.
MlirOp Lgamma(MlirOp &operand);
chlo::NextAfterOp
Creates a new chlo.next_after
operation.
MlirOp NextAfter(MlirOp &x, MlirOp &y);
chlo::PolygammaOp
Creates a new chlo.polygamma
operation.
MlirOp Polygamma(MlirOp &n, MlirOp &x);
chlo::RaggedDotOp
Creates a new chlo.ragged_dot
operation.
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
Creates a new chlo.sinh
operation.
MlirOp Sinh(MlirOp &operand);
chlo::SquareOp
Creates a new chlo.square
operation.
MlirOp Square(MlirOp &operand);
chlo::TanOp
Creates a new chlo.tan
operation.
MlirOp Tan(MlirOp &operand);
chlo::TopKOp
Creates a new chlo.top_k
operation.
SmallVector<MlirOp, 2> TopK(MlirOp &operand, uint64_t k);
chlo::ZetaOp
Creates a new chlo.zeta
operation.
MlirOp Zeta(MlirOp &x, MlirOp &q);
Skipped Operations
Unable to generate builder for the following operations: