1 /** 2 * Copyright 2020 Huawei Technologies Co., Ltd 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 #ifndef MINDSPORE_CCSRC_CXX_API_GRAPH_GRAPH_IMPL_H 17 #define MINDSPORE_CCSRC_CXX_API_GRAPH_GRAPH_IMPL_H 18 #include <functional> 19 #include <map> 20 #include <string> 21 #include <vector> 22 #include <memory> 23 #include <utility> 24 #include "include/api/cell.h" 25 #include "include/api/graph.h" 26 #include "cxx_api/graph/graph_data.h" 27 #include "utils/utils.h" 28 29 namespace mindspore { 30 class GraphCell::GraphImpl { 31 public: GraphImpl()32 GraphImpl() : graph_(nullptr), graph_context_(nullptr) {} 33 virtual ~GraphImpl() = default; 34 MutableGraphData()35 std::shared_ptr<Graph::GraphData> &MutableGraphData() const { return graph_->graph_data_; } SetGraph(const std::shared_ptr<Graph> & graph)36 void SetGraph(const std::shared_ptr<Graph> &graph) { graph_ = graph; } SetContext(const std::shared_ptr<Context> & context)37 void SetContext(const std::shared_ptr<Context> &context) { graph_context_ = context; } 38 39 virtual Status Run(const std::vector<MSTensor> &inputs, std::vector<MSTensor> *outputs) = 0; 40 virtual Status Load(uint32_t device_id) = 0; 41 42 virtual std::vector<MSTensor> GetInputs() = 0; 43 virtual std::vector<MSTensor> GetOutputs() = 0; 44 45 protected: 46 std::shared_ptr<Graph> graph_; 47 std::shared_ptr<Context> graph_context_; 48 }; 49 } // namespace mindspore 50 #endif // MINDSPORE_CCSRC_CXX_API_GRAPH_GRAPH_IMPL_H 51