Searched refs:einsum_context (Results 1 – 2 of 2) sorted by relevance
240 bool GenerateBatchMatmulContextFromEinsum(const OpContext& einsum_context,
251 bool IsEinsumCorrectlyFormed(const OpContext& einsum_context) { in IsEinsumCorrectlyFormed() argument252 const auto& op_info = einsum_context.op_info; in IsEinsumCorrectlyFormed()1111 const OpContext& einsum_context, OpContext* batch_matmul_context, in GenerateBatchMatmulContextFromEinsum() argument1138 if (!IsEinsumCorrectlyFormed(einsum_context)) return false; in GenerateBatchMatmulContextFromEinsum()1139 const auto& op_info = einsum_context.op_info; in GenerateBatchMatmulContextFromEinsum()1229 *batch_matmul_context = einsum_context; in GenerateBatchMatmulContextFromEinsum()