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
500 const CSRSparseMatrix* a_matrix; in Compute() local
1051 auto& a_matrix = op_info.inputs(0); in CountMatMulOperations() local1175 OpInfo::TensorProperties* a_matrix = batch_matmul_op_info.add_inputs(); in GenerateBatchMatmulContextFromEinsum() local1327 OpInfo::TensorProperties* a_matrix = matmul_op_info.add_inputs(); in CountBatchMatMulOperations() local
2499 const DeviceMemory<T> &a_matrix = *a_ptrs_to_wrappers[b]; in DoBlasInternalImpl() local2633 const auto *a_matrix = in DoBlasInternalImpl() local