MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / IsDynamicStitchOp

Function IsDynamicStitchOp

serving/processor/framework/graph_optimizer.cc:962–966  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

960}
961
962bool IsDynamicStitchOp(Node* n) {
963 return n->op_def().name() == "DynamicStitch" ||
964 n->op_def().name() == "DynamicStitchFast" ||
965 n->op_def().name() == "ParallelDynamicStitch";
966}
967
968Node* FindDynamicStitchNode(Node* gather_node) {
969 // Supported pattern:

Callers 1

FindDynamicStitchNodeFunction · 0.85

Calls 2

nameMethod · 0.65
op_defMethod · 0.45

Tested by

no test coverage detected