ind += (current_index[i] % shape[i]) * mul;
    mul *= shape[i];
  }
  return ind;
}

// Helper method that increment index represented in 'current_index_ptr'
// in the shape of 'result_shape'.
void IncrementIndex(ArrayRef<int64_t> result_shape,
                    llvm::SmallVectorImpl<int64_t> &current_index) {
  for (int i = result_shape.size() - 1; i >= 0; --i) {
    current_index[i]++;

    ind += (current_index[i] % shape[i]) * mul;
    mul *= shape[i];
  }
  return ind;
}

// Helper method that increment index represented in 'current_index_ptr'
// in the shape of 'result_shape'.
void IncrementIndex(ArrayRef<int64_t> result_shape,
                    std::vector<int64_t>* current_index_ptr) {
  std::vector<int64_t>& current_index = *current_index_ptr;
  for (int i = result_shape.size() - 1; i >= 0; --i) {