Searched refs:HloReduceWindowInstruction (Results 1 – 6 of 6) sorted by relevance
126 Cast<HloReduceWindowInstruction>(hlo), in MakeElementGenerator()
123 const HloReduceWindowInstruction* reduce_window,
641 const HloReduceWindowInstruction* reduce_window, in EmitElementalReduceWindow()
1952 HloReduceWindowInstruction::HloReduceWindowInstruction( in HloReduceWindowInstruction() function in xla::HloReduceWindowInstruction1961 HloInstructionProto HloReduceWindowInstruction::ToProto() const { in ToProto()1967 std::vector<string> HloReduceWindowInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()1976 bool HloReduceWindowInstruction::IdenticalSlowPath( in IdenticalSlowPath()1981 static_cast<const HloReduceWindowInstruction&>(other); in IdenticalSlowPath()1987 HloReduceWindowInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()1991 return absl::make_unique<HloReduceWindowInstruction>( in CloneWithNewOperandsImpl()
1093 class HloReduceWindowInstruction : public HloInstruction {1095 explicit HloReduceWindowInstruction(const Shape& shape,
1057 return absl::make_unique<HloReduceWindowInstruction>( in CreateReduceWindow()