Searched refs:matmul_dims (Results 1 – 3 of 3) sorted by relevance
212 MatMulDimensions matmul_dims; member
613 batch_mat_mul.matmul_dims.n = 0; in CountBatchMatMulDimProduct()614 batch_mat_mul.matmul_dims.m = 0; in CountBatchMatMulDimProduct()615 batch_mat_mul.matmul_dims.k = 0; in CountBatchMatMulDimProduct()622 dimension_product *= batch_mat_mul.matmul_dims.n; in CountBatchMatMulDimProduct()623 dimension_product *= batch_mat_mul.matmul_dims.m; in CountBatchMatMulDimProduct()624 dimension_product *= batch_mat_mul.matmul_dims.k; in CountBatchMatMulDimProduct()
1343 batch_mat_mul->matmul_dims.m = (transpose_a.b()) in CountBatchMatMulOperations()1346 batch_mat_mul->matmul_dims.k = (transpose_a.b()) in CountBatchMatMulOperations()1349 batch_mat_mul->matmul_dims.n = (transpose_b.b()) in CountBatchMatMulOperations()