Searched defs:ReduceOp (Results 1 – 10 of 10) sorted by relevance
34 struct TORCH_API ReduceOp : torch::CustomClassHolder { struct36 enum RedOpType : uint8_t {51 ReduceOp(RedOpType op) : op_(op) { in ReduceOp() argument57 ReduceOp( in ReduceOp() function75 operator RedOpType() const { in operator RedOpType()79 bool operator==(const std::uint8_t other) { in operator ==()84 bool operator==(const ReduceOp::RedOpType other) { in operator ==()89 bool operator==(const ReduceOp& other) { in operator ==()93 RedOpType op_ = SUM;101 c10::intrusive_ptr<_SupplementBase> supplement_;
341 struct ReduceOp { struct342 using traits = function_traits<decltype(&ops_t::reduce)>;343 using arg_t = typename std::decay<typename traits::template arg<0>::type>::type;345 using InputCalculator = OffsetCalculator<1, index_t>;346 using OutputCalculator = OffsetCalculator<2, index_t>;348 static constexpr bool can_accumulate_in_output =352 static constexpr int input_vec_size = ReduceConfig::input_vec_size;354 ops_t ops;355 arg_t ident;356 ReduceConfig config;[all …]
24 class ReduceOp(enum.Enum): class
53 ReduceOp = reduce_util.ReduceOp variable
141 ReduceOp( in ReduceOp() function154 ReduceOp( in ReduceOp() function
62 class ReduceOp; variable
289 reduce_scatter_reduce_op: Optional[Union[dist.ReduceOp, dist.ReduceOp.RedOpType]],
208 def supports_complex(reduceOp: ReduceOp) -> bool:538 redop: Optional[ReduceOp] = None,
1741 enum ReduceOp : uint16_t { enum
23377 SDValue AArch64TargetLowering::LowerPredReductionToSVE(SDValue ReduceOp, in LowerPredReductionToSVE()