Searched defs:a_matrix (Results 1 – 6 of 6) sorted by relevance
46 const CSRSparseMatrix* a_matrix; in Compute() local
202 const CSRSparseMatrix* a_matrix; in Compute() local
315 const CSRSparseMatrix* a_matrix; in Compute() local
503 const CSRSparseMatrix* a_matrix; in Compute() local
1062 auto& a_matrix = op_info.inputs(0); in CountMatMulOperations() local1186 OpInfo::TensorProperties* a_matrix = batch_matmul_op_info.add_inputs(); in GenerateBatchMatmulContextFromEinsum() local1338 OpInfo::TensorProperties* a_matrix = matmul_op_info.add_inputs(); in CountBatchMatMulOperations() local
2197 const auto *a_matrix = reinterpret_cast<const __nv_bfloat16 *>( in DoBlasInternalImpl() local2433 const DeviceMemory<T> &a_matrix = *a_ptrs_to_wrappers[b]; in DoBlasInternalImpl() local2565 const auto *a_matrix = reinterpret_cast<const __nv_bfloat16 *>( in DoBlasInternalImpl() local2601 const auto *a_matrix = reinterpret_cast<const __half *>( in DoBlasInternalImpl() local