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