Searched defs:mat1_ptr (Results 1 – 4 of 4) sorted by relevance
1189 const Dtype* mat1_ptr, in gemm_and_bias()1397 const void* mat1_ptr, in scaled_gemm()1573 const int8_t* mat1_ptr, in int8_gemm()
57 const scalar_t* mat1_ptr = mat1_data + b * M * K; in sampled_addmm_sparse_csr_kernel_impl() local
466 const scalar_t* mat1_ptr = args.mata->const_data_ptr<scalar_t>(); in addmm_out_cuda_impl() local
1274 mat1_ptr, argument