コード例 #1
0
def _test_loop():
    x = ops.Const(5)
    y = ops.Const(0)
    with ops.loop():
        ops.stop_if(ops.EQ([x, ops.Const(0)]))
        ops.Add([x, ops.Const(-1)], [x])
        ops.Add([y, ops.Const(1)], [y])
    return y
コード例 #2
0
ファイル: net_builder_test.py プロジェクト: Ralfhund/caffe2
def _test_loop():
    x = ops.Const(5)
    y = ops.Const(0)
    with ops.loop():
        ops.stop_if(ops.EQ([x, ops.Const(0)]))
        ops.Add([x, ops.Const(-1)], [x])
        ops.Add([y, ops.Const(1)], [y])
    return y
コード例 #3
0
def _test_inner_stop(x):
    ops.stop_if(ops.LT([x, ops.Const(5)]))
コード例 #4
0
ファイル: net_builder_test.py プロジェクト: Ralfhund/caffe2
def _test_inner_stop(x):
    ops.stop_if(ops.LT([x, ops.Const(5)]))