| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_fft_test.cc | 29 EXPECT_EQ(::xla::FftType::IRFFT, static_cast<::tensorflow::int32>( in TEST()
30 ::tensorflow::xla::FftType::IRFFT)); in TEST()
|
| D | runtime_fft_impl.h | 34 IRFFT = 3, // Inverse real FFT; fft_length / 2 + 1 complex in, enumerator
223 case FftType::IRFFT: in EigenFftWithRank()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | IRFFT.pbtxt | 2 name: "IRFFT"
17 name: "IRFFT"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | IRFFT.pbtxt | 2 name: "IRFFT"
17 name: "IRFFT"
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | fft_ops.cc | 54 if (fft_type_ == FftType::RFFT || fft_type_ == FftType::IRFFT) { in Compile()
65 if (fft_type_ == FftType::IRFFT) { in Compile()
161 : GenericFftOp(ctx, /*fft_type=*/FftType::IRFFT, /*fft_rank=*/FFTRank) {} in IRFFTOp()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_IRFFT.pbtxt | 2 graph_op_name: "IRFFT"
39 Along the axis `IRFFT` is computed on, if `fft_length / 2 + 1` is smaller
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_IRFFT.pbtxt | 2 graph_op_name: "IRFFT"
|
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_IRFFT.pbtxt | 2 graph_op_name: "IRFFT"
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | attribute_exporter.cc | 148 case mlir::mhlo::FftType::IRFFT: in ConvertFftType()
149 return xla::FftType::IRFFT; in ConvertFftType()
|
| D | attribute_importer.cc | 135 case FftType::IRFFT: in ConvertFftType()
136 return mlir::mhlo::FftType::IRFFT; in ConvertFftType()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops_base_enums.td | 42 def HLO_FFT_TYPE_IRFFT : StrEnumAttrCase<"IRFFT">;
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | fft_thunk.cc | 71 case FftType::IRFFT: in FftTypeToSeType()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 1003 const bool inverse = fft_type == FftType::IFFT || fft_type == FftType::IRFFT; in Sweep()
1004 const bool input_is_truncated = fft_type == FftType::IRFFT; in Sweep()
1139 const bool input_is_truncated = fft_type == FftType::IRFFT; in CopyDataFromInput()
1247 const bool output_is_float = fft_type == FftType::IRFFT; in CopyDataToOutput()
1295 if (fft_type == FftType::IRFFT && output_elt_type != PrimitiveType::F32) { in CheckParameters()
1299 if (fft_type != FftType::IRFFT && output_elt_type != PrimitiveType::C64) { in CheckParameters()
|
| D | shape_inference_test.cc | 904 FftType type = FftType::IRFFT; in TEST_F()
915 FftType type = FftType::IRFFT; in TEST_F()
934 FftType type = FftType::IRFFT; in TEST_F()
|
| D | hlo_parser_test.cc | 843 …ROOT %fft = f32[5,64,128,64]{3,2,1,0} fft(c64[5,64,128,33]{3,2,1,0} %input), fft_type=IRFFT, fft_l… in CreateTestCases()
|
| D | shape_inference.cc | 1923 case IRFFT: { in InferFftShape()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_compiler.cc | 584 .value("IRFFT", FftType::IRFFT); in BuildXlaCompilerSubmodule()
|
| D | xla_client_test.py | 1417 ops.Fft(ops.Constant(c, a), xla_client.FftType.IRFFT, [3, 4, 8])
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | op_select_allowlist.md | 211 * `raw_ops.IRFFT`
|
| /external/tensorflow/tensorflow/compiler/tf2xla/g3doc/ |
| D | gpu_supported_ops.md | 102 `IRFFT` |
|
| D | cpu_supported_ops.md | 102 `IRFFT` |
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 561 IRFFT = 3; // Inverse real FFT; fft_length / 2 + 1 complex in, enumerator
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 1386 : : : IRFFT to right-size the :
1401 | `IRFFT` | Inverse real-to-complex FFT (i.e. takes complex, returns real). |
1412 (effectively) performed first (RFFT; last for IRFFT), which is why the innermost
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1557 TEST_F(OpTest, IRFFT) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf.mlir | 1899 …// CHECK: "mhlo.fft"(%[[SLICED]]) {fft_length = dense<5> : tensor<1xi64>, fft_type = "IRFFT"} : (t…
1900 %0 = "tf.IRFFT"(%arg0, %fftlength) : (tensor<8xcomplex<f32>>, tensor<1xi32>) -> tensor<5xf32>
|