stablehlo MLIR Dialect Builder API

روش‌های سازنده

stablehlo::AbsOp

یک عملیات جدید stablehlo.abs ایجاد می‌کند.

MlirOp Abs(MlirOp &operand);

stablehlo::AddOp

یک عملیات جدید stablehlo.add ایجاد می‌کند.

MlirOp Add(MlirOp &lhs, MlirOp &rhs);

stablehlo::AfterAllOp

یک عملیات جدید stablehlo.after_all ایجاد می‌کند.

MlirOp AfterAll(MlirBuilder &builder, ArrayRef<MlirOp> inputs);

stablehlo::AllGatherOp

یک عملیات جدید stablehlo.all_gather ایجاد می‌کند.

SmallVector<MlirOp> AllGather(MlirBuilder &builder, TypeRange resultTypes, ArrayRef<MlirOp> operands, uint64_t all_gather_dim, ::mlir::DenseIntElementsAttr replica_groups, /*optional*/::mlir::stablehlo::ChannelHandleAttr channel_handle = {}, /*optional*/bool use_global_device_ids = false);

stablehlo::AllReduceOp

یک عملیات جدید stablehlo.all_reduce ایجاد می‌کند.

این عملیات دارای یک ناحیه بدنه است که از طریق یک تابع فراخوانی (callback) ساخته شده است.

SmallVector<MlirOp> AllReduce(MlirBuilder &builder, ArrayRef<MlirOp> operands, const RegionBuilderCallback &computation, ::mlir::DenseIntElementsAttr replica_groups, /*optional*/::mlir::stablehlo::ChannelHandleAttr channel_handle = {}, /*optional*/bool use_global_device_ids = false);

stablehlo::AllToAllOp

یک عملیات جدید stablehlo.all_to_all ایجاد می‌کند.

SmallVector<MlirOp> AllToAll(MlirBuilder &builder, ArrayRef<MlirOp> operands, uint64_t split_dimension, uint64_t concat_dimension, uint64_t split_count, ::mlir::DenseIntElementsAttr replica_groups, /*optional*/::mlir::stablehlo::ChannelHandleAttr channel_handle = {});

stablehlo::AndOp

یک عملیات جدید stablehlo.and ایجاد می‌کند.

MlirOp And(MlirOp &lhs, MlirOp &rhs);

stablehlo::Atan2Op

یک عملیات جدید stablehlo.atan2 ایجاد می‌کند.

MlirOp Atan2(MlirOp &lhs, MlirOp &rhs);

stablehlo::BatchNormGradOp

یک عملیات جدید stablehlo.batch_norm_grad ایجاد می‌کند.

SmallVector<MlirOp, 3> BatchNormGrad(MlirOp &operand, MlirOp &scale, MlirOp &mean, MlirOp &variance, MlirOp &grad_output, ::llvm::APFloat epsilon, uint64_t feature_index);

stablehlo::BatchNormInferenceOp

یک عملیات جدید stablehlo.batch_norm_inference ایجاد می‌کند.

MlirOp BatchNormInference(MlirOp &operand, MlirOp &scale, MlirOp &offset, MlirOp &mean, MlirOp &variance, ::llvm::APFloat epsilon, uint64_t feature_index);

stablehlo::BatchNormTrainingOp

یک عملیات جدید stablehlo.batch_norm_training ایجاد می‌کند.

SmallVector<MlirOp, 3> BatchNormTraining(MlirOp &operand, MlirOp &scale, MlirOp &offset, ::llvm::APFloat epsilon, uint64_t feature_index);

stablehlo::BitcastConvertOp

یک عملیات جدید stablehlo.bitcast_convert ایجاد می‌کند.

MlirOp BitcastConvert(Type resultType, MlirOp &operand);

stablehlo::BroadcastInDimOp

یک عملیات جدید stablehlo.broadcast_in_dim ایجاد می‌کند.

MlirOp BroadcastInDim(Type resultType, MlirOp &operand, ::llvm::ArrayRef<int64_t> broadcast_dimensions);

