Public
  1. Public

onnxruntime

Public

Commits

Dmitri SmirnovGitHub
a20f8037f6d
Dmitri Smirnov authored and GitHub committed e6f85f338e7
Refactor TensorAt, prepare for release (#5180)

* Refactor TensorAt
  locations* must be const and int64_t since our dims are int64_t
  Remove unnecessary copy of locations.
  Remove unnecesary casting and C-casting. Simplify implementation.
  Add a check for string type.
  Make CXX api return T& to fully expose C API in C++, const std::vector& by value as it
  covers more ground and eliminate redundant copy.
  Eliminate inner loop, compute strides first.