def pattern(self):
return dict(nodes=[
('input', dict()),
('add_const',
dict(op='Const',
value=lambda v: check_value(
v, lambda x: np.allclose(x, 3.0, atol=1e-6)))),
('add', dict(op='Add')),
('relu', dict(op='ReLU')),
('min_const',
dict(op='Const',
value=lambda v: check_value(
v, lambda x: np.allclose(x, 6.0, atol=1e-6)))),
('min', dict(op='Minimum')),
('mul_const',
dict(op='Const',
value=lambda v: check_value(
v, lambda x: np.allclose(x, 1.0 / 6.0, atol=1e-6)))),
('mul', dict(op='Mul')),
],
edges=[
('input', 'add', {
'out': 0
}),
('add_const', 'add', {}),
('add', 'relu', {}),
('relu', 'min', {}),
('min_const', 'min', {}),
('min', 'mul', {}),
('mul_const', 'mul', {}),
])
def pattern(**kwargs):
return dict(nodes=[
('shape', dict(op='ShapeOf')),
('random_uniform', dict(op='RandomUniform')),
('mul', dict(op='Mul')),
('add_const',
dict(op='Const',
value=lambda v: check_value(
v, lambda x: np.allclose(x, 0.0, atol=0)))),
('add', dict(op='Add')),
('add2', dict(op='Add')),
('floor', dict(op='Floor')),
],
edges=[
('shape', 'random_uniform'),
('random_uniform', 'mul'),
('mul', 'add', {
'in': 0
}),
('add_const', 'add', {
'in': 1
}),
('add', 'add2'),
('add2', 'floor'),
])
def pattern(self):
return dict(nodes=[
('input', dict()),
('add', dict(op='Add')),
('const_0',
dict(op='Const',
value=lambda v: check_value(
v, lambda x: np.allclose(x, 0.0, atol=1e-6)))),
('const_3',
dict(op='Const',
value=lambda v: check_value(
v, lambda x: np.allclose(x, 3.0, atol=1e-6)))),
('const_6',
dict(op='Const',
value=lambda v: check_value(
v, lambda x: np.allclose(x, 6.0, atol=1e-6)))),
('const_1_6',
dict(op='Const',
value=lambda v: check_value(
v, lambda x: np.allclose(x, 1.0 / 6.0, atol=1e-6)))),
('clamp', dict(op='Clamp')),
('mul', dict(op='Mul')),
('mul_2', dict(op='Mul')),
],
edges=[
('input', 'add', {}),
('input', 'mul', {}),
('const_3', 'add', {}),
('add', 'clamp', {
'in': 0
}),
('const_0', 'clamp', {
'in': 1
}),
('const_6', 'clamp', {
'in': 2
}),
('clamp', 'mul', {}),
('mul', 'mul_2', {}),
('const_1_6', 'mul_2', {}),
])
def pattern(self):
return dict(
nodes=[
('exp', dict(op='Exp')),
('add', dict(op='Add')),
('const_1', dict(op='Const', value=lambda v: check_value(v, lambda x: np.allclose(x, 1.0, atol=1e-6)))),
('ln', dict(op='Log')),
],
edges=[
('exp', 'add', {}),
('const_1', 'add', {}),
('add', 'ln', {}),
])
