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
271 const auto& a_input = op_info.inputs(0); in IsEinsumCorrectlyFormed() local1155 const auto& a_input = op_info.inputs(0); in GenerateBatchMatmulContextFromEinsum() local1267 const auto& a_input = op_info.inputs(0); in CountBatchMatMulOperations() local