Searched defs:mat2_ptr (Results 1 – 6 of 6) sorted by relevance
80 const auto mat2_ptr = get_nested_tensor_impl(mat2); in matmul_with_bmm_nested() local238 const auto mat2_ptr = get_nested_tensor_impl(mat2_contig); in matmul_nested() local
1191 const Dtype* mat2_ptr, in gemm_and_bias()1401 const void* mat2_ptr, in scaled_gemm()1575 const int8_t* mat2_ptr, in int8_gemm()
58 const scalar_t* mat2_ptr = mat2_data + b * N * K; in sampled_addmm_sparse_csr_kernel_impl() local
289 auto mat2_ptr = mat2.is_nested() ? get_nested_tensor_impl(mat2) : mat2.unsafeGetTensorImpl(); in bmm_nested_cuda() local
467 const scalar_t* mat2_ptr = args.matb->const_data_ptr<scalar_t>(); in addmm_out_cuda_impl() local
1255 def _sampled_addmm_kernel(