[mlir][tosa] Relax tosa.apply_scale operations

[rsuderman]

Summary

Apply scale may operate on vectors, scalars, or tensors during
tiling. Relax the requirements to avoid failures.

mlir/include/mlir/Dialect/Tosa/IR/TosaTypesBase.td

 //===----------------------------------------------------------------------===//
 
 class Tosa_TypeLike<list<Type> types, string description = ""> : TypeConstraint<Or<[
      AnyTypeOf<types>.predicate,
      VectorOf<types>.predicate,
      TensorOf<types>.predicate]>,
      description>;
 
+def Tosa_IntLike : Tosa_TypeLike<[Tosa_Int], "signless-integer-like">;
 def Tosa_Int8Like : Tosa_TypeLike<[Tosa_Int8], "signless-integer-8-bit-like">;
 def Tosa_Int16Like : Tosa_TypeLike<[Tosa_Int16], "signless-integer-16-bit-like">;
 def Tosa_Int32Like : Tosa_TypeLike<[Tosa_Int32], "signless-integer-32-bit-like">;
 def Tosa_Int64Like : Tosa_TypeLike<[Tosa_Int64], "signless-integer-64-bit-like">;
 
 //===----------------------------------------------------------------------===//
 // Attribute predicates and classes.
 //===----------------------------------------------------------------------===//

mlir/include/mlir/Dialect/Tosa/IR/TosaUtilOps.td

let description = [{
     multiple quantized operations (mul, convolution, rescale, matmul, pooling).
 
     The commonplace implementation is to use i64 operations to avoid integer
     overflow with target specific implementations can use native operations to
     avoid wider than necessary types.
   }];
 
   let arguments = (ins
-    Tosa_Int:$value,
-    Tosa_Int:$multiplier,
+    Tosa_IntLike:$value,
+    Tosa_IntLike:$multiplier,
     Tosa_Int8Like:$shift,
     BoolAttr:$double_round
   );
 
   let results = (outs
-    Tosa_Int:$output
+    Tosa_IntLike:$output
   );
 }
 
 //===----------------------------------------------------------------------===//
 // Operator: yield
 //===----------------------------------------------------------------------===//
 def Tosa_YieldOp : Tosa_Op<"yield", [
        Terminator,