روش های سازنده
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);
عملیات نادیده گرفته شده
ایجاد سازنده برای عملیات زیر ممکن نیست: