Home
last modified time | relevance | path

Searched refs:dim_offset (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfft_thunk.cc137 auto dim_offset = input_shape_.dimensions_size() - fft_rank + i; in ExecuteOnStream() local
139 input_embed[i] = input_shape_.dimensions(dim_offset); in ExecuteOnStream()
140 input_distance *= input_shape_.dimensions(dim_offset); in ExecuteOnStream()
141 output_embed[i] = output_shape_.dimensions(dim_offset); in ExecuteOnStream()
142 output_distance *= output_shape_.dimensions(dim_offset); in ExecuteOnStream()
Dir_emitter_unnested.cc1156 llvm::Value* dim_offset = in EmitScatter() local
1158 input_window_multidim[operand_dim] = dim_offset; in EmitScatter()
/external/tensorflow/tensorflow/core/kernels/
Dfft_ops.cc397 auto dim_offset = input_shape.dims() - fft_rank + i; in DoFFT() local
398 input_embed[i] = input_shape.dim_size(dim_offset); in DoFFT()
399 input_distance *= input_shape.dim_size(dim_offset); in DoFFT()
400 output_embed[i] = output_shape.dim_size(dim_offset); in DoFFT()
401 output_distance *= output_shape.dim_size(dim_offset); in DoFFT()