/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
runtime_single_threaded_fft.cc | 27 int32 fft_rank, int64 input_batch, int64 fft_length0, int64 fft_length1, 30 fft_rank, input_batch, fft_length0, fft_length1, local
|
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_single_threaded_fft.h | 25 void* operand, tensorflow::int32 fft_type, tensorflow::int32 fft_rank,
|
runtime_fft_impl.h | 208 int32 fft_type, int32 fft_rank, int64 input_batch, 210 switch (fft_rank) {
|
ir_emitter.cc | 1296 const int fft_rank = fft_length.size(); local [all...] |
/external/tensorflow/tensorflow/python/ops/signal/ |
fft_ops.py | 31 def _infer_fft_length_for_rfft(input_tensor, fft_rank): 33 # A TensorShape for the inner fft_rank dimensions. 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): 46 # A TensorShape for the inner fft_rank dimensions. 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): 63 """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) 384 const int fft_rank = Rank(); variable [all...] |
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
fft_thunk.cc | 122 const int64 fft_rank = fft_length_.size(); local 123 CHECK_LE(fft_rank, 3); 125 for (int i = 0; i < input_shape_.dimensions_size() - fft_rank; ++i) { 136 for (int i = 0; i < fft_rank; ++i) { 137 auto dim_offset = input_shape_.dimensions_size() - fft_rank + i; 148 stream, fft_rank, fft_length, input_embed, input_stride,
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
fft_ops.cc | 43 int fft_rank) 44 : XlaOpKernel(ctx), fft_type_(fft_type), fft_rank_(fft_rank) {} 114 : GenericFftOp(ctx, /*fft_type=*/FftType::FFT, /*fft_rank=*/FFTRank) {} 126 : GenericFftOp(ctx, /*fft_type=*/FftType::IFFT, /*fft_rank=*/FFTRank) {} 139 : GenericFftOp(ctx, /*fft_type=*/FftType::RFFT, /*fft_rank=*/FFTRank) {} 151 : GenericFftOp(ctx, /*fft_type=*/FftType::IRFFT, /*fft_rank=*/FFTRank) {}
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
shape_inference.cc | 1821 const int64 fft_rank = fft_length.size(); local [all...] |