stablehlo::BroadcastOp

یک عملیات جدید stablehlo.broadcast ایجاد می‌کند.

MlirOp Broadcast(MlirOp &operand, ::llvm::ArrayRef<int64_t> broadcast_sizes);

stablehlo::CbrtOp

یک عملیات جدید stablehlo.cbrt ایجاد می‌کند.

MlirOp Cbrt(MlirOp &operand, /*optional*/::mlir::stablehlo::ResultAccuracyAttr result_accuracy = {});

stablehlo::CeilOp

یک عملیات جدید stablehlo.ceil ایجاد می‌کند.

MlirOp Ceil(MlirOp &operand);

stablehlo::CholeskyOp

یک عملیات جدید stablehlo.cholesky ایجاد می‌کند.

MlirOp Cholesky(MlirOp &a, /*optional*/bool lower = false);

stablehlo::ClampOp

یک عملیات جدید stablehlo.clamp ایجاد می‌کند.

MlirOp Clamp(MlirOp &min, MlirOp &operand, MlirOp &max);

stablehlo::ClzOp

یک عملیات جدید stablehlo.count_leading_zeros ایجاد می‌کند.

MlirOp Clz(MlirOp &operand);

stablehlo::CollectiveBroadcastOp

یک عملیات جدید stablehlo.collective_broadcast ایجاد می‌کند.

MlirOp CollectiveBroadcast(MlirOp &operand, ::mlir::DenseIntElementsAttr replica_groups, /*optional*/::mlir::stablehlo::ChannelHandleAttr channel_handle = {});

stablehlo::CollectivePermuteOp

یک عملیات جدید stablehlo.collective_permute ایجاد می‌کند.

MlirOp CollectivePermute(MlirOp &operand, ::mlir::DenseIntElementsAttr source_target_pairs, /*optional*/::mlir::stablehlo::ChannelHandleAttr channel_handle = {});

stablehlo::CompareOp

یک عملیات جدید stablehlo.compare ایجاد می‌کند.

MlirOp Compare(MlirOp &lhs, MlirOp &rhs, ::mlir::stablehlo::ComparisonDirection comparison_direction, /*optional*/::mlir::stablehlo::ComparisonTypeAttr compare_type = {});

stablehlo::ComplexOp

یک عملیات جدید stablehlo.complex ایجاد می‌کند.

MlirOp Complex(MlirOp &lhs, MlirOp &rhs);

stablehlo::ConcatenateOp

یک عملیات جدید stablehlo.concatenate ایجاد می‌کند.

MlirOp Concatenate(MlirBuilder &builder, ArrayRef<MlirOp> inputs, uint64_t dimension);

stablehlo::ConstantOp

یک عملیات جدید stablehlo.constant ایجاد می‌کند.

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

stablehlo::ConvertOp

یک عملیات جدید stablehlo.convert ایجاد می‌کند.

MlirOp Convert(Type resultType, MlirOp &operand);

stablehlo::ConvolutionOp

یک عملیات جدید stablehlo.convolution ایجاد می‌کند.

MlirOp Convolution(Type resultType, MlirOp &lhs, MlirOp &rhs, ::mlir::stablehlo::ConvDimensionNumbersAttr dimension_numbers, uint64_t feature_group_count, uint64_t batch_group_count, /*optional*/::mlir::DenseI64ArrayAttr window_strides = {}, /*optional*/::mlir::DenseIntElementsAttr padding = {}, /*optional*/::mlir::DenseI64ArrayAttr lhs_dilation = {}, /*optional*/::mlir::DenseI64ArrayAttr rhs_dilation = {}, /*optional*/::mlir::DenseBoolArrayAttr window_reversal = {}, /*optional*/::mlir::ArrayAttr precision_config = {});

stablehlo::CosineOp

یک عملیات جدید stablehlo.cosine ایجاد می‌کند.

