Searched defs:b_matrix (Results 1 – 2 of 2) sorted by relevance
699 auto& b_matrix = op_info.inputs(1); in CountMatMulOperations() local849 OpInfo::TensorProperties* b_matrix = matmul_op_info.add_inputs(); in CountBatchMatMulOperations() local1246 auto b_matrix = op_info.inputs(3); in PredictSparseTensorDenseMatMul() local
2259 const DeviceMemory<T> &b_matrix = *b_ptrs_to_wrappers[b]; in DoBlasGemmBatchedInternal() local2397 const auto *b_matrix = in DoBlasGemmStridedBatched() local