Searched refs:edge_pattern (Results 1 – 3 of 3) sorted by relevance
| /dports/misc/py-tvm/incubator-tvm-0.6.1/src/relay/pass/ |
| H A D | fuse_ops.cc | 257 if (edge_pattern == kBroadcast && in VisitExpr_()
261 edge_pattern = kElemWise; in VisitExpr_()
417 OpPatternKind* edge_pattern) { in LeastCommonAncestor() argument
422 edge_pattern[0] = CombinePattern( in LeastCommonAncestor()
423 edge_pattern[0], rhs->pattern); in LeastCommonAncestor()
426 edge_pattern[0] = CombinePattern( in LeastCommonAncestor()
427 edge_pattern[0], lhs->pattern); in LeastCommonAncestor()
430 edge_pattern[0] = CombinePattern( in LeastCommonAncestor()
432 edge_pattern[0] = CombinePattern( in LeastCommonAncestor()
461 *edge_pattern = CombinePattern(*edge_pattern, link->value.pattern); in LeastCommonAncestor()
[all …]
|
| /dports/misc/tvm/incubator-tvm-0.6.1/src/relay/pass/ |
| H A D | fuse_ops.cc | 257 if (edge_pattern == kBroadcast && in VisitExpr_()
261 edge_pattern = kElemWise; in VisitExpr_()
417 OpPatternKind* edge_pattern) { in LeastCommonAncestor() argument
422 edge_pattern[0] = CombinePattern( in LeastCommonAncestor()
423 edge_pattern[0], rhs->pattern); in LeastCommonAncestor()
426 edge_pattern[0] = CombinePattern( in LeastCommonAncestor()
427 edge_pattern[0], lhs->pattern); in LeastCommonAncestor()
430 edge_pattern[0] = CombinePattern( in LeastCommonAncestor()
432 edge_pattern[0] = CombinePattern( in LeastCommonAncestor()
461 *edge_pattern = CombinePattern(*edge_pattern, link->value.pattern); in LeastCommonAncestor()
[all …]
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/relay/transforms/ |
| H A D | fuse_ops.cc | 261 OpPatternKind edge_pattern = op_pattern; in VisitExpr_() local
264 edge_pattern = kElemWise; in VisitExpr_()
266 this->Update(call->args[i], node, edge_pattern); in VisitExpr_()
417 edge_pattern[0] = CombinePattern(edge_pattern[0], rhs->pattern); in LeastCommonAncestor()
420 edge_pattern[0] = CombinePattern(edge_pattern[0], lhs->pattern); in LeastCommonAncestor()
423 edge_pattern[0] = CombinePattern(edge_pattern[0], lhs->pattern); in LeastCommonAncestor()
424 edge_pattern[0] = CombinePattern(edge_pattern[0], rhs->pattern); in LeastCommonAncestor()
439 OpPatternKind* edge_pattern) { in LeastCommonAncestor() argument
452 *edge_pattern = CombinePattern(*edge_pattern, link->value.pattern); in LeastCommonAncestor()
455 parent = LeastCommonAncestor(parent, get_node(link->value), edge_pattern); in LeastCommonAncestor()
[all …]
|