Searched refs:HloFftInstruction (Results 1 – 3 of 3) sorted by relevance
165 HloFftInstruction::HloFftInstruction(const Shape& shape, in HloFftInstruction() function in xla::HloFftInstruction173 HloInstructionProto HloFftInstruction::ToProto() const { in ToProto()182 std::vector<string> HloFftInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()188 bool HloFftInstruction::IdenticalSlowPath( in IdenticalSlowPath()192 const auto& casted_other = static_cast<const HloFftInstruction&>(other); in IdenticalSlowPath()197 std::unique_ptr<HloInstruction> HloFftInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()201 return absl::make_unique<HloFftInstruction>(shape, new_operands[0], fft_type_, in CloneWithNewOperandsImpl()
102 class HloFftInstruction : public HloInstruction {104 explicit HloFftInstruction(const Shape& shape, HloInstruction* operand,
793 return absl::make_unique<HloFftInstruction>(shape, operand, fft_type, in CreateFft()3223 return Cast<HloFftInstruction>(this)->fft_type(); in fft_type()3227 return Cast<HloFftInstruction>(this)->fft_length(); in fft_length()