Searched refs:mat_a_strategy (Results 1 – 1 of 1) sorted by relevance
35 void SetDevMatrixShape(const Dimensions &mat_a_strategy, const Dimensions &mat_b_strategy, bool tra… in SetDevMatrixShape() argument38 size_t mat_a_size = mat_a_strategy.size(); in SetDevMatrixShape()46 dev_matrix_shape->push_back(mat_a_strategy.at(i)); in SetDevMatrixShape()59 dev_matrix_shape->push_back(mat_a_strategy.at(SECOND_FROM_END(mat_a_size))); in SetDevMatrixShape()60 dev_matrix_shape->push_back(mat_a_strategy.back()); in SetDevMatrixShape()159 Dimensions mat_a_strategy = stra.at(0); in CheckStrategy() local162 size_t mat_a_size = mat_a_strategy.size(); in CheckStrategy()172 if (!transpose_b_ && (mat_a_strategy.back() != mat_b_strategy.at(SECOND_FROM_END(mat_b_size)))) { in CheckStrategy()174 …<< ", the transpose_b is false, the shard num of first input's column is " << mat_a_strategy.back() in CheckStrategy()177 } else if (transpose_b_ && (mat_a_strategy.back() != mat_b_strategy.back())) { in CheckStrategy()[all …]