add support for torchvision.ops.stochastic_depth

Author: mshr-h

python/tvm/relax/frontend/torch/fx_translator.py

@@ -1509,6 +1509,7 @@ def create_convert_map(self):
             "softmax": self._softmax,
             "log_softmax": self._log_softmax,
             "dropout": lambda node: self.env[node.args[0]],
+            "stochastic_depth": lambda node: self.env[node.args[0]],
             "clamp": self._clamp,
             "relu": lambda node: self.block_builder.emit(relax.op.nn.relu(self.env[node.args[0]])),
             "leaky_relu": self._leakyrelu,