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