Searched refs:batch_matmul_context (Results 1 – 2 of 2) sorted by relevance
241 OpContext* batch_matmul_context,
1111 const OpContext& einsum_context, OpContext* batch_matmul_context, in GenerateBatchMatmulContextFromEinsum() argument1134 if (batch_matmul_context == nullptr) { in GenerateBatchMatmulContextFromEinsum()1229 *batch_matmul_context = einsum_context; in GenerateBatchMatmulContextFromEinsum()1230 batch_matmul_context->op_info = batch_matmul_op_info; in GenerateBatchMatmulContextFromEinsum()