| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| runtime_fft.cc | 30 int32 fft_rank, int64 input_batch, int64 fft_length0, int64 fft_length1,
35 operand, fft_type, fft_rank, input_batch,
|
| runtime_fft.h | 25 void* operand, tensorflow::int32 fft_type, tensorflow::int32 fft_rank,
|
| runtime_fft_impl.h | 215 int32 fft_type, int32 fft_rank, int64 input_batch,
217 switch (fft_rank) {
233 LOG(FATAL) << "Unsupported FFT rank " << fft_rank;
|
| ir_emitter.cc | [all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| spectral_ops.py | 47 def _infer_fft_length_for_rfft(input_tensor, fft_rank):
49 # A TensorShape for the inner fft_rank dimensions.
50 fft_shape = input_tensor.get_shape()[-fft_rank:]
54 return _array_ops.shape(input_tensor)[-fft_rank:]
60 def _infer_fft_length_for_irfft(input_tensor, fft_rank):
62 # A TensorShape for the inner fft_rank dimensions.
63 fft_shape = input_tensor.get_shape()[-fft_rank:]
67 fft_length = _array_ops.unstack(_array_ops.shape(input_tensor)[-fft_rank:])
78 def _maybe_pad_for_rfft(input_tensor, fft_rank, fft_length, is_reverse=False):
79 """Pads `input_tensor` to `fft_length` on its inner-most `fft_rank` dims.""
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| fft_ops.cc | 44 const int fft_rank = Rank(); variable
46 ctx, input_shape.dims() >= fft_rank,
47 errors::InvalidArgument("Input must have rank of at least ", fft_rank,
60 fft_length.shape().dim_size(0) == fft_rank,
62 fft_rank, "]"));
65 for (int i = 0; i < fft_rank; ++i) {
70 bool inner_most = (i == fft_rank - 1);
73 auto input_index = input_shape.dims() - fft_rank + i;
86 output_shape.set_dim(output_shape.dims() - fft_rank + i, dim);
89 for (int i = 0; i < fft_rank; ++i)
375 const int fft_rank = Rank(); variable
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| fft_thunk.cc | 137 const int64 fft_rank = fft_length_.size(); local
138 CHECK_LE(fft_rank, 3);
140 for (int i = 0; i < input_shape_.dimensions_size() - fft_rank; ++i) {
151 for (int i = 0; i < fft_rank; ++i) {
152 auto dim_offset = input_shape_.dimensions_size() - fft_rank + i;
163 stream, fft_rank, fft_length, input_embed, input_stride,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| fft_ops.cc | 42 int fft_rank)
43 : XlaOpKernel(ctx), fft_type_(fft_type), fft_rank_(fft_rank) {}
83 : GenericFftOp(ctx, /*fft_type=*/FftType::FFT, /*fft_rank=*/FFTRank) {}
93 : GenericFftOp(ctx, /*fft_type=*/FftType::IFFT, /*fft_rank=*/FFTRank) {}
103 : GenericFftOp(ctx, /*fft_type=*/FftType::RFFT, /*fft_rank=*/FFTRank) {}
113 : GenericFftOp(ctx, /*fft_type=*/FftType::IRFFT, /*fft_rank=*/FFTRank) {}
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| shape_inference.cc | 1749 const int64 fft_rank = fft_length.size(); local
[all...] |
