Searched defs:b_input (Results 1 – 2 of 2) sorted by relevance
73 const T b_input = ldg(b + ((ADJ_B) ? j * b_cols + k : k * b_cols + j)); in SparseTensorDenseMatMulKernel() local
272 const auto& b_input = op_info.inputs(1); in IsEinsumCorrectlyFormed() local1156 const auto& b_input = op_info.inputs(1); in GenerateBatchMatmulContextFromEinsum() local1268 const auto& b_input = op_info.inputs(1); in CountBatchMatMulOperations() local