MlirOp Cosine(MlirOp &operand, /*optional*/::mlir::stablehlo::ResultAccuracyAttr result_accuracy = {});

stablehlo::CreateTokenOp

یک عملیات جدید stablehlo.create_token ایجاد می‌کند.

MlirOp CreateToken(MlirBuilder &builder);

stablehlo::CrossReplicaSumOp

یک عملیات جدید stablehlo.cross-replica-sum ایجاد می‌کند.

MlirOp CrossReplicaSum(MlirOp &operand, ::mlir::DenseIntElementsAttr replica_groups);

stablehlo::CustomCallOp

یک عملیات جدید stablehlo.custom_call ایجاد می‌کند.

SmallVector<MlirOp> CustomCall(MlirBuilder &builder, TypeRange resultTypes, ArrayRef<MlirOp> inputs, ::llvm::StringRef call_target_name, /*optional*/bool has_side_effect = false, /*optional*/::mlir::Attribute backend_config = {}, /*optional*/::mlir::stablehlo::CustomCallApiVersion api_version = ::mlir::stablehlo::CustomCallApiVersion::API_VERSION_ORIGINAL, /*optional*/::mlir::ArrayAttr called_computations = {}, /*optional*/::mlir::ArrayAttr operand_layouts = {}, /*optional*/::mlir::ArrayAttr result_layouts = {}, /*optional*/::mlir::ArrayAttr output_operand_aliases = {});

stablehlo::DivOp

یک عملیات جدید stablehlo.divide ایجاد می‌کند.

MlirOp Div(MlirOp &lhs, MlirOp &rhs);

stablehlo::DotGeneralOp

یک عملیات جدید stablehlo.dot_general ایجاد می‌کند.

MlirOp DotGeneral(Type resultType, MlirOp &lhs, MlirOp &rhs, ::mlir::stablehlo::DotDimensionNumbersAttr dot_dimension_numbers, /*optional*/::mlir::ArrayAttr precision_config = {}, /*optional*/::mlir::stablehlo::DotAlgorithmAttr algorithm = {});

stablehlo::DotOp

یک عملیات جدید stablehlo.dot ایجاد می‌کند.

MlirOp Dot(Type resultType, MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::ArrayAttr precision_config = {});

stablehlo::DynamicBroadcastInDimOp

یک عملیات جدید stablehlo.dynamic_broadcast_in_dim ایجاد می‌کند.

MlirOp DynamicBroadcastInDim(Type resultType, MlirOp &operand, MlirOp &output_dimensions, ::llvm::ArrayRef<int64_t> broadcast_dimensions, /*optional*/::mlir::DenseI64ArrayAttr known_expanding_dimensions = {}, /*optional*/::mlir::DenseI64ArrayAttr known_nonexpanding_dimensions = {});

stablehlo::DynamicConvOp

یک عملیات جدید stablehlo.dynamic_conv ایجاد می‌کند.

MlirOp DynamicConv(Type resultType, MlirOp &lhs, MlirOp &rhs, MlirOp &padding, ::mlir::stablehlo::ConvDimensionNumbersAttr dimension_numbers, uint64_t feature_group_count, uint64_t batch_group_count, /*optional*/::mlir::DenseI64ArrayAttr window_strides = {}, /*optional*/::mlir::DenseI64ArrayAttr lhs_dilation = {}, /*optional*/::mlir::DenseI64ArrayAttr rhs_dilation = {}, /*optional*/::mlir::DenseBoolArrayAttr window_reversal = {}, /*optional*/::mlir::ArrayAttr precision_config = {});

stablehlo::DynamicGatherOp

یک عملیات جدید stablehlo.dynamic_gather ایجاد می‌کند.

MlirOp DynamicGather(MlirOp &operand, MlirOp &start_indices, MlirOp &slice_sizes, ::mlir::stablehlo::GatherDimensionNumbersAttr dimension_numbers, /*optional*/bool indices_are_sorted = false);

stablehlo::DynamicIotaOp

