Searched refs:rfft2d (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | rfft2d.cc | 35 namespace rfft2d { namespace
465 static TfLiteRegistration r = {rfft2d::Init, rfft2d::Free, rfft2d::Prepare, in Register_RFFT2D()
466 rfft2d::Eval}; in Register_RFFT2D()
|
| D | BUILD | 643 "rfft2d.cc",
|
| /external/tensorflow/tensorflow/python/ops/signal/ |
| D | fft_ops.py | 190 rfft2d = _rfft_wrapper(gen_spectral_ops.rfft2d, 2, "rfft2d") variable
192 dispatch.add_dispatch_support(rfft2d))
465 _ops.RegisterGradient("IRFFT2D")(_irfft_grad_helper(2, rfft2d))
|
| D | signal.py | 48 from tensorflow.python.ops.signal.fft_ops import rfft2d
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.spectral.pbtxt | 52 name: "rfft2d"
|
| D | tensorflow.signal.pbtxt | 108 name: "rfft2d"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.signal.pbtxt | 108 name: "rfft2d"
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | fft_test.py | 146 return signal.rfft2d(x, fft_length=[x.shape[-2], x.shape[-1]])
|
| /external/tensorflow/tensorflow/python/kernel_tests/signal/ |
| D | fft_ops_test.py | 336 return fft_ops.rfft2d
550 2: [gen_spectral_ops.rfft2d, gen_spectral_ops.irfft2d],
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 1419 TF::RFFT2DOp rfft2d = rewriter.create<TF::RFFT2DOp>( in matchAndRewrite() local
1426 rfft_op.getLoc(), output_type, rfft2d.getResult(), squeeze_dim); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | legalize-tf.mlir | 1986 func @rfft2d(%arg0: tensor<10x20x10x30xf32>, %arg1: tensor<2xi32>) -> tensor<10x20x10x30xcomplex<f3…
1989 // CHECK-LABEL: rfft2d
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | control_flow_ops_test.py | 2360 (fft_ops.rfft2d,),
|
| D | pfor.py | 4638 @RegisterPForWithArgs("RFFT2D", gen_spectral_ops.rfft2d, "Tcomplex")
|