Searched defs:b_matrix (Results 1 – 4 of 4) sorted by relevance
203 const CSRSparseMatrix* b_matrix; in Compute() local
316 const CSRSparseMatrix* b_matrix; in Compute() local
1052 auto& b_matrix = op_info.inputs(1); in CountMatMulOperations() local1179 OpInfo::TensorProperties* b_matrix = batch_matmul_op_info.add_inputs(); in GenerateBatchMatmulContextFromEinsum() local1335 OpInfo::TensorProperties* b_matrix = matmul_op_info.add_inputs(); in CountBatchMatMulOperations() local1837 auto b_matrix = op_info.inputs(3); in PredictSparseTensorDenseMatMul() local
2500 const DeviceMemory<T> &b_matrix = *b_ptrs_to_wrappers[b]; in DoBlasInternalImpl() local2635 const auto *b_matrix = in DoBlasInternalImpl() local