tensorflow/compiler/tf2xla/rearrange_function_argument.cc
// Given original input types and argument index mapping, return the new input
// types.
Given original input types and argument index mapping, return the new input
types.
Given original input types and argument index mapping return the new input types .
std::vector<DataType> ShuffleInputDataTypeAttribute(
    const std::vector<DataType>& in_types,
    const std::vector<int>& index_mapping) {
  std::vector<DataType> result(index_mapping.size());
  for (int i = 0, end = in_types.size(); i < end; i++) {
    result[index_mapping.at(i)] = in_types[i];
  }
  return result;
}
std :: vector < DataType > ShuffleInputDataTypeAttribute ( const std :: vector < DataType > & in_types , const std :: vector < int > & index_mapping ) { std :: vector < DataType > result ( index_mapping . size ( ) ) ; for ( int i = 0 , end = in_types . size ( ) ; i < end ; i ++ ) { result [ index_mapping . at ( i ) ] = in_types [ i ] ; } return result ; }
Given original input types and argument index mapping, return the new input
types.
Given original input types and argument index mapping return the new input types .