Searched defs:CreateReduce (Results 1 – 2 of 2) sorted by relevance
475 Reduce CreateReduce(const std::set<Axis>& axis_to_reduce, const BHWC& src_shape, in CreateReduce() function482 Reduce CreateReduce(const std::set<Axis>& axis_to_reduce, in CreateReduce() function
1286 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateReduce( in CreateReduce() function in xla::HloInstruction1295 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateReduce( in CreateReduce() function in xla::HloInstruction