feat(imperative/opr-mm): add broadcast

GitOrigin-RevId: 83640255c758b85a4cbb7d40fb20b822d4a8b30f

src/opr-mm/impl/collective_comm.cpp

@@ -810,7 +810,7 @@ void CollectiveComm::init_output_static_infer_desc() {
         };
 
         auto get_shape_from_server = [this](TensorShape& dest, const InpVal&) {
-            if (!m_enable_shape_infer) {
+            if (!m_enable_shape_infer && !owner_graph()->options().imperative_proxy_graph) {
                 return false;
             }