Searched refs:dim_offset (Results 1 – 4 of 4) sorted by relevance
161 auto dim_offset = input_shape_.dimensions_size() - fft_rank + i; in ExecuteOnStream() local163 input_embed[i] = input_shape_.dimensions(dim_offset); in ExecuteOnStream()164 input_distance *= input_shape_.dimensions(dim_offset); in ExecuteOnStream()165 output_embed[i] = output_shape_.dimensions(dim_offset); in ExecuteOnStream()166 output_distance *= output_shape_.dimensions(dim_offset); in ExecuteOnStream()
2678 llvm::Value* dim_offset = in EmitScatter() local2680 input_window_multidim[operand_dim] = dim_offset; in EmitScatter()
168 auto dim_offset = ndims_output - ndims_input; in ExpandInputDimsToOutputShape() local169 DCHECK(dim_offset > 0); in ExpandInputDimsToOutputShape()174 reshaped_dims->at(dim_idx + dim_offset) = input_dims[dim_idx]; in ExpandInputDimsToOutputShape()
446 auto dim_offset = input_shape.dims() - fft_rank + i; in DoFFT() local447 input_embed[i] = input_shape.dim_size(dim_offset); in DoFFT()448 input_distance *= input_shape.dim_size(dim_offset); in DoFFT()449 output_embed[i] = output_shape.dim_size(dim_offset); in DoFFT()450 output_distance *= output_shape.dim_size(dim_offset); in DoFFT()