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, Type resultType, 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::CompositeOp

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

SmallVector<MlirOp> Composite(MlirBuilder &builder, Type resultType, ArrayRef<MlirOp> inputs, ::llvm::StringRef name, ::llvm::StringRef decomposition, /*optional*/::mlir::DictionaryAttr composite_attributes = {}, /*optional*/uint32_t version = 0);

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, Type resultType, 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(Type resultType, 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(Type resultType, 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 جدید ایجاد می کند.

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

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 جدید ایجاد می کند.

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

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);

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

ایجاد سازنده برای عملیات زیر ممکن نیست: