Home
last modified time | relevance | path

Searched refs:subgraph_id (Results 1 – 25 of 39) sorted by relevance

12

/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/executor/
H A Dpointwise_fusion_pass.cc200 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 Dpointwise_fusion_pass.cc200 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 Ddefault_subgraph_property.cc59 const int subgraph_id = 0) const { in CreateSubgraphNode() argument
62 n->attrs.name = "_CachedOp" + std::to_string(subgraph_id); in CreateSubgraphNode()
H A Ddefault_subgraph_property_v2.cc64 const int subgraph_id = 0) const override { in CreateSubgraphNode() argument
67 n->attrs.name = "_CachedOp" + std::to_string(subgraph_id); in CreateSubgraphNode()
H A Dsubgraph_property.h294 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 Dbuild_subgraph.cc348 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 Ddefault_subgraph_property.cc59 const int subgraph_id = 0) const { in CreateSubgraphNode() argument
62 n->attrs.name = "_CachedOp" + std::to_string(subgraph_id); in CreateSubgraphNode()
H A Ddefault_subgraph_property_v2.cc64 const int subgraph_id = 0) const override { in CreateSubgraphNode() argument
67 n->attrs.name = "_CachedOp" + std::to_string(subgraph_id); in CreateSubgraphNode()
H A Dsubgraph_property.h294 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 Dbuild_subgraph.cc348 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 Dmkldnn_transformer_property.h85 const int subgraph_id = 0) const override {
105 node_name << NameMapping.at(op_name) << "_" << std::to_string(subgraph_id);
H A Dmkldnn_fc_property.h185 const int subgraph_id = 0) const override {
203 node_name << std::to_string(subgraph_id);
H A Dmkldnn_conv_property.h201 const int subgraph_id = 0) const override {
232 node_name << std::to_string(subgraph_id);
H A Dmkldnn_post_quantize_align_scale_property.h146 const int subgraph_id = 0) const override {
H A Dmkldnn_post_quantize_property.h133 const int subgraph_id = 0) const override {
H A Dmkldnn_transformer_post_quantize_property.h152 const int subgraph_id = 0) const override {
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/subgraph/mkldnn/
H A Dmkldnn_transformer_property.h85 const int subgraph_id = 0) const override {
105 node_name << NameMapping.at(op_name) << "_" << std::to_string(subgraph_id);
H A Dmkldnn_fc_property.h185 const int subgraph_id = 0) const override {
203 node_name << std::to_string(subgraph_id);
H A Dmkldnn_conv_property.h201 const int subgraph_id = 0) const override {
232 node_name << std::to_string(subgraph_id);
H A Dmkldnn_post_quantize_align_scale_property.h146 const int subgraph_id = 0) const override {
H A Dmkldnn_post_quantize_property.h133 const int subgraph_id = 0) const override {
/dports/misc/mxnet/incubator-mxnet-1.9.0/example/extensions/lib_subgraph/
H A Dsubgraph_lib.cc210 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 Dsubgraph_lib.cc210 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 Dcustom_subgraph_property.h371 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 Dcustom_subgraph_property.h371 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);

12