Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/common/
Dmeta_graph_utils.cc88 std::vector<size_t> GetInputNodeIdx(const schema::MetaGraphT &graphT, const size_t &nodeIdx, const … in GetInputNodeIdx()
111 std::vector<size_t> GetOutputNodeIdx(const schema::MetaGraphT &graphT, const size_t &nodeIdx, in GetOutputNodeIdx()
218 size_t nodeIdx = 0; in IsolateNode() local
263 STATUS IsolateOneWayNode(schema::MetaGraphT *graphT, size_t nodeIdx, bool removeTensor) { in IsolateOneWayNode()
320 STATUS IsolateOneWayNode(schema::MetaGraphT *graph, size_t subGraphIdx, size_t nodeIdx, bool remove… in IsolateOneWayNode()
329 size_t nodeIdx = 0; in IsolateOneWayNode() local
Dgraph_util.cc287 STATUS AddTensor2Node(schema::MetaGraphT *graphT, uint32_t nodeIdx, std::unique_ptr<TensorT> tensor, in AddTensor2Node()
306 STATUS ReplaceTensorOfNode(schema::MetaGraphT *graphT, uint32_t nodeIdx, uint32_t inTensorIdx, in ReplaceTensorOfNode()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/legacy_optimizer/fusion/
Dfusion_pass.cc136 auto nodeIdx = nodeQueue.front(); in MatchOnePattern() local
160 for (auto nodeIdx : entries) { in MatchOnePattern() local
228 bool FusionPass::MatchTree(const schema::MetaGraphT &graph, size_t nodeIdx, const std::shared_ptr<P… in MatchTree()
287 bool FusionPass::CheckMatchParams(const schema::MetaGraphT &graph, size_t nodeIdx, in CheckMatchParams()
Dfusion_pattern.h35 int32_t nodeIdx = -1; member
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/legacy_optimizer/graph/
Ddropout_node_remove_pass.cc28 STATUS IsolateDropoutNode(schema::MetaGraphT *graphT, size_t nodeIdx) { in IsolateDropoutNode()
/third_party/mindspore/mindspore-src/source/mindspore/core/mindrt/src/thread/
Dhqueue.h85 int32_t nodeIdx = free_index; in Enqueue() local