یک عملیات جدید stablehlo.dynamic_iota ایجاد می‌کند.

MlirOp DynamicIota(Type resultType, MlirOp &output_shape, uint64_t iota_dimension);

stablehlo::DynamicPadOp

یک عملیات جدید stablehlo.dynamic_pad ایجاد می‌کند.

MlirOp DynamicPad(Type resultType, MlirOp &operand, MlirOp &padding_value, MlirOp &edge_padding_low, MlirOp &edge_padding_high, MlirOp &interior_padding);

stablehlo::DynamicReshapeOp

یک عملیات جدید stablehlo.dynamic_reshape ایجاد می‌کند.

MlirOp DynamicReshape(Type resultType, MlirOp &operand, MlirOp &output_shape);

stablehlo::DynamicSliceOp

یک عملیات جدید stablehlo.dynamic_slice ایجاد می‌کند.

MlirOp DynamicSlice(MlirOp &operand, ArrayRef<MlirOp> start_indices, ::llvm::ArrayRef<int64_t> slice_sizes);

stablehlo::DynamicUpdateSliceOp

یک عملیات جدید stablehlo.dynamic_update_slice ایجاد می‌کند.

MlirOp DynamicUpdateSlice(MlirOp &operand, MlirOp &update, ArrayRef<MlirOp> start_indices);

stablehlo::EinsumOp

یک عملیات جدید stablehlo.einsum ایجاد می‌کند.

MlirOp Einsum(Type resultType, MlirOp &lhs, MlirOp &rhs, ::llvm::StringRef einsum_config);

stablehlo::ExpOp

یک عملیات جدید stablehlo.exponential ایجاد می‌کند.

MlirOp Exp(MlirOp &operand, /*optional*/::mlir::stablehlo::ResultAccuracyAttr result_accuracy = {});

stablehlo::Expm1Op

یک عملیات جدید stablehlo.exponential_minus_one ایجاد می‌کند.

MlirOp Expm1(MlirOp &operand, /*optional*/::mlir::stablehlo::ResultAccuracyAttr result_accuracy = {});

stablehlo::FftOp

یک عملیات جدید stablehlo.fft ایجاد می‌کند.

MlirOp Fft(MlirOp &operand, ::mlir::stablehlo::FftType fft_type, ::llvm::ArrayRef<int64_t> fft_length);

stablehlo::FloorOp

یک عملیات جدید stablehlo.floor ایجاد می‌کند.

MlirOp Floor(MlirOp &operand);

stablehlo::GatherOp

یک عملیات جدید stablehlo.gather ایجاد می‌کند.

MlirOp Gather(MlirOp &operand, MlirOp &start_indices, ::mlir::stablehlo::GatherDimensionNumbersAttr dimension_numbers, ::llvm::ArrayRef<int64_t> slice_sizes, /*optional*/bool indices_are_sorted = false);

stablehlo::GetDimensionSizeOp

یک عملیات جدید stablehlo.get_dimension_size ایجاد می‌کند.

MlirOp GetDimensionSize(MlirOp &operand, uint64_t dimension);

stablehlo::GetTupleElementOp

یک عملیات جدید stablehlo.get_tuple_element ایجاد می‌کند.

MlirOp GetTupleElement(MlirOp &operand, uint32_t index);

stablehlo::IfOp

یک عملیات جدید stablehlo.if ایجاد می‌کند.

این عملیات دارای یک ناحیه بدنه است که از طریق یک تابع فراخوانی (callback) ساخته شده است.

SmallVector<MlirOp> If(MlirOp &pred, const RegionBuilderCallback &true_branch, const RegionBuilderCallback &false_branch);

stablehlo::ImagOp

یک عملیات جدید stablehlo.imag ایجاد می‌کند.

MlirOp Imag(MlirOp &operand);

stablehlo::InfeedOp

یک عملیات جدید stablehlo.infeed ایجاد می‌کند.

