Searched defs:fft_shape (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | fft_ops.cc | 55 uint64 fft_shape[3] = {0, 0, 0}; in Compute() local
159 void DoFFT(OpKernelContext* ctx, const Tensor& in, uint64* fft_shape, in DoFFT()
213 void DoRealForwardFFT(OpKernelContext* ctx, uint64* fft_shape, in DoRealForwardFFT()
251 void DoRealBackwardFFT(OpKernelContext* ctx, uint64* fft_shape, in DoRealBackwardFFT()
436 void DoFFT(OpKernelContext* ctx, const Tensor& in, uint64* fft_shape, in DoFFT()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | runtime_fft_impl.h | 46 const std::array<int64_t, 3> fft_shape = { in EigenFftC2C() local
69 const std::array<int64_t, 3> fft_shape = { in EigenFftR2C() local
108 const std::array<int64_t, 3> fft_shape = { in EigenFftC2R() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | randomized_tests.cc | 2257 Tensor fft_shape = test::AsTensor<int32>(AsInt32s({dims[dims.size() - 1]})); in TEST_F() local
2266 Tensor fft_shape = test::AsTensor<int32>( in TEST_F() local
2276 Tensor fft_shape = test::AsTensor<int32>(AsInt32s( in TEST_F() local
2288 Tensor fft_shape = test::AsTensor<int32>(AsInt32s({orig_size})); in TEST_F() local
2301 Tensor fft_shape = test::AsTensor<int32>(AsInt32s({orig_size})); in TEST_F() local
2314 Tensor fft_shape = test::AsTensor<int32>(AsInt32s({orig_size})); in TEST_F() local
|