Searched refs:SECOND_FROM_END (Results 1 – 3 of 3) sorted by relevance
45 for (size_t i = 0; i < SECOND_FROM_END(mat_a_size); ++i) { in SetDevMatrixShape()53 for (size_t i = 0; i < SECOND_FROM_END(mat_b_size); ++i) { in SetDevMatrixShape()59 dev_matrix_shape->push_back(mat_a_strategy.at(SECOND_FROM_END(mat_a_size))); in SetDevMatrixShape()66 dev_matrix_shape->push_back(mat_b_strategy.at(SECOND_FROM_END(mat_b_size))); in SetDevMatrixShape()141 for (size_t j = 0; j < SECOND_FROM_END(short_size); ++j) { in CheckRelevantDimension()172 if (!transpose_b_ && (mat_a_strategy.back() != mat_b_strategy.at(SECOND_FROM_END(mat_b_size)))) { in CheckStrategy()175 … ", but the shard num of second input's row is " << mat_b_strategy.at(SECOND_FROM_END(mat_b_size)); in CheckStrategy()219 size_t relevant_dimension_index = SECOND_FROM_END(dimension); in InferForwardCommunication()263 …t_tensor_map.erase(output_tensor_map.begin() + static_cast<different_type>(SECOND_FROM_END(size))); in InferTensorMap()582 (LongToSize(slice_shape[SECOND_FROM_END(slice_shape.size())]) % align_size != 0)) { in CheckForTensorSliceValid()
486 constexpr size_t SECOND_FROM_END(size_t s) { return s - 2; } in SECOND_FROM_END() function
592 …(void)slice_shape.erase(slice_shape.begin() + static_cast<different_type>(SECOND_FROM_END(shape_si… in HandleAdaFactorOpt()593 …(void)origin_shape.erase(origin_shape.begin() + static_cast<different_type>(SECOND_FROM_END(shape_… in HandleAdaFactorOpt()594 …(void)tensor_map.erase(tensor_map.begin() + static_cast<different_type>(SECOND_FROM_END(shape_size… in HandleAdaFactorOpt()