Searched defs:b_matrix (Results 1 – 4 of 4) sorted by relevance
1064 auto& b_matrix = op_info.inputs(1); in CountMatMulOperations() local1191 OpInfo::TensorProperties* b_matrix = batch_matmul_op_info.add_inputs(); in GenerateBatchMatmulContextFromEinsum() local1347 OpInfo::TensorProperties* b_matrix = matmul_op_info.add_inputs(); in CountBatchMatMulOperations() local1871 auto b_matrix = op_info.inputs(3); in PredictSparseTensorDenseMatMul() local
1067 const auto *b_matrix = reinterpret_cast<const __nv_bfloat16 *>( in DoBlasInternalImpl() local1313 const DeviceMemory<T> &b_matrix = *b_ptrs_to_wrappers[b]; in DoBlasInternalImpl() local1445 const auto *b_matrix = reinterpret_cast<const __nv_bfloat16 *>( in DoBlasInternalImpl() local1480 const auto *b_matrix = reinterpret_cast<const __half *>( in DoBlasInternalImpl() local
204 const CSRSparseMatrix* b_matrix; in Compute() local
316 const CSRSparseMatrix* b_matrix; in Compute() local