| /third_party/mindspore/mindspore/ccsrc/cxx_api/ |
| D | cell.cc | 70 GraphCell::GraphCell(const Graph &graph) : graph_(std::make_shared<Graph>(graph)) { MS_EXCEPTION_IF… in GraphCell() function in mindspore::GraphCell
72 GraphCell::GraphCell(const std::shared_ptr<Graph> &graph) : graph_(graph) { MS_EXCEPTION_IF_NULL(gr… in GraphCell() function in mindspore::GraphCell
74 GraphCell::GraphCell(Graph &&graph) : graph_(std::make_shared<Graph>(graph)) { MS_EXCEPTION_IF_NULL… in GraphCell() function in mindspore::GraphCell
76 void GraphCell::SetContext(const std::shared_ptr<Context> &context) { in SetContext()
78 executor_ = Factory<GraphCell::GraphImpl>::Instance().Create(g_device_target); in SetContext()
88 Status GraphCell::Run(const std::vector<MSTensor> &inputs, std::vector<MSTensor> *outputs) { in Run()
90 executor_ = Factory<GraphCell::GraphImpl>::Instance().Create(g_device_target); in Run()
100 Status GraphCell::Load(uint32_t device_id) { in Load()
102 executor_ = Factory<GraphCell::GraphImpl>::Instance().Create(g_device_target); in Load()
112 std::vector<MSTensor> GraphCell::GetInputs() { in GetInputs()
[all …]
|
| /third_party/mindspore/mindspore/lite/src/cxx_api/ |
| D | cell.cc | 56 GraphCell::GraphCell(const Graph &graph) : graph_(std::shared_ptr<Graph>(new (std::nothrow) Graph(g… in GraphCell() function in mindspore::GraphCell
62 GraphCell::GraphCell(const std::shared_ptr<Graph> &graph) : graph_(graph) { in GraphCell() function in mindspore::GraphCell
68 GraphCell::GraphCell(Graph &&graph) : graph_(std::shared_ptr<Graph>(new (std::nothrow) Graph(graph)… in GraphCell() function in mindspore::GraphCell
74 Status GraphCell::Run(const std::vector<MSTensor> &inputs, std::vector<MSTensor> *outputs) { in Run()
79 Status GraphCell::Load(uint32_t device_id) { in Load()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/cxx_api/ |
| D | model_test.cc | 34 auto status = model.Build(GraphCell(graph), nullptr, nullptr); in TEST_F()
41 GraphCell graph_cell; in TEST_F()
57 ASSERT_TRUE(model.Build(GraphCell(graph), context, nullptr) == kSuccess); in TEST_F()
73 ASSERT_TRUE(model.Build(GraphCell(graph), context, nullptr) == kSuccess); in TEST_F()
92 ASSERT_TRUE(model.Build(GraphCell(graph), context, nullptr) == kSuccess); in TEST_F()
113 ASSERT_TRUE(model.Build(GraphCell(graph), context, nullptr) == kSuccess); in TEST_F()
130 ASSERT_TRUE(model.Build(GraphCell(graph), context, train_cfg) == kSuccess); in TEST_F()
166 ASSERT_TRUE(model.Build(GraphCell(graph), context, train_cfg) == kSuccess); in TEST_F()
198 ASSERT_TRUE(model.Build(GraphCell(graph), context, train_cfg) == kSuccess); in TEST_F()
201 ASSERT_TRUE(model.Build(GraphCell(graph), context, train_cfg) == kSuccess); in TEST_F()
[all …]
|
| /third_party/mindspore/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/include/api/ |
| D | cell.h | 89 class MS_API GraphCell final : public Cell<GraphCell> {
93 GraphCell() = default;
94 ~GraphCell() override = default;
96 explicit GraphCell(const Graph &);
97 explicit GraphCell(Graph &&);
98 explicit GraphCell(const std::shared_ptr<Graph> &);
|
| D | graph.h | 41 friend class GraphCell;
|
| D | model.h | 41 Status Build(GraphCell graph, const std::shared_ptr<Context> &model_context = nullptr);
|
| /third_party/mindspore/include/api/ |
| D | cell.h | 90 class MS_API GraphCell final : public Cell<GraphCell> {
94 GraphCell() = default;
95 ~GraphCell() override = default;
97 explicit GraphCell(const Graph &);
98 explicit GraphCell(Graph &&);
99 explicit GraphCell(const std::shared_ptr<Graph> &);
|
| D | graph.h | 41 friend class GraphCell;
|
| /third_party/mindspore/mindspore/ccsrc/cxx_api/model/ms/ |
| D | ms_model.h | 50 std::shared_ptr<GraphCell> GenerateGraphCell(const std::vector<std::vector<int64_t>> &dims);
53 std::map<std::string, std::shared_ptr<GraphCell>> dynamic_size_graph_map_;
|
| D | ms_model.cc | 49 std::shared_ptr<GraphCell> MsModel::GenerateGraphCell(const std::vector<std::vector<int64_t>> &dims… in GenerateGraphCell()
76 auto graph_cell = std::make_shared<GraphCell>(graph); in GenerateGraphCell()
102 auto graph_cell = std::make_shared<GraphCell>(graph); in Build()
|
| /third_party/mindspore/tests/st/control/ |
| D | test_while_mindir.py | 51 loaded_net = nn.GraphCell(graph)
73 loaded_net = nn.GraphCell(graph)
143 loaded_net = nn.GraphCell(graph)
|
| D | test_switch_layer_mindir.py | 58 loaded_net = nn.GraphCell(graph)
98 loaded_net = nn.GraphCell(graph)
|
| D | test_if_mindir.py | 146 loaded_net = nn.GraphCell(graph)
169 loaded_net = nn.GraphCell(graph)
|
| D | test_recrusive_mindir.py | 65 loaded_net = nn.GraphCell(graph)
|
| /third_party/mindspore/tests/st/cpp/model/ |
| D | test_control.cc | 46 ASSERT_TRUE(control_model.Build(GraphCell(graph), context) == kSuccess); in TEST_F()
123 ASSERT_TRUE(control_model.Build(GraphCell(graph), context) == kSuccess); in TEST_F()
188 ASSERT_TRUE(control_model.Build(GraphCell(graph), context) == kSuccess); in TEST_F()
234 ASSERT_TRUE(control_model.Build(GraphCell(graph), context) == kSuccess); in TEST_F()
303 ASSERT_TRUE(control_model.Build(GraphCell(graph), context) == kSuccess); in TEST_F()
360 ASSERT_TRUE(control_model.Build(GraphCell(graph), context) == kSuccess); in TEST_F()
417 ASSERT_TRUE(control_model.Build(GraphCell(graph), context) == kSuccess); in TEST_F()
|
| D | test_tensor_add.cc | 40 ASSERT_TRUE(tensor_add.Build(GraphCell(graph), context) == kSuccess); in TEST_F()
|
| D | test_zero_copy.cc | 70 ASSERT_TRUE(resnet50.Build(GraphCell(graph), context) == kSuccess); in TEST_F()
118 ASSERT_TRUE(resnet50.Build(GraphCell(graph), context) == kSuccess); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/cxx_api/model/acl/ |
| D | acl_model.cc | 38 graph_cell_ = std::make_shared<GraphCell>(graph_); in Build()
79 auto graph_cell = std::make_shared<GraphCell>(graph); in Build()
|
| /third_party/mindspore/mindspore/nn/ |
| D | __init__.py | 23 from .cell import Cell, GraphKernel, GraphCell
|
| /third_party/mindspore/mindspore/ccsrc/cxx_api/graph/acl/ |
| D | acl_graph_impl.h | 31 class AclGraphImpl : public GraphCell::GraphImpl {
|
| /third_party/mindspore/mindspore/ccsrc/cxx_api/graph/ |
| D | graph_impl.h | 30 class GraphCell::GraphImpl {
|
| /third_party/mindspore/mindspore/ccsrc/cxx_api/model/acl/acl_vm/ |
| D | acl_multi_graph_session.h | 43 std::map<GraphId, GraphCell> graphs_ = {};
|
| /third_party/mindspore/mindspore/ccsrc/cxx_api/graph/gpu/ |
| D | gpu_graph_impl.h | 30 class GPUGraphImpl : public GraphCell::GraphImpl {
|
| /third_party/mindspore/mindspore/ccsrc/cxx_api/model/ |
| D | model_impl.h | 65 std::shared_ptr<GraphCell> graph_cell_ = nullptr;
|
| /third_party/mindspore/mindspore/ccsrc/cxx_api/graph/ascend/ |
| D | ascend_graph_impl.h | 33 class AscendGraphImpl : public GraphCell::GraphImpl {
|