Searched refs:matmul_op_info (Results 1 – 1 of 1) sorted by relevance
824 OpInfo matmul_op_info; in CountBatchMatMulOperations() local825 matmul_op_info.set_op("MatMul"); in CountBatchMatMulOperations()832 (*matmul_op_info.mutable_attr())["transpose_a"] = transpose_a; in CountBatchMatMulOperations()839 (*matmul_op_info.mutable_attr())["transpose_b"] = transpose_b; in CountBatchMatMulOperations()841 OpInfo::TensorProperties* a_matrix = matmul_op_info.add_inputs(); in CountBatchMatMulOperations()849 OpInfo::TensorProperties* b_matrix = matmul_op_info.add_inputs(); in CountBatchMatMulOperations()859 ops += CountMatMulOperations(matmul_op_info, &matmul_unknown_shapes); in CountBatchMatMulOperations()