def extract(node):
Eltwise.update_node_stat(node, {'operation': 'max'})
return __class__.enabled
def replace_pattern(graph: Graph, match: [str, Node]):
Eltwise.update_node_stat(
match['op'], {'operation': op_to_operation_map[match['op'].type]})
def extract(node):
Eltwise.update_node_stat(node, {
'operation': 'floor_mod',
'force_precision': 'I32',
})
return __class__.enabled