Searched defs:fft_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_fft_impl.h | 50 const std::array<int64, 3> fft_shape = { in EigenFftC2C() local
73 const std::array<int64, 3> fft_shape = { in EigenFftR2C() local
112 const std::array<int64, 3> fft_shape = { in EigenFftC2R() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fft_ops.cc | 54 uint64 fft_shape[3] = {0, 0, 0}; in Compute() local
154 void DoFFT(OpKernelContext* ctx, const Tensor& in, uint64* fft_shape, in DoFFT()
208 void DoRealForwardFFT(OpKernelContext* ctx, uint64* fft_shape, in DoRealForwardFFT()
243 void DoRealBackwardFFT(OpKernelContext* ctx, uint64* fft_shape, in DoRealBackwardFFT()
425 void DoFFT(OpKernelContext* ctx, const Tensor& in, uint64* fft_shape, in DoFFT()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1531 Tensor fft_shape = test::AsTensor<int32>(AsInt32s({dims[dims.size() - 1]})); in TEST_F() local
1540 Tensor fft_shape = test::AsTensor<int32>( in TEST_F() local
1550 Tensor fft_shape = test::AsTensor<int32>(AsInt32s( in TEST_F() local
1562 Tensor fft_shape = test::AsTensor<int32>(AsInt32s({orig_size})); in TEST_F() local
1575 Tensor fft_shape = test::AsTensor<int32>(AsInt32s({orig_size})); in TEST_F() local
1588 Tensor fft_shape = test::AsTensor<int32>(AsInt32s({orig_size})); in TEST_F() local
|