OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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