chore: import upstream snapshot with attribution

python/tvm/relax/op/nn/__init__.py

@@ -0,0 +1,61 @@
+# isort: skip_file
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+"""Neural network related operators."""
+
+from .nn import (
+    adaptive_avg_pool1d,
+    adaptive_avg_pool2d,
+    adaptive_avg_pool3d,
+    attention,
+    attention_bias,
+    attention_var_len,
+    avg_pool1d,
+    avg_pool2d,
+    avg_pool3d,
+    batch_flatten,
+    batch_norm,
+    conv1d,
+    conv1d_transpose,
+    conv2d,
+    conv2d_transpose,
+    conv3d,
+    conv3d_transpose,
+    cross_entropy_with_logits,
+    dropout,
+    gelu,
+    gelu_tanh,
+    group_norm,
+    instance_norm,
+    layer_norm,
+    leakyrelu,
+    log_softmax,
+    max_pool1d,
+    max_pool2d,
+    max_pool3d,
+    nll_loss,
+    pad,
+    pixel_shuffle,
+    prelu,
+    relu,
+    relu6,
+    rms_norm,
+    selu,
+    silu,
+    softmax,
+    softplus,
+)

python/tvm/relax/op/nn/_ffi_api.py

@@ -0,0 +1,21 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+"""Constructor APIs"""
+
+import tvm_ffi
+
+tvm_ffi.init_ffi_api("relax.op.nn", __name__)