SmallVector<MlirOp> Infeed(TypeRange resultTypes, MlirOp &token, ::llvm::StringRef infeed_config = "", /*optional*/::mlir::ArrayAttr layout = {});

stablehlo::IotaOp

یک عملیات جدید stablehlo.iota ایجاد می‌کند.

MlirOp Iota(MlirBuilder &builder, Type resultType, uint64_t iota_dimension);

stablehlo::IsFiniteOp

یک عملیات جدید stablehlo.is_finite ایجاد می‌کند.

MlirOp IsFinite(MlirOp &x);

stablehlo::Log1pOp

یک عملیات جدید stablehlo.log_plus_one ایجاد می‌کند.

MlirOp Log1p(MlirOp &operand, /*optional*/::mlir::stablehlo::ResultAccuracyAttr result_accuracy = {});

stablehlo::LogOp

یک عملیات جدید stablehlo.log ایجاد می‌کند.

MlirOp Log(MlirOp &operand, /*optional*/::mlir::stablehlo::ResultAccuracyAttr result_accuracy = {});

stablehlo::LogisticOp

یک عملیات جدید stablehlo.logistic ایجاد می‌کند.

MlirOp Logistic(MlirOp &operand, /*optional*/::mlir::stablehlo::ResultAccuracyAttr result_accuracy = {});

stablehlo::MapOp

یک عملیات جدید stablehlo.map ایجاد می‌کند.

این عملیات دارای یک ناحیه بدنه است که از طریق یک تابع فراخوانی (callback) ساخته شده است.

MlirOp Map(MlirBuilder &builder, ArrayRef<MlirOp> inputs, const RegionBuilderCallback &computation, ::llvm::ArrayRef<int64_t> dimensions);

stablehlo::MaxOp

یک عملیات جدید stablehlo.maximum ایجاد می‌کند.

MlirOp Max(MlirOp &lhs, MlirOp &rhs);

stablehlo::MinOp

یک عملیات جدید stablehlo.minimum ایجاد می‌کند.

MlirOp Min(MlirOp &lhs, MlirOp &rhs);

stablehlo::MulOp

یک عملیات جدید stablehlo.multiply ایجاد می‌کند.

MlirOp Mul(MlirOp &lhs, MlirOp &rhs);

stablehlo::NegOp

یک عملیات جدید stablehlo.negate ایجاد می‌کند.

MlirOp Neg(MlirOp &operand);

stablehlo::NotOp

یک عملیات جدید stablehlo.not ایجاد می‌کند.

MlirOp Not(MlirOp &operand);

stablehlo::OptimizationBarrierOp

یک عملیات جدید stablehlo.optimization_barrier ایجاد می‌کند.

SmallVector<MlirOp> OptimizationBarrier(MlirBuilder &builder, ArrayRef<MlirOp> operand);

stablehlo::OrOp

یک عملیات جدید stablehlo.or ایجاد می‌کند.

MlirOp Or(MlirOp &lhs, MlirOp &rhs);

stablehlo::OutfeedOp

یک عملیات جدید stablehlo.outfeed ایجاد می‌کند.

MlirOp Outfeed(ArrayRef<MlirOp> inputs, MlirOp &token, ::llvm::StringRef outfeed_config = "");

stablehlo::PadOp

یک عملیات جدید stablehlo.pad ایجاد می‌کند.

MlirOp Pad(MlirOp &operand, MlirOp &padding_value, ::llvm::ArrayRef<int64_t> edge_padding_low, ::llvm::ArrayRef<int64_t> edge_padding_high, ::llvm::ArrayRef<int64_t> interior_padding);

stablehlo::PartitionIdOp

یک عملیات جدید stablehlo.partition_id ایجاد می‌کند.

MlirOp PartitionId(MlirBuilder &builder);

stablehlo::PopulationCountOp

یک عملیات جدید stablehlo.popcnt ایجاد می‌کند.

MlirOp PopulationCount(MlirOp &operand);

stablehlo::PowOp

یک عملیات جدید stablehlo.power ایجاد می‌کند.

