Searched defs:a_input (Results 1 – 1 of 1) sorted by relevance
271 const auto& a_input = op_info.inputs(0); in IsEinsumCorrectlyFormed() local1144 const auto& a_input = op_info.inputs(0); in GenerateBatchMatmulContextFromEinsum() local1256 const auto& a_input = op_info.inputs(0); in CountBatchMatMulOperations() local