Searched refs:y_row (Results 1 – 3 of 3) sorted by relevance
45 int64_t y_row = y_last[static_cast<size_t>(transpose_b)]; in MatMulInferShape() local48 if (x_col != y_row) { in MatMulInferShape()50 … << "got " << x_col << " and " << y_row << " , with x1 shape " << x_shp in MatMulInferShape()
58 int64_t y_row = y_last[static_cast<size_t>(transpose_b)]; in BatchMatmulInferShape() local61 if (x_col != y_row) { in BatchMatmulInferShape()63 … << "got " << x_col << " and " << y_row << " , with x1 shape " << x_shp in BatchMatmulInferShape()
299 auto y_row = y_shp[(transpose_b ? 1 : 0)]; in InferImplMatMul() local300 if (x_col != y_row) { in InferImplMatMul()301 MS_LOG(EXCEPTION) << "MatMul shape error, got x_col: " << x_col << ", y_row: " << y_row in InferImplMatMul()367 auto y_row = y_shp[offset + (transpose_b ? 1 : 0)]; in InferImplBatchMatMul() local368 if (x_col != y_row) { in InferImplBatchMatMul()369 MS_LOG(EXCEPTION) << "BatchMatMul shape error, got x_col: " << x_col << ", y_row: " << y_row in InferImplBatchMatMul()