Searched refs:different_type (Results 1 – 9 of 9) sorted by relevance
263 …(void)output_tensor_map.erase(output_tensor_map.begin() + static_cast<different_type>(SECOND_FROM_… in InferTensorMap()273 mat_a_tensor_map.begin() + static_cast<different_type>(LAST_INDEX(size) - mat_a_dimension_)); in InferTensorMap()278 …(void)mat_b_tensor_map.erase(mat_b_tensor_map.begin() + static_cast<different_type>(THIRD_FROM_END… in InferTensorMap()282 mat_b_tensor_map.begin() + static_cast<different_type>(LAST_INDEX(size) - mat_b_dimension_)); in InferTensorMap()288 …mat_b_tensor_map.begin() + static_cast<different_type>(LAST_INDEX(mat_b_tensor_map.size())), last_… in InferTensorMap()
189 gamma_strategy.begin() + static_cast<different_type>(gamma_diff)); in GenerateGammaAndBetaStrategies()191 …ategy.erase(beta_strategy.begin(), beta_strategy.begin() + static_cast<different_type>(beta_diff)); in GenerateGammaAndBetaStrategies()
225 …output_tensor_map.begin() + static_cast<different_type>(inputs_shape_[0].size() - IntToSize(axes_i… in InferTensorMapAxesInt()226 output_tensor_map.begin() + static_cast<different_type>(inputs_shape_[0].size())); in InferTensorMapAxesInt()
147 … tensor_map_out.begin() + static_cast<different_type>(inputs_shape_.at(1).size() - 1)); in InferTensorMap()152 …(void)tensor_map_in_index.erase(tensor_map_in_index.begin() + static_cast<different_type>(inputs_s… in InferTensorMap()
73 …if (std::accumulate(stra[2].begin(), stra[2].begin() + static_cast<different_type>(stra[1].size())… in CheckStrategy()
131 … expend_map.begin() + static_cast<different_type>(dev_matrix_size - strategy_size)); in SetTensorMap()
49 using different_type = std::vector<int64_t>::difference_type; variable
1122 input0_strategy.begin() + static_cast<different_type>(size_diff)); in GenerateStrategiesForBroadcastLeft()1173 input1_strategy.begin() + static_cast<different_type>(size_diff)); in GenerateStrategiesForBroadcastRight()
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()