Searched refs:batch_matmul_op_context (Results 1 – 1 of 1) sorted by relevance
1811 OpContext batch_matmul_op_context; in PredictEinsum() local1814 op_context, &batch_matmul_op_context, &found_unknown_shapes); in PredictEinsum()1822 return PredictNodeCosts(batch_matmul_op_context, node_costs); in PredictEinsum()