| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | fft_ops.cc | 38 using xla::FftType;
42 explicit GenericFftOp(OpKernelConstruction* ctx, FftType fft_type, in GenericFftOp()
54 if (fft_type_ == FftType::RFFT || fft_type_ == FftType::IRFFT) { in Compile()
65 if (fft_type_ == FftType::IRFFT) { in Compile()
103 const FftType fft_type_;
114 : GenericFftOp(ctx, /*fft_type=*/FftType::FFT, /*fft_rank=*/FFTRank) {} in FFTOp()
126 : GenericFftOp(ctx, /*fft_type=*/FftType::IFFT, /*fft_rank=*/FFTRank) {} in IFFTOp()
139 : GenericFftOp(ctx, /*fft_type=*/FftType::RFFT, /*fft_rank=*/FFTRank) {} in RFFTOp()
161 : GenericFftOp(ctx, /*fft_type=*/FftType::IRFFT, /*fft_rank=*/FFTRank) {} in IRFFTOp()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | fft_thunk.cc | 63 se::fft::Type FftTypeToSeType(FftType type) { in FftTypeToSeType()
65 case FftType::FFT: in FftTypeToSeType()
67 case FftType::IFFT: in FftTypeToSeType()
69 case FftType::IRFFT: in FftTypeToSeType()
71 case FftType::RFFT: in FftTypeToSeType()
95 FftThunk::FftThunk(FftType fft_type, absl::Span<const int64> fft_length, in FftThunk()
|
| D | fft_thunk.h | 65 FftThunk(FftType fft_type, absl::Span<const int64> fft_length,
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_fft_impl.h | 176 case ::xla::FftType::FFT: in EigenFftWithRank()
182 case ::xla::FftType::IFFT: in EigenFftWithRank()
188 case ::xla::FftType::RFFT: in EigenFftWithRank()
193 case ::xla::FftType::IRFFT: in EigenFftWithRank()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 987 void Sweep(int64 fft_rank, FftType fft_type, in Sweep()
991 const bool inverse = fft_type == FftType::IFFT || fft_type == FftType::IRFFT; in Sweep()
992 const bool input_is_truncated = fft_type == FftType::IRFFT; in Sweep()
993 const bool output_is_truncated = fft_type == FftType::RFFT; in Sweep()
1119 int64 fft_rank, FftType fft_type, int64 fft_size, in CopyDataFromInput()
1127 const bool input_is_truncated = fft_type == FftType::IRFFT; in CopyDataFromInput()
1172 int64 fft_rank, FftType fft_type, in CopyDataToOutput()
1178 const bool output_is_truncated = fft_type == FftType::RFFT; in CopyDataToOutput()
1209 int64 fft_rank, FftType fft_type, int64 fft_size, in CopyDataFromInput()
1215 const bool input_is_float = fft_type == FftType::RFFT; in CopyDataFromInput()
[all …]
|
| D | shape_inference_test.cc | 624 static void Pass(const Shape& shape, FftType type, in Pass()
632 static void Fail(const Shape& shape, FftType type, in Fail()
643 FftType type = FftType::FFT; in TEST_F()
653 FftType type = FftType::FFT; in TEST_F()
661 FftType type = FftType::IFFT; in TEST_F()
671 FftType type = FftType::IFFT; in TEST_F()
679 FftType type = FftType::RFFT; in TEST_F()
690 FftType type = FftType::RFFT; in TEST_F()
710 FftType type = FftType::RFFT; in TEST_F()
718 FftType type = FftType::IRFFT; in TEST_F()
[all …]
|
| D | shape_inference.h | 116 static StatusOr<Shape> InferFftShape(const Shape& in, FftType fft_type,
|
| D | hlo_instructions.h | 105 FftType fft_type,
107 FftType fft_type() const { return fft_type_; } in fft_type()
128 FftType fft_type_ = FftType::FFT;
|
| D | hlo_instruction.h | 565 const Shape& shape, HloInstruction* operand, FftType fft_type,
1558 FftType fft_type() const;
|
| D | hlo_parser.cc | 321 bool ParseFftType(FftType* result);
1169 optional<FftType> fft_type; in ParseInstructionRhs()
2806 FftType result; in ParseAttributeHelper()
2810 static_cast<optional<FftType>*>(attr_out_ptr)->emplace(result); in ParseAttributeHelper()
3903 bool HloParserImpl::ParseFftType(FftType* result) { in ParseFftType()
|
| D | hlo.proto | 148 xla.FftType fft_type = 31;
|
| D | hlo_instruction.cc | 869 const Shape& shape, HloInstruction* operand, FftType fft_type, in CreateFft()
3552 FftType HloInstruction::fft_type() const { in fft_type()
|
| D | hlo_instructions.cc | 167 HloInstruction* operand, FftType fft_type, in HloFftInstruction()
|
| D | shape_inference.cc | 1842 const Shape& in, const FftType fft_type, in InferFftShape()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla.cc | 867 py::enum_<FftType>(m, "FftType") in PYBIND11_MODULE()
868 .value("FFT", FftType::FFT) in PYBIND11_MODULE()
869 .value("IFFT", FftType::IFFT) in PYBIND11_MODULE()
870 .value("RFFT", FftType::RFFT) in PYBIND11_MODULE()
871 .value("IRFFT", FftType::IRFFT); in PYBIND11_MODULE()
|
| D | xla_client.py | 1637 FftType = _xla.FftType variable
|
| D | xla_client_test.py | 1447 c.Fft(c.Constant(a), xla_client.FftType.FFT, shape[-3:])
1452 c.Fft(c.Constant(a), xla_client.FftType.IFFT, shape[-3:])
1458 c.Fft(c.Constant(b), xla_client.FftType.RFFT, shape[-3:])
1463 c.Fft(c.Constant(a), xla_client.FftType.IRFFT, [3, 4, 8])
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 123 static xla::FftType Convert_fft_type(llvm::StringRef fft_type_str) { in Convert_fft_type()
124 xla::FftType fft_type_enum; in Convert_fft_type()
128 return xla::FftType::FFT; in Convert_fft_type()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 465 XlaOp Fft(XlaOp operand, FftType fft_type,
851 friend XlaOp Fft(XlaOp operand, FftType fft_type,
1439 XlaOp Fft(XlaOp operand, FftType fft_type, absl::Span<const int64> fft_length);
|
| D | xla_builder.cc | 1323 XlaOp XlaBuilder::Fft(XlaOp operand, const FftType fft_type, in Fft()
3089 XlaOp Fft(const XlaOp operand, FftType fft_type, in Fft()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | hlo_ops_base.td | 588 // These mirror the XLA FftType proto enum.
594 def HLO_FftTypeAttr : StrEnumAttr<"FftType",
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 529 enum FftType { enum
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 1311 | `fft_type` | `FftType` | See the table below. |
1323 | `FftType` | Semantics |
|