HomeSort by relevance Sort by last modified time
    Searched defs:fft_rank (Results 1 - 3 of 3) sorted by null

  /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/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/
shape_inference.cc 1749 const int64 fft_rank = fft_length.size(); local
    [all...]

Completed in 426 milliseconds