Searched refs:HloMapInstruction (Results 1 – 6 of 6) sorted by relevance
120 return ir_emitter_->EmitElementalMap(*Cast<HloMapInstruction>(hlo), in MakeElementGenerator()
118 const HloMapInstruction& map_instr,
635 const HloMapInstruction& map_instr, in EmitElementalMap()
907 HloMapInstruction::HloMapInstruction(const Shape& shape, in HloMapInstruction() function in xla::HloMapInstruction921 HloInstructionProto HloMapInstruction::ToProto() const { in ToProto()929 bool HloMapInstruction::IsElementwiseImpl( in IsElementwiseImpl()945 std::vector<string> HloMapInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()950 bool HloMapInstruction::IdenticalSlowPath( in IdenticalSlowPath()957 std::unique_ptr<HloInstruction> HloMapInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()960 return absl::make_unique<HloMapInstruction>(shape, new_operands, to_apply()); in CloneWithNewOperandsImpl()
580 class HloMapInstruction : public HloInstruction {582 explicit HloMapInstruction(const Shape& shape,
777 return absl::make_unique<HloMapInstruction>(shape, operands, map_computation); in CreateMap()