Searched refs:batch2_ptr (Results 1 – 1 of 1) sorted by relevance
560 const scalar_t* batch2_ptr = batch2_->const_data_ptr<scalar_t>(); in baddbmm_out_cuda_impl() local571 batch2_ptr, ldb, in baddbmm_out_cuda_impl()580 batch2_ptr, ldb, batch2_->strides()[0], in baddbmm_out_cuda_impl()