Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for IRFFT (0.06 sec)

  1. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir

      // CHECK: "mhlo.fft"(%[[SLICED]]) <{fft_length = dense<8> : tensor<1xi64>, fft_type = #mhlo<fft_type IRFFT>}> : (tensor<5xcomplex<f32>>
      %0 = "tf.IRFFT"(%arg0, %fftlength) : (tensor<8xcomplex<f32>>, tensor<1xi32>) -> tensor<8xf32>
      func.return %0 : tensor<8xf32>
    }
    
    // -----
    
    // CHECK-LABEL: fft_1D_dynamic
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 335.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

        }
    
        int64_t expected_dim = fft_length;
        std::string fft_string = "RFFT";
        if (typeid(OpTy) == typeid(TF::IRFFTOp)) {
          expected_dim = fft_length / 2 + 1;
          fft_string = "IRFFT";
        }
        Location loc = op.getLoc();
    
        // The inner-most dim cannot be dynamic.
        if (input_ty.isDynamicDim(input_shape.size() - 1)) {
          return failure();
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
Back to top