Add shape inference for AtenAsStrided (#4076)

Added shape inference for AtenAsStridedOp

Author: keshavj-cerebras

projects/ltc/csrc/base_lazy_backend/shape_inference.cpp

@@ -513,6 +513,13 @@ std::vector<torch::lazy::Shape> compute_shape_scalar_tensor(
   return {Shape(dtype.value_or(s.type()), c10::ArrayRef<int64_t>{})};
 }
 
+std::vector<torch::lazy::Shape>
+compute_shape_as_strided(const at::Tensor &self, at::IntArrayRef size,
+                         at::IntArrayRef stride,
+                         c10::optional<int64_t> storage_offset) {
+  return {Shape(self.scalar_type(), size.vec())};
+}
+
 std::vector<torch::lazy::Shape> compute_shape_roll(const at::Tensor &self,
                                                    at::IntArrayRef shifts,
                                                    at::IntArrayRef dims) {