Searched defs:einsum_context (Results 1 – 1 of 1) sorted by relevance
251 bool IsEinsumCorrectlyFormed(const OpContext& einsum_context) { in IsEinsumCorrectlyFormed()1111 const OpContext& einsum_context, OpContext* batch_matmul_context, in GenerateBatchMatmulContextFromEinsum()