[PATCH] D85250: [llvm] Expose type and element count-related APIs on TensorSpec

[Mircea Trofin via Phabricator via llvm-commits]

mtrofin added a comment.

In D85250#2194715 <https://reviews.llvm.org/D85250#2194715>, @davidxl wrote:

> What is the use case of the new APIs?

They are a small but necessary part of a change that allows us to import a specification of additional outputs that the model can produce and then need to be captured in the training log. This enables "final reward" training algorithms. Since they are relatively "general goodness", I didn't add that context here.


Repository:
  rG LLVM Github Monorepo

CHANGES SINCE LAST ACTION
  https://reviews.llvm.org/D85250/new/

https://reviews.llvm.org/D85250