Searched refs:HloFftInstruction (Results 1 – 4 of 4) sorted by relevance
168 HloFftInstruction::HloFftInstruction(const Shape& shape, in HloFftInstruction() function in xla::HloFftInstruction176 HloInstructionProto HloFftInstruction::ToProto() const { in ToProto()185 std::vector<string> HloFftInstruction::ExtraAttributesToStringImpl( in ExtraAttributesToStringImpl()191 bool HloFftInstruction::IdenticalSlowPath( in IdenticalSlowPath()195 const auto& casted_other = static_cast<const HloFftInstruction&>(other); in IdenticalSlowPath()200 std::unique_ptr<HloInstruction> HloFftInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()204 return absl::make_unique<HloFftInstruction>(shape, new_operands[0], fft_type_, in CloneWithNewOperandsImpl()
104 class HloFftInstruction : public HloInstruction {106 explicit HloFftInstruction(const Shape& shape, HloInstruction* operand,
996 return absl::make_unique<HloFftInstruction>(shape, operand, fft_type, in CreateFft()3903 return Cast<HloFftInstruction>(this)->fft_type(); in fft_type()3907 return Cast<HloFftInstruction>(this)->fft_length(); in fft_length()
1286 auto hlo_fft = xla::Cast<xla::HloFftInstruction>(instr); in EmitFftOp()