Searched refs:HloReduceInstruction (Results 1 – 8 of 8) sorted by relevance
901 HloReduceInstruction::HloReduceInstruction( in HloReduceInstruction() function in xla::HloReduceInstruction913 HloInstructionProto HloReduceInstruction::ToProto() const { in ToProto()921 std::vector<string> HloReduceInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()926 bool HloReduceInstruction::IdenticalSlowPath( in IdenticalSlowPath()930 const auto& casted_other = static_cast<const HloReduceInstruction&>(other); in IdenticalSlowPath()937 std::unique_ptr<HloInstruction> HloReduceInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()941 return absl::make_unique<HloReduceInstruction>(shape, new_operands, in CloneWithNewOperandsImpl()
258 const HloReduceInstruction* reduce,
585 class HloReduceInstruction : public HloInstruction {587 explicit HloReduceInstruction(const Shape& shape,
2405 HloReduceInstruction* reduce = Cast<HloReduceInstruction>(instr); in HandleReduce()
1290 auto instruction = absl::WrapUnique(new HloReduceInstruction( in CreateReduce()1304 return absl::make_unique<HloReduceInstruction>( in CreateReduce()3593 return operand_num >= Cast<HloReduceInstruction>(this)->input_count() in OperandElementUse()
2510 auto reduce_instr = Cast<HloReduceInstruction>(hlo); in MakeElementGenerator()2709 const HloReduceInstruction* reduce, in EmitElementalReduce()
4513 HloReduceInstruction* reduce = Cast<HloReduceInstruction>(hlo); in HandleReduce()
949 auto* reduce = xla::Cast<xla::HloReduceInstruction>(instr); in EmitReduceOp()