Searched defs:a_input (Results 1 – 2 of 2) sorted by relevance
69 const T a_input = ldg(a_values + a_ix); in SparseTensorDenseMatMulKernel() local
272 const auto& a_input = op_info.inputs(0); in IsEinsumCorrectlyFormed() local1156 const auto& a_input = op_info.inputs(0); in GenerateBatchMatmulContextFromEinsum() local1268 const auto& a_input = op_info.inputs(0); in CountBatchMatMulOperations() local