[all-commits] [llvm/llvm-project] ec83c7: [MLGO] Make TFLiteUtils throw an error if some fea...

Aiden Grossman via All-commits all-commits at lists.llvm.org
Sat Sep 10 15:59:30 PDT 2022


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: ec83c7e358ecd7db9af2d980b6d528f5ea6865a4
      https://github.com/llvm/llvm-project/commit/ec83c7e358ecd7db9af2d980b6d528f5ea6865a4
  Author: Aiden Grossman <agrossman154 at yahoo.com>
  Date:   2022-09-10 (Sat, 10 Sep 2022)

  Changed paths:
    M llvm/lib/Analysis/TFLiteUtils.cpp
    M llvm/unittests/Analysis/TFUtilsTest.cpp

  Log Message:
  -----------
  [MLGO] Make TFLiteUtils throw an error if some features haven't been passed to the model

In the Tensorflow C lib utilities, an error gets thrown if some features
haven't gotten passed into the model (due to differences in ordering
which now don't exist with the transition to TFLite). However, this is
not currently the case when using TFLiteUtils. This patch makes some
minor changes to throw an error when not all inputs of the model have
been passed, which when not handled will result in a seg fault within
TFLite.

Reviewed By: mtrofin

Differential Revision: https://reviews.llvm.org/D133451




More information about the All-commits mailing list