Searched defs:b_input (Results 1 – 7 of 7) sorted by relevance
73 const T b_input = ldg(b + ((ADJ_B) ? j * b_cols + k : k * b_cols + j)); in SparseTensorDenseMatMulKernel() local
273 const auto& b_input = op_info.inputs(1); in IsEinsumCorrectlyFormed() local1157 const auto& b_input = op_info.inputs(1); in GenerateBatchMatmulContextFromEinsum() local1269 const auto& b_input = op_info.inputs(1); in CountBatchMatMulOperations() local
5363 let b_input: [$ety; $ecount] = [$($in_b),*]; localVariable