Searched defs:dim_offset (Results 1 – 9 of 9) sorted by relevance
42 auto dim_offset = ndims_output - ndims_input; in ExpandInputDimsToOutputShape() local
130 auto dim_offset = input_shape.dimensions_size() - fft_rank + i; in RunFft() local
2391 llvm::Value* dim_offset = in EmitScatter() local
78 int64_t dim_offset = 0; in MergeSliceAndSelectToIndices() local
330 orig, dim_offset=0, keepdim_offset=1, dim_name="dim", single_dim=False, reduce=True argument
457 auto dim_offset = input_shape.dims() - fft_rank + i; in DoFFT() local
157 const int64_t dim_offset = packed_dim + 1; in calculate_padded_sizes() local
2870 int64_t dim_offset = 0; member
392 unsigned dim_offset : 5; member