Migrate LiteRT ATS unary op graph generation to use litert::tensor API.

PiperOrigin-RevId: 917866082

Author: gcarranza-1

litert/tensor/arithmetic.h

@@ -165,6 +165,30 @@ Tensor<Mixins...> Relu6(Tensor<Mixins...> a,
   return output;
 }
 
+template <class... Mixins>
+Tensor<Mixins...> ReluN1To1(Tensor<Mixins...> a,
+                            source_location loc = source_location::current()) {
+  Tensor<Mixins...> output =
+      ElementwiseOp<graph::ReluN1To1Operation<Mixins...>>(loc, a);
+  return output;
+}
+
+template <class... Mixins>
+Tensor<Mixins...> ZerosLike(Tensor<Mixins...> a,
+                            source_location loc = source_location::current()) {
+  Tensor<Mixins...> output =
+      ElementwiseOp<graph::ZerosLikeOperation<Mixins...>>(loc, a);
+  return output;
+}
+
+template <class... Mixins>
+Tensor<Mixins...> Relu0To1(Tensor<Mixins...> a,
+                           source_location loc = source_location::current()) {
+  Tensor<Mixins...> output =
+      ElementwiseOp<graph::Relu0To1Operation<Mixins...>>(loc, a);
+  return output;
+}
+
 template <class... Mixins>
 Tensor<Mixins...> LeakyRelu(Tensor<Mixins...> a, float alpha = 0.2f,
                             source_location loc = source_location::current()) {

litert/tensor/arithmetic_graph.h

@@ -83,6 +83,27 @@ struct Relu6Operation : virtual Operation,
   absl::string_view GetName() const override { return "Relu6"; }
 };
 
+template <class... Mixins>
+struct ReluN1To1Operation
+    : virtual Operation,
+      virtual OpMixin<struct ReluN1To1OperationTag, Mixins>... {
+  absl::string_view GetName() const override { return "ReluN1To1"; }
+};
+
+template <class... Mixins>
+struct ZerosLikeOperation
+    : virtual Operation,
+      virtual OpMixin<struct ZerosLikeOperationTag, Mixins>... {
+  absl::string_view GetName() const override { return "ZerosLike"; }
+};
+
+template <class... Mixins>
+struct Relu0To1Operation
+    : virtual Operation,
+      virtual OpMixin<struct Relu0To1OperationTag, Mixins>... {
+  absl::string_view GetName() const override { return "Relu0To1"; }
+};
+
 struct LeakyReluOperationData {
   float alpha;
 };

litert/tensor/datatypes.h

@@ -22,6 +22,10 @@ limitations under the License.
 
 #include "litert/tensor/internal/fp16.h"
 
+namespace tflite {
+struct half;
+}
+
 namespace litert::tensor {
 
 enum class Type {
@@ -266,6 +270,9 @@ struct ApiType<bf16_t> : internal::StorageImpl<Type::kBF16, bf16_t> {};
 template <>
 struct ApiType<fp16_t> : internal::StorageImpl<Type::kFP16, fp16_t> {};
 
+template <>
+struct ApiType<tflite::half> : internal::StorageImpl<Type::kFP16, fp16_t> {};
+
 template <>
 struct ApiType<float> : internal::StorageImpl<Type::kFP32, float> {};