Searched defs:ReduceAdd (Results 1 – 6 of 6) sorted by relevance
62 struct ReduceAdd : public autograd::Node { struct63 explicit ReduceAdd(const at::Device& destination_device) in ReduceAdd() argument65 ~ReduceAdd() override {} in ~ReduceAdd() argument70 "ReduceAdd can only be used during the backward pass of data parallel."); in apply() argument
44 struct ReduceAdd { struct45 inline __device__ T operator()(const T a, const T b) const { in operator ()()
31 class ReduceAdd { class
304 class ReduceAdd { class
46 class ReduceAdd { class
2624 TEST_P(HloEvaluatorBf16Test, ReduceAdd) { in TEST_P() argument