Why Gemfury? Push, build, and install  RubyGems npm packages Python packages Maven artifacts PHP packages Go Modules Debian packages RPM packages NuGet packages

Repository URL to install this package:

Details    
mmcv / ops / csrc / pytorch / npu / common_util.h
Size: Mime:
#ifndef MMCV_OPS_CSRC_COMMON__UTIL_HPP_
#define MMCV_OPS_CSRC_COMMON__UTIL_HPP_
const int SIZE = 8;

c10::SmallVector<int64_t, SIZE> array_to_vector(c10::IntArrayRef shape) {
  c10::SmallVector<int64_t, SIZE> shape_small_vec;
  for (uint64_t i = 0; i < shape.size(); i++) {
    shape_small_vec.emplace_back(shape[i]);
  }
}

#endif  // MMCV_OPS_CSRC_COMMON__UTIL_HPP_