- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for rfft2d (0.09 sec)
-
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
} // ----- // CHECK-LABEL: func @rfft_1D func.func @rfft_1D(%arg0: tensor<8xf32>) -> tensor<5xcomplex<f32>> { %fftlength = "tf.Const"() {value = dense<[8]> : tensor<1xi32>} : () -> (tensor<1xi32>) // CHECK: "mhlo.fft"(%arg0) <{fft_length = dense<8> : tensor<1xi64>, fft_type = #mhlo<fft_type RFFT>}> : (tensor<8xf32> %0 = "tf.RFFT"(%arg0, %fftlength) : (tensor<8xf32>, tensor<1xi32>) -> tensor<5xcomplex<f32>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
symbolizeFftType(fft_string).value()), rewriter.getI64TensorAttr(fft_length)); return success(); } }; using ConvertRFFTOp = ConvertFFTOp<TF::RFFTOp>; using ConvertIRFFTOp = ConvertFFTOp<TF::IRFFTOp>; // The base class to convert TensorFlow FusedBatchNormGrad*Op to HLO // BatchNormGradOp for training and a sequence of binary ops for inference.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)