MlirOp Pow(MlirOp &lhs, MlirOp &rhs);

stablehlo::RealDynamicSliceOp

یک عملیات جدید stablehlo.real_dynamic_slice ایجاد می‌کند.

MlirOp RealDynamicSlice(Type resultType, MlirOp &operand, MlirOp &start_indices, MlirOp &limit_indices, MlirOp &strides);

stablehlo::RealOp

یک عملیات جدید stablehlo.real ایجاد می‌کند.

MlirOp Real(MlirOp &operand);

stablehlo::RecvOp

یک عملیات جدید stablehlo.recv ایجاد می‌کند.

SmallVector<MlirOp> Recv(TypeRange resultTypes, MlirOp &token, ::mlir::stablehlo::ChannelHandleAttr channel_handle, /*optional*/bool is_host_transfer = false, /*optional*/::mlir::DenseIntElementsAttr source_target_pairs = {});

stablehlo::ReduceOp

یک عملیات جدید stablehlo.reduce ایجاد می‌کند.

این عملیات دارای یک ناحیه بدنه است که از طریق یک تابع فراخوانی (callback) ساخته شده است.

SmallVector<MlirOp> Reduce(MlirBuilder &builder, ArrayRef<MlirOp> inputs, ArrayRef<MlirOp> init_values, const RegionBuilderCallback &body, ::llvm::ArrayRef<int64_t> dimensions);

stablehlo::ReducePrecisionOp

یک عملیات جدید stablehlo.reduce_precision ایجاد می‌کند.

MlirOp ReducePrecision(MlirOp &operand, uint32_t exponent_bits, uint32_t mantissa_bits);

stablehlo::ReduceScatterOp

یک عملیات جدید stablehlo.reduce_scatter ایجاد می‌کند.

این عملیات دارای یک ناحیه بدنه است که از طریق یک تابع فراخوانی (callback) ساخته شده است.

MlirOp ReduceScatter(Type resultType, MlirOp &operand, const RegionBuilderCallback &computation, uint64_t scatter_dimension, ::mlir::DenseIntElementsAttr replica_groups, /*optional*/::mlir::stablehlo::ChannelHandleAttr channel_handle = {}, /*optional*/bool use_global_device_ids = false);

stablehlo::ReduceWindowOp

یک عملیات جدید stablehlo.reduce_window ایجاد می‌کند.

این عملیات دارای یک ناحیه بدنه است که از طریق یک تابع فراخوانی (callback) ساخته شده است.

SmallVector<MlirOp> ReduceWindow(MlirBuilder &builder, ArrayRef<MlirOp> inputs, ArrayRef<MlirOp> init_values, const RegionBuilderCallback &body, ::llvm::ArrayRef<int64_t> window_dimensions, /*optional*/::mlir::DenseI64ArrayAttr window_strides = {}, /*optional*/::mlir::DenseI64ArrayAttr base_dilations = {}, /*optional*/::mlir::DenseI64ArrayAttr window_dilations = {}, /*optional*/::mlir::DenseIntElementsAttr padding = {});

stablehlo::RemOp

یک عملیات جدید stablehlo.remainder ایجاد می‌کند.

MlirOp Rem(MlirOp &lhs, MlirOp &rhs);

stablehlo::ReplicaIdOp

یک عملیات جدید stablehlo.replica_id ایجاد می‌کند.

MlirOp ReplicaId(MlirBuilder &builder);

stablehlo::ReshapeOp

یک عملیات جدید stablehlo.reshape ایجاد می‌کند.

MlirOp Reshape(Type resultType, MlirOp &operand);

stablehlo::ReturnOp

یک عملیات جدید stablehlo.return ایجاد می‌کند.

این عملیات، پایان‌دهنده‌ی یک Region است. فقط می‌توان آن را در فراخوانی تابع RegionBuilder هنگام ساخت بدنه‌ی یک op فراخوانی کرد.

void Return(RegionBuilder &builder, ArrayRef<MlirOp> results);

