Searched defs:fft_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_fft_impl.h | 42 const std::array<int64, 3> fft_shape = { in EigenFftC2C() local
65 const std::array<int64, 3> fft_shape = { in EigenFftR2C() local
104 const std::array<int64, 3> fft_shape = { in EigenFftC2R() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fft_ops.cc | 52 uint64 fft_shape[3] = {0, 0, 0}; in Compute() local
125 void DoFFT(OpKernelContext* ctx, const Tensor& in, uint64* fft_shape, in DoFFT()
376 void DoFFT(OpKernelContext* ctx, const Tensor& in, uint64* fft_shape, in DoFFT()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1530 Tensor fft_shape = test::AsTensor<int32>(AsInt32s({dims[dims.size() - 1]})); in TEST_F() local
1539 Tensor fft_shape = test::AsTensor<int32>( in TEST_F() local
1549 Tensor fft_shape = test::AsTensor<int32>(AsInt32s( in TEST_F() local
1561 Tensor fft_shape = test::AsTensor<int32>(AsInt32s({orig_size})); in TEST_F() local
1574 Tensor fft_shape = test::AsTensor<int32>(AsInt32s({orig_size})); in TEST_F() local
1587 Tensor fft_shape = test::AsTensor<int32>(AsInt32s({orig_size})); in TEST_F() local
|