/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/executor/ |
H A D | pointwise_fusion_pass.cc | 200 const int subgraph_id = subgraph_assignment[i]; in CopyAndReplaceSubgraphs() local 201 if (subgraph_id != -1) { in CopyAndReplaceSubgraphs() 202 auto& info = subgraphs[subgraph_id]; in CopyAndReplaceSubgraphs() 205 if (their_subgraph == subgraph_id) { in CopyAndReplaceSubgraphs() 243 if (subgraph_id == -1) { in CopyAndReplaceSubgraphs() 248 auto& info = subgraphs[subgraph_id]; in CopyAndReplaceSubgraphs() 261 if (subgraph_id == subgraph_assignment[dep]) { in CopyAndReplaceSubgraphs() 270 if (subgraph_id == -1) { in CopyAndReplaceSubgraphs() 296 const int subgraph_id = subgraph_assignment[i]; in CopyAndReplaceSubgraphs() local 302 auto& info = subgraphs[subgraph_id]; in CopyAndReplaceSubgraphs() [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/executor/ |
H A D | pointwise_fusion_pass.cc | 200 const int subgraph_id = subgraph_assignment[i]; in CopyAndReplaceSubgraphs() local 201 if (subgraph_id != -1) { in CopyAndReplaceSubgraphs() 202 auto& info = subgraphs[subgraph_id]; in CopyAndReplaceSubgraphs() 205 if (their_subgraph == subgraph_id) { in CopyAndReplaceSubgraphs() 243 if (subgraph_id == -1) { in CopyAndReplaceSubgraphs() 248 auto& info = subgraphs[subgraph_id]; in CopyAndReplaceSubgraphs() 261 if (subgraph_id == subgraph_assignment[dep]) { in CopyAndReplaceSubgraphs() 270 if (subgraph_id == -1) { in CopyAndReplaceSubgraphs() 296 const int subgraph_id = subgraph_assignment[i]; in CopyAndReplaceSubgraphs() local 302 auto& info = subgraphs[subgraph_id]; in CopyAndReplaceSubgraphs() [all …]
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/subgraph/ |
H A D | default_subgraph_property.cc | 59 const int subgraph_id = 0) const { in CreateSubgraphNode() argument 62 n->attrs.name = "_CachedOp" + std::to_string(subgraph_id); in CreateSubgraphNode()
|
H A D | default_subgraph_property_v2.cc | 64 const int subgraph_id = 0) const override { in CreateSubgraphNode() argument 67 n->attrs.name = "_CachedOp" + std::to_string(subgraph_id); in CreateSubgraphNode()
|
H A D | subgraph_property.h | 294 const int subgraph_id = 0) const { 309 const int subgraph_id = 0) const { 310 return CreateSubgraphNode(sym, subgraph_id); 322 const int subgraph_id = 0) const { 325 return CreateSubgraphNode(sym, bridge->GetV1ptr(), subgraph_id); 338 const int subgraph_id = 0) const {
|
H A D | build_subgraph.cc | 348 const BiDirectedNode* node, const size_t snid, size_t* subgraph_id) { in SelectSubgraphNodes() argument 358 PreSelectSubgraphNodes(*g, subgraph_selector, *subgraph_id, snid, simple_nodes, in SelectSubgraphNodes() 387 (*subgraph_id)++; in SelectSubgraphNodes() 404 size_t subgraph_id = 0; in FindSubgraphs() local 409 snode.get(), i, &subgraph_id); in FindSubgraphs() 606 const size_t subgraph_id, in CreateSubgraphNode() argument 653 nnvm::ObjectPtr n = subg_prop->CreateSubgraphNode(sym, subgraph_selector, subgraph_id); in CreateSubgraphNode() 703 const size_t subgraph_id) { in AdjustSubgraphNode() argument 710 subg_prop->AdjustSubgraphNode(node_list, subgraph_selector, subgraph_id); in AdjustSubgraphNode()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/subgraph/ |
H A D | default_subgraph_property.cc | 59 const int subgraph_id = 0) const { in CreateSubgraphNode() argument 62 n->attrs.name = "_CachedOp" + std::to_string(subgraph_id); in CreateSubgraphNode()
|
H A D | default_subgraph_property_v2.cc | 64 const int subgraph_id = 0) const override { in CreateSubgraphNode() argument 67 n->attrs.name = "_CachedOp" + std::to_string(subgraph_id); in CreateSubgraphNode()
|
H A D | subgraph_property.h | 294 const int subgraph_id = 0) const { 309 const int subgraph_id = 0) const { 310 return CreateSubgraphNode(sym, subgraph_id); 322 const int subgraph_id = 0) const { 325 return CreateSubgraphNode(sym, bridge->GetV1ptr(), subgraph_id); 338 const int subgraph_id = 0) const {
|
H A D | build_subgraph.cc | 348 const BiDirectedNode* node, const size_t snid, size_t* subgraph_id) { in SelectSubgraphNodes() argument 358 PreSelectSubgraphNodes(*g, subgraph_selector, *subgraph_id, snid, simple_nodes, in SelectSubgraphNodes() 387 (*subgraph_id)++; in SelectSubgraphNodes() 404 size_t subgraph_id = 0; in FindSubgraphs() local 409 snode.get(), i, &subgraph_id); in FindSubgraphs() 606 const size_t subgraph_id, in CreateSubgraphNode() argument 653 nnvm::ObjectPtr n = subg_prop->CreateSubgraphNode(sym, subgraph_selector, subgraph_id); in CreateSubgraphNode() 703 const size_t subgraph_id) { in AdjustSubgraphNode() argument 710 subg_prop->AdjustSubgraphNode(node_list, subgraph_selector, subgraph_id); in AdjustSubgraphNode()
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/subgraph/mkldnn/ |
H A D | mkldnn_transformer_property.h | 85 const int subgraph_id = 0) const override { 105 node_name << NameMapping.at(op_name) << "_" << std::to_string(subgraph_id);
|
H A D | mkldnn_fc_property.h | 185 const int subgraph_id = 0) const override { 203 node_name << std::to_string(subgraph_id);
|
H A D | mkldnn_conv_property.h | 201 const int subgraph_id = 0) const override { 232 node_name << std::to_string(subgraph_id);
|
H A D | mkldnn_post_quantize_align_scale_property.h | 146 const int subgraph_id = 0) const override {
|
H A D | mkldnn_post_quantize_property.h | 133 const int subgraph_id = 0) const override {
|
H A D | mkldnn_transformer_post_quantize_property.h | 152 const int subgraph_id = 0) const override {
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/subgraph/mkldnn/ |
H A D | mkldnn_transformer_property.h | 85 const int subgraph_id = 0) const override { 105 node_name << NameMapping.at(op_name) << "_" << std::to_string(subgraph_id);
|
H A D | mkldnn_fc_property.h | 185 const int subgraph_id = 0) const override { 203 node_name << std::to_string(subgraph_id);
|
H A D | mkldnn_conv_property.h | 201 const int subgraph_id = 0) const override { 232 node_name << std::to_string(subgraph_id);
|
H A D | mkldnn_post_quantize_align_scale_property.h | 146 const int subgraph_id = 0) const override {
|
H A D | mkldnn_post_quantize_property.h | 133 const int subgraph_id = 0) const override {
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/example/extensions/lib_subgraph/ |
H A D | subgraph_lib.cc | 210 MXReturnValue myReviewSubgraph(const mxnet::ext::Graph *subgraph, int subgraph_id, bool* accept, in myReviewSubgraph() argument 218 std::cout << "subgraph " << subgraph_id << ": " << std::endl; in myReviewSubgraph()
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/example/extensions/lib_subgraph/ |
H A D | subgraph_lib.cc | 210 MXReturnValue myReviewSubgraph(const mxnet::ext::Graph *subgraph, int subgraph_id, bool* accept, in myReviewSubgraph() argument 218 std::cout << "subgraph " << subgraph_id << ": " << std::endl; in myReviewSubgraph()
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/subgraph/partitioner/ |
H A D | custom_subgraph_property.h | 371 const int subgraph_id = 0) const { 399 CHECK(call_review_subgraph_(review_subgraph_, subgraph_json.c_str(), subgraph_id, 427 n->attrs.name = "_op" + std::to_string(subgraph_id);
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/subgraph/partitioner/ |
H A D | custom_subgraph_property.h | 371 const int subgraph_id = 0) const { 399 CHECK(call_review_subgraph_(review_subgraph_, subgraph_json.c_str(), subgraph_id, 427 n->attrs.name = "_op" + std::to_string(subgraph_id);
|