stablehlo::ReverseOp

یک عملیات جدید stablehlo.reverse ایجاد می‌کند.

MlirOp Reverse(MlirOp &operand, ::llvm::ArrayRef<int64_t> dimensions);

stablehlo::RngOp

یک عملیات جدید stablehlo.rng ایجاد می‌کند.

MlirOp Rng(MlirOp &a, MlirOp &b, MlirOp &shape, ::mlir::stablehlo::RngDistribution rng_distribution);

stablehlo::RoundNearestEvenOp

یک عملیات جدید stablehlo.round_nearest_even ایجاد می‌کند.

MlirOp RoundNearestEven(MlirOp &operand);

stablehlo::RoundOp

یک عملیات جدید stablehlo.round_nearest_afz ایجاد می‌کند.

MlirOp Round(MlirOp &operand);

stablehlo::RsqrtOp

یک عملیات جدید stablehlo.rsqrt ایجاد می‌کند.

MlirOp Rsqrt(MlirOp &operand, /*optional*/::mlir::stablehlo::ResultAccuracyAttr result_accuracy = {});

stablehlo::ScatterOp

یک عملیات جدید stablehlo.scatter ایجاد می‌کند.

این عملیات دارای یک ناحیه بدنه است که از طریق یک تابع فراخوانی ساخته شده است.

SmallVector<MlirOp> Scatter(ArrayRef<MlirOp> inputs, MlirOp &scatter_indices, ArrayRef<MlirOp> updates, const RegionBuilderCallback &update_computation, ::mlir::stablehlo::ScatterDimensionNumbersAttr scatter_dimension_numbers, /*optional*/bool indices_are_sorted = false, /*optional*/bool unique_indices = false);

stablehlo::SelectAndScatterOp

یک عملیات جدید stablehlo.select_and_scatter ایجاد می‌کند.

این عملیات دارای یک ناحیه بدنه است که از طریق یک تابع فراخوانی (callback) ساخته شده است.

MlirOp SelectAndScatter(MlirOp &operand, MlirOp &source, MlirOp &init_value, const RegionBuilderCallback &select, const RegionBuilderCallback &scatter, /*optional*/::mlir::DenseI64ArrayAttr window_dimensions = {}, /*optional*/::mlir::DenseI64ArrayAttr window_strides = {}, /*optional*/::mlir::DenseIntElementsAttr padding = {});

stablehlo::SelectOp

یک عملیات جدید stablehlo.select ایجاد می‌کند.

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

stablehlo::SendOp

یک عملیات جدید stablehlo.send ایجاد می‌کند.

MlirOp Send(ArrayRef<MlirOp> inputs, MlirOp &token, ::mlir::stablehlo::ChannelHandleAttr channel_handle, /*optional*/bool is_host_transfer = false, /*optional*/::mlir::DenseIntElementsAttr source_target_pairs = {});

stablehlo::SetDimensionSizeOp

یک عملیات جدید stablehlo.set_dimension_size ایجاد می‌کند.

MlirOp SetDimensionSize(MlirOp &operand, MlirOp &size, uint64_t dimension);

stablehlo::ShiftLeftOp

یک عملیات جدید stablehlo.shift_left ایجاد می‌کند.

MlirOp ShiftLeft(MlirOp &lhs, MlirOp &rhs);

stablehlo::ShiftRightArithmeticOp

یک عملیات جدید stablehlo.shift_right_arithmetic ایجاد می‌کند.

MlirOp ShiftRightArithmetic(MlirOp &lhs, MlirOp &rhs);

stablehlo::ShiftRightLogicalOp

یک عملیات جدید stablehlo.shift_right_logical ایجاد می‌کند.

MlirOp ShiftRightLogical(MlirOp &lhs, MlirOp &rhs);

stablehlo::SignOp

یک عملیات جدید stablehlo.sign ایجاد می‌کند.

MlirOp Sign(MlirOp &operand);

stablehlo::SineOp

