Searched refs:i64array (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| H A D | legalize_tf.cc | 545 SmallVector<int64_t, 4> i64array; in matchAndRewrite() local
549 i64array.emplace_back(value); in matchAndRewrite()
553 llvm::makeArrayRef(i64array), rewriter.getIntegerType(64)); in matchAndRewrite()
615 SmallVector<int64_t, 4> i64array; in matchAndRewrite() local
619 i64array.emplace_back(value); in matchAndRewrite()
623 llvm::makeArrayRef(i64array), rewriter.getIntegerType(64)); in matchAndRewrite()
|
| H A D | legalize_tfl.cc | 879 SmallVector<int64_t, 4> i64array({1, 1, 1, 1}); in matchAndRewrite() local
888 i64array[1] = kernel_h; in matchAndRewrite()
889 i64array[2] = kernel_w; in matchAndRewrite()
905 llvm::makeArrayRef(i64array), rewriter.getIntegerType(64)); in matchAndRewrite()
956 SmallVector<int64_t, 4> i64array({1, 1, 1, 1}); in matchAndRewrite() local
965 i64array[1] = kernel_h; in matchAndRewrite()
966 i64array[2] = kernel_w; in matchAndRewrite()
982 RankedTensorType::get(i64array, rewriter.getIntegerType(64)); in matchAndRewrite()
|