Searched refs:matmul_op_info (Results 1 – 1 of 1) sorted by relevance
1310 OpInfo matmul_op_info; in CountBatchMatMulOperations() local1311 matmul_op_info.set_op("MatMul"); in CountBatchMatMulOperations()1318 (*matmul_op_info.mutable_attr())["transpose_a"] = transpose_a; in CountBatchMatMulOperations()1325 (*matmul_op_info.mutable_attr())["transpose_b"] = transpose_b; in CountBatchMatMulOperations()1327 OpInfo::TensorProperties* a_matrix = matmul_op_info.add_inputs(); in CountBatchMatMulOperations()1335 OpInfo::TensorProperties* b_matrix = matmul_op_info.add_inputs(); in CountBatchMatMulOperations()1356 ops += CountMatMulOperations(matmul_op_info, &matmul_unknown_shapes); in CountBatchMatMulOperations()