[all-commits] [llvm/llvm-project] 3edd89: fix mlgo regalloc test model generation for tflite

Yundi Qian via All-commits all-commits at lists.llvm.org
Tue Aug 9 12:37:08 PDT 2022


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 3edd8978c3129d15e364abb3632a0db478891415
      https://github.com/llvm/llvm-project/commit/3edd8978c3129d15e364abb3632a0db478891415
  Author: yundiqian <yundi at google.com>
  Date:   2022-08-09 (Tue, 09 Aug 2022)

  Changed paths:
    M llvm/lib/Analysis/models/gen-regalloc-eviction-test-model.py

  Log Message:
  -----------
  fix mlgo regalloc test model generation for tflite

To move from TF C API to TFLite, we found that the argmax op in TFLite does not work for int64 inputs, so cast the int64 inputs to int32 inputs to make TFLite argmax op work

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




More information about the All-commits mailing list