Home
last modified time | relevance | path

Searched defs:compute_nodes (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dkernel_fusion.h40 std::vector<AnfNodePtr> compute_nodes; member
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/
Dfusion_tbe_json_creator.cc70 auto compute_nodes = fusion_scope_info.compute_nodes; in GenOpListJson() local
97 auto compute_nodes = fusion_scope_info.compute_nodes; in CheckInput() local
123 void FusionBuildTbeJsonCreator::GenDataJson(const std::vector<AnfNodePtr> &compute_nodes, in GenDataJson()
/third_party/mindspore/tests/ut/cpp/tbe/
Dtbe_json_creator_test.cc290 std::vector<AnfNodePtr> compute_nodes = {relu1, relu2}; in TEST_F() local
349 std::vector<AnfNodePtr> compute_nodes = {add, conv2d}; in TEST_F() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_adapter.cc285 bool TbeAdapter::IsSpecialFusionComputeNode(const std::vector<mindspore::AnfNodePtr> &compute_nodes… in IsSpecialFusionComputeNode()
389 auto compute_nodes = fusion_scope_info.compute_nodes; in GetSpecDataInput() local
Dtbe_kernel_build.cc994 const std::vector<mindspore::AnfNodePtr> &compute_nodes, in GenFusionScopeJson()
1214 const std::vector<mindspore::AnfNodePtr> &compute_nodes, in GetInputLayers()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.cc1085 std::vector<CNodePtr> compute_nodes; in AssignDynamicMemory() local