Searched refs:dim_offset (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dfft_thunk.cc152 auto dim_offset = input_shape_.dimensions_size() - fft_rank + i; local
154 input_embed[i] = input_shape_.dimensions(dim_offset);
155 input_distance *= input_shape_.dimensions(dim_offset);
156 output_embed[i] = output_shape_.dimensions(dim_offset);
157 output_distance *= output_shape_.dimensions(dim_offset);
/external/tensorflow/tensorflow/core/kernels/
H A Dfft_ops.cc388 auto dim_offset = input_shape.dims() - fft_rank + i; variable
389 input_embed[i] = input_shape.dim_size(dim_offset);
390 input_distance *= input_shape.dim_size(dim_offset);
391 output_embed[i] = output_shape.dim_size(dim_offset);
392 output_distance *= output_shape.dim_size(dim_offset);

Completed in 141 milliseconds