A pattern that matches another pattern with certain data type Parameters ---------- pattern: tvm.relax.dpl.DFPattern The input pattern that needs type annotation. dtype: str The dtype to match.
| 590 | |
| 591 | @register_df_node |
| 592 | class DataTypePattern(DFPattern): |
| 593 | """A pattern that matches another pattern with certain data type |
| 594 | |
| 595 | Parameters |
| 596 | ---------- |
| 597 | pattern: tvm.relax.dpl.DFPattern |
| 598 | The input pattern that needs type annotation. |
| 599 | |
| 600 | dtype: str |
| 601 | The dtype to match. |
| 602 | """ |
| 603 | |
| 604 | def __init__(self, pattern: "DFPattern", dtype: str): |
| 605 | self.__init_handle_by_constructor__(ffi.DataTypePattern, pattern, dtype) # type: ignore |
| 606 | |
| 607 | |
| 608 | @register_df_node |
no outgoing calls
no test coverage detected
searching dependent graphs…