Searched defs:offset_calculator (Results 1 – 3 of 3) sorted by relevance
12 int numel, T* out, const T* x1, const T* x2, OffsetCalc offset_calculator, in cross_kernel()39 auto offset_calculator = make_element_offset_calculator<3>(iter); in launch_cross_kernel() local
81 OffsetCalc offset_calculator, in glu_backward_kernel()116 const auto offset_calculator = make_element_offset_calculator<3>(iter); in launch_glu_backward_kernel() local
118 const auto offset_calculator = make_offset_calculator<2>(iter); in unpack_pivots_cuda_kernel() local