یک عملیات جدید stablehlo.sine ایجاد می‌کند.

MlirOp Sine(MlirOp &operand, /*optional*/::mlir::stablehlo::ResultAccuracyAttr result_accuracy = {});

stablehlo::SliceOp

یک عملیات جدید stablehlo.slice ایجاد می‌کند.

MlirOp Slice(MlirOp &operand, ::llvm::ArrayRef<int64_t> start_indices, ::llvm::ArrayRef<int64_t> limit_indices, ::llvm::ArrayRef<int64_t> strides);

stablehlo::SortOp

یک عملیات جدید stablehlo.sort ایجاد می‌کند.

این عملیات دارای یک ناحیه بدنه است که از طریق یک تابع فراخوانی (callback) ساخته شده است.

SmallVector<MlirOp> Sort(MlirBuilder &builder, ArrayRef<MlirOp> inputs, const RegionBuilderCallback &comparator, /*optional*/uint64_t dimension = -1, /*optional*/bool is_stable = false);

stablehlo::SqrtOp

یک عملیات جدید stablehlo.sqrt ایجاد می‌کند.

MlirOp Sqrt(MlirOp &operand, /*optional*/::mlir::stablehlo::ResultAccuracyAttr result_accuracy = {});

stablehlo::SubtractOp

یک عملیات جدید stablehlo.subtract ایجاد می‌کند.

MlirOp Subtract(MlirOp &lhs, MlirOp &rhs);

stablehlo::TanOp

یک عملیات جدید stablehlo.tan ایجاد می‌کند.

MlirOp Tan(MlirOp &operand, /*optional*/::mlir::stablehlo::ResultAccuracyAttr result_accuracy = {});

stablehlo::TanhOp

یک عملیات جدید stablehlo.tanh ایجاد می‌کند.

MlirOp Tanh(MlirOp &operand, /*optional*/::mlir::stablehlo::ResultAccuracyAttr result_accuracy = {});

stablehlo::TorchIndexSelectOp

یک عملیات جدید stablehlo.torch_index_select ایجاد می‌کند.

MlirOp TorchIndexSelect(Type resultType, MlirOp &operand, MlirOp &index, uint64_t dim, uint64_t batch_dims);

stablehlo::TransposeOp

یک عملیات جدید stablehlo.transpose ایجاد می‌کند.

MlirOp Transpose(MlirOp &operand, ::llvm::ArrayRef<int64_t> permutation);

stablehlo::TupleOp

یک عملیات جدید stablehlo.tuple ایجاد می‌کند.

MlirOp Tuple(MlirBuilder &builder, ArrayRef<MlirOp> val);

stablehlo::UnaryEinsumOp

یک عملیات جدید stablehlo.unary_einsum ایجاد می‌کند.

MlirOp UnaryEinsum(Type resultType, MlirOp &operand, ::llvm::StringRef einsum_config);

stablehlo::UniformDequantizeOp

یک عملیات جدید stablehlo.uniform_dequantize ایجاد می‌کند.

MlirOp UniformDequantize(MlirOp &operand);

stablehlo::UniformQuantizeOp

یک عملیات جدید stablehlo.uniform_quantize ایجاد می‌کند.

MlirOp UniformQuantize(Type resultType, MlirOp &operand);

stablehlo::WhileOp

یک عملیات جدید stablehlo.while ایجاد می‌کند.

این عملیات دارای یک ناحیه بدنه است که از طریق یک تابع فراخوانی (callback) ساخته شده است.

SmallVector<MlirOp> While(MlirBuilder &builder, ArrayRef<MlirOp> operand, const RegionBuilderCallback &cond, const RegionBuilderCallback &body);

stablehlo::XorOp

یک عملیات جدید stablehlo.xor ایجاد می‌کند.

MlirOp Xor(MlirOp &lhs, MlirOp &rhs);

عملیات‌های نادیده گرفته شده

قادر به تولید سازنده برای عملیات زیر نیست: