Searched defs:arg_out_ptr (Results 1 – 1 of 1) sorted by relevance
212 index_t* arg_out_ptr = arg_out_data + m * K; in spmm_reduce_arg_kernel_impl() local324 index_t* arg_out_ptr = arg_out_data + m * K; in spmm_reduce_backward_input_arg_kernel_impl() local406 const index_t* arg_out_ptr = arg_out_data + m * K; in spmm_reduce_backward_other_arg_kernel_impl() local