Home
last modified time | relevance | path

Searched refs:fft_shape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfft_ops.cc54 uint64 fft_shape[3] = {0, 0, 0}; in Compute() local
68 fft_shape[i] = fft_length_as_vec(i); in Compute()
74 !IsForward() && inner_most ? fft_shape[i] / 2 + 1 : fft_shape[i]; in Compute()
85 uint64 dim = IsForward() && inner_most && fft_shape[i] != 0 in Compute()
86 ? fft_shape[i] / 2 + 1 in Compute()
87 : fft_shape[i]; in Compute()
92 fft_shape[i] = in Compute()
129 DoFFT(ctx, in, fft_shape, out); in Compute()
138 virtual void DoFFT(OpKernelContext* ctx, const Tensor& in, uint64* fft_shape,
154 void DoFFT(OpKernelContext* ctx, const Tensor& in, uint64* fft_shape, in DoFFT() argument
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_fft_impl.h50 const std::array<int64, 3> fft_shape = { in EigenFftC2C() local
56 dims[i + 1] = fft_shape[i]; in EigenFftC2C()
73 const std::array<int64, 3> fft_shape = { in EigenFftR2C() local
81 in_dims[i + 1] = fft_shape[i]; in EigenFftR2C()
82 out_dims[i + 1] = i == FFTRank - 1 ? fft_shape[i] / 2 + 1 : fft_shape[i]; in EigenFftR2C()
112 const std::array<int64, 3> fft_shape = { in EigenFftC2R() local
120 in_dims[i + 1] = i == FFTRank - 1 ? fft_shape[i] / 2 + 1 : fft_shape[i]; in EigenFftC2R()
121 out_dims[i + 1] = fft_shape[i]; in EigenFftC2R()
139 neg_sizes[FFTRank] = fft_shape[FFTRank - 1] - in_dims[FFTRank]; in EigenFftC2R()
/external/tensorflow/tensorflow/python/ops/signal/
Dfft_ops.py36 fft_shape = input_tensor.get_shape()[-fft_rank:]
39 if not fft_shape.is_fully_defined():
43 return _ops.convert_to_tensor(fft_shape.as_list(), _dtypes.int32)
49 fft_shape = input_tensor.get_shape()[-fft_rank:]
52 if not fft_shape.is_fully_defined():
58 fft_length = fft_shape.as_list()
66 fft_shape = _tensor_util.constant_value_as_shape(fft_length)
76 if fft_shape.is_fully_defined() and input_tensor.shape.ndims is not None:
78 input_fft_shape = input_tensor.shape[-fft_shape.ndims:]
83 fft_shape = fft_shape[:-1].concatenate(
[all …]
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc1531 Tensor fft_shape = test::AsTensor<int32>(AsInt32s({dims[dims.size() - 1]})); in TEST_F() local
1533 OpTestBuilder("RFFT").RandomInput(DT_FLOAT, dims).Input(fft_shape)); in TEST_F()
1540 Tensor fft_shape = test::AsTensor<int32>( in TEST_F() local
1543 OpTestBuilder("RFFT2D").RandomInput(DT_FLOAT, dims).Input(fft_shape)); in TEST_F()
1550 Tensor fft_shape = test::AsTensor<int32>(AsInt32s( in TEST_F() local
1553 OpTestBuilder("RFFT3D").RandomInput(DT_FLOAT, dims).Input(fft_shape)); in TEST_F()
1562 Tensor fft_shape = test::AsTensor<int32>(AsInt32s({orig_size})); in TEST_F() local
1565 .Input(fft_shape)); in TEST_F()
1575 Tensor fft_shape = test::AsTensor<int32>(AsInt32s({orig_size})); in TEST_F() local
1578 .Input(fft_shape)); in TEST_F()
[all …]