Searched defs:offset_calc (Results 1 – 7 of 7) sorted by relevance
83 auto offset_calc = make_offset_calculator<3>(iter); in gpu_index_kernel() local129 const auto offset_calc = make_offset_calculator<2>(iter); in index_fill_kernel_impl() local167 const auto offset_calc = make_offset_calculator<3>(iter); in index_copy_kernel_impl() local293 const auto offset_calc = make_offset_calculator<2>(iter); in cuda_take_put_kernel() local440 const auto offset_calc = make_offset_calculator<2, /*signed_strides=*/true>(iter); in flip_kernel_impl() local
273 auto offset_calc = ::make_offset_calculator<traits::arity + 1>(iter); in gpu_kernel_impl_nocast() local337 auto offset_calc = ::make_offset_calculator<traits::arity + 1>(iter); in gpu_kernel_impl() local
139 auto offset_calc = make_offset_calculator<3>(iter); in operator ()() local360 auto offset_calc = make_offset_calculator<2>(iter); in operator ()() local
65 auto offset_calc = make_offset_calculator<3>(iter); in _compute_linear_combination_internal_kernel() local
86 auto offset_calc = make_offset_calculator<3>(iter); in _unfold_backward_internal_kernel() local
98 auto offset_calc = make_offset_calculator<6>(iter); in binary_op_intersection_kernel() local
162 auto offset_calc = make_offset_calculator<1>(iter); in distribution_nullary_kernel() local