| /external/tensorflow/tensorflow/python/ops/signal/ |
| D | fft_ops.py | 31 def _infer_fft_length_for_rfft(input_tensor, fft_rank): argument
34 fft_shape = input_tensor.get_shape()[-fft_rank:]
38 return _array_ops.shape(input_tensor)[-fft_rank:]
44 def _infer_fft_length_for_irfft(input_tensor, fft_rank): argument
47 fft_shape = input_tensor.get_shape()[-fft_rank:]
51 fft_length = _array_ops.unstack(_array_ops.shape(input_tensor)[-fft_rank:])
62 def _maybe_pad_for_rfft(input_tensor, fft_rank, fft_length, is_reverse=False): argument
96 input_fft_shape = _array_ops.shape(input_tensor)[-fft_rank:]
97 outer_dims = _math_ops.maximum(0, input_rank - fft_rank)
110 def _rfft_wrapper(fft_fn, fft_rank, default_name): argument
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fft_ops.cc | 44 const int fft_rank = Rank(); in Compute() local
46 ctx, input_shape.dims() >= fft_rank, in Compute()
47 errors::InvalidArgument("Input must have rank of at least ", fft_rank, in Compute()
60 fft_length.shape().dim_size(0) == fft_rank, in Compute()
62 fft_rank, "]")); in Compute()
65 for (int i = 0; i < fft_rank; ++i) { in Compute()
70 bool inner_most = (i == fft_rank - 1); in Compute()
73 auto input_index = input_shape.dims() - fft_rank + i; in Compute()
86 output_shape.set_dim(output_shape.dims() - fft_rank + i, dim); in Compute()
89 for (int i = 0; i < fft_rank; ++i) { in Compute()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | fft_thunk.cc | 122 const int64 fft_rank = fft_length_.size(); in ExecuteOnStream() local
123 CHECK_LE(fft_rank, 3); in ExecuteOnStream()
125 for (int i = 0; i < input_shape_.dimensions_size() - fft_rank; ++i) { in ExecuteOnStream()
136 for (int i = 0; i < fft_rank; ++i) { in ExecuteOnStream()
137 auto dim_offset = input_shape_.dimensions_size() - fft_rank + i; in ExecuteOnStream()
148 stream, fft_rank, fft_length, input_embed, input_stride, in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_single_threaded_fft.cc | 27 int32 fft_rank, int64 input_batch, int64 fft_length0, int64 fft_length1, in __xla_cpu_runtime_EigenSingleThreadedFft() argument
30 fft_rank, input_batch, fft_length0, fft_length1, in __xla_cpu_runtime_EigenSingleThreadedFft()
|
| D | runtime_fft.cc | 30 int32 fft_rank, int64 input_batch, int64 fft_length0, int64 fft_length1, in __xla_cpu_runtime_EigenFft() argument
35 operand, fft_type, fft_rank, input_batch, in __xla_cpu_runtime_EigenFft()
|
| D | runtime_single_threaded_fft.h | 25 void* operand, tensorflow::int32 fft_type, tensorflow::int32 fft_rank,
|
| D | runtime_fft.h | 25 void* operand, tensorflow::int32 fft_type, tensorflow::int32 fft_rank,
|
| D | runtime_fft_impl.h | 208 int32 fft_type, int32 fft_rank, int64 input_batch, in EigenFftImpl() argument
210 switch (fft_rank) { in EigenFftImpl()
|
| D | ir_emitter.cc | 1296 const int fft_rank = fft_length.size(); in HandleFft() local
1301 b_.getInt32(fft_rank), b_.getInt64(input_batch), in HandleFft()
1302 b_.getInt64(fft_rank > 0 ? fft_length[0] : 0), in HandleFft()
1303 b_.getInt64(fft_rank > 1 ? fft_length[1] : 0), in HandleFft()
1304 b_.getInt64(fft_rank > 2 ? fft_length[2] : 0)}); in HandleFft()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | fft_ops.cc | 43 int fft_rank) in GenericFftOp() argument
44 : XlaOpKernel(ctx), fft_type_(fft_type), fft_rank_(fft_rank) {} in GenericFftOp()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 1821 const int64 fft_rank = fft_length.size(); in InferFftShape() local
1822 if (fft_rank < 1 || fft_rank > 3) { in InferFftShape()
1823 return InvalidArgument("FFT only supports ranks 1-3; got %d.", fft_rank); in InferFftShape()
1826 if (x.dimensions_size() < fft_rank) { \ in InferFftShape()
1830 fft_rank, x.dimensions_size()); \ in InferFftShape()
1848 for (int i = 0; i < fft_rank; i++) { in InferFftShape()
1849 if (in.dimensions(in.dimensions_size() - fft_rank + i) != in InferFftShape()
1854 in.dimensions_size() - fft_rank + i, in InferFftShape()
1855 in.dimensions(in.dimensions_size() - fft_rank + i), in InferFftShape()
1864 fft_length[fft_rank - 1] / 2 + 1); in InferFftShape()
[all …]
|