(self, api_item_yaml)
| 105 | } |
| 106 | |
| 107 | def get_api_name(self, api_item_yaml): |
| 108 | if 'op' in api_item_yaml: |
| 109 | return api_item_yaml['op'] |
| 110 | elif 'backward_op' in api_item_yaml: |
| 111 | return api_item_yaml['backward_op'] |
| 112 | else: |
| 113 | raise ValueError("op or backward_op are not in api_yaml.") |
| 114 | |
| 115 | def get_api_func_name(self): |
| 116 | return self.api |