Python FunctionBenchmark Examples

Example #1

File: test_reduction.py Project: zwsong/nnabla

def test_reduction_axis(inspecs, reduction, axis, nnabla_opts):
    func = getattr(F, reduction)
    fb = FunctionBenchmark(
        func, inspecs, [], dict(axis=axis),
        nnabla_opts.ext, nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #2

File: test_loss.py Project: zwsong/nnabla

def test_binary_classification_loss(inspecs, loss, nnabla_opts):
    func = getattr(F, loss)
    fb = FunctionBenchmark(
        func, inspecs, [], {},
        nnabla_opts.ext, nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #3

File: test_logical.py Project: zwsong/nnabla

def test_pairwise_logical(inspecs, op, nnabla_opts):
    func = getattr(F, op)
    fb = FunctionBenchmark(
        func, inspecs, [], {},
        nnabla_opts.ext, nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #4

File: test_embed.py Project: zwsong/nnabla

def test_embed(inspecs, n_inputs, n_features, nnabla_opts):
    fb = FunctionBenchmark(
        PF.embed, inspecs, [],
        dict(n_inputs=n_inputs, n_features=n_features),
        nnabla_opts.ext, nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #5

File: test_activation.py Project: zwsong/nnabla

def test_activation(inspecs, activation, nnabla_opts):
    func = getattr(F, activation)
    fb = FunctionBenchmark(
        func, inspecs, [], {},
        nnabla_opts.ext, nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #6

File: test_logical.py Project: zwsong/nnabla

def test_logical_not(inspecs, nnabla_opts):
    func = F.logical_not
    fb = FunctionBenchmark(
        func, inspecs, [], {},
        nnabla_opts.ext, nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #7

File: test_loss.py Project: zwsong/nnabla

def test_categorical_classification_loss(inspecs, loss, nnabla_opts):
    func = getattr(F, loss)
    fb = FunctionBenchmark(
        func, inspecs, [], dict(axis=1),
        nnabla_opts.ext, nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #8

File: test_pooling.py Project: zwsong/nnabla

def test_pooling(inspecs, pool, nnabla_opts):
    if pool == 'average':
        func = F.average_pooling
    elif pool == 'max':
        func = F.max_pooling
    fb = FunctionBenchmark(
        func, inspecs, [], dict(kernel=(2, 2), stride=(2, 2)),
        nnabla_opts.ext, nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #9

File: test_normalization_functions.py Project: sony/nnabla

def test_instance_normalization(inspec_and_axis, nnabla_opts):
    inspec, axis = inspec_and_axis
    fb = FunctionBenchmark(PF.instance_normalization, inspec, [],
                           dict(channel_axis=axis), nnabla_opts.ext,
                           nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #10

File: test_reduction.py Project: aswifi/nnabla

def test_reduction_axis(inspecs, reduction, axis, nnabla_opts):
    func = getattr(F, reduction)
    fb = FunctionBenchmark(
        func, inspecs, [], dict(axis=axis),
        nnabla_opts.ext, nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #11

File: test_embed.py Project: aswifi/nnabla

def test_embed(inspecs, n_inputs, n_features, nnabla_opts):
    fb = FunctionBenchmark(
        PF.embed, inspecs, [],
        dict(n_inputs=n_inputs, n_features=n_features),
        nnabla_opts.ext, nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #12

File: test_arithmetic.py Project: sony/nnabla

def test_pairwise_arithmetic(inspecs, op, nnabla_opts):
    func = getattr(F, op)
    fb = FunctionBenchmark(
        func, inspecs, [], {},
        nnabla_opts.ext, nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #13

File: test_pooling.py Project: aswifi/nnabla

def test_pooling(inspecs, pool, nnabla_opts):
    if pool == 'average':
        func = F.average_pooling
    elif pool == 'max':
        func = F.max_pooling
    fb = FunctionBenchmark(
        func, inspecs, [], dict(kernel=(2, 2), stride=(2, 2)),
        nnabla_opts.ext, nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #14

def test_cumprod(seed, test_case, exclusive, reverse, with_mask, nnabla_opts):
    x_shape = test_case.shape
    axis = test_case.axis

    def init(shape):
        rng = np.random.RandomState(seed)
        return create_cumprod_input(rng, shape, axis, with_mask)

    need_grad = True

    inputs = [Inspec(x_shape, init, need_grad)]

    func_kwargs = dict(
        axis=axis,
        exclusive=exclusive,
        reverse=reverse,
    )
    fb = FunctionBenchmark(F.cumprod, inputs, [], func_kwargs, nnabla_opts.ext,
                           nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #15

File: test_loss.py Project: sony/nnabla

def test_categorical_classification_loss(inspecs, loss, nnabla_opts):
    func = getattr(F, loss)
    fb = FunctionBenchmark(func, inspecs, [], dict(axis=1), nnabla_opts.ext,
                           nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #16

def test_softmax(inspecs, axis, nnabla_opts):
    fb = FunctionBenchmark(F.softmax, inspecs, [], dict(axis=axis),
                           nnabla_opts.ext, nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #17

File: test_logical.py Project: sony/nnabla

def test_logical_not(inspecs, nnabla_opts):
    func = F.logical_not
    fb = FunctionBenchmark(func, inspecs, [], {}, nnabla_opts.ext,
                           nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #18

File: test_convolution.py Project: aswifi/nnabla

def test_convolution(inputs, func_kwargs, nnabla_opts):
    fb = FunctionBenchmark(
        PF.convolution, inputs, [], func_kwargs,
        nnabla_opts.ext, nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #19

File: test_batch_normalization.py Project: stevevista/nbla-go-zero

def test_bn(inspecs, batch_stat, nnabla_opts):
    fb = FunctionBenchmark(PF.batch_normalization, inspecs, [],
                           dict(batch_stat=batch_stat), nnabla_opts.ext,
                           nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #20

File: test_logical.py Project: sony/nnabla

def test_scalar_logical(inspecs, op, nnabla_opts):
    func = getattr(F, op)
    fb = FunctionBenchmark(func, inspecs, [1], {}, nnabla_opts.ext,
                           nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #21

File: test_loss.py Project: sony/nnabla

def test_binary_classification_loss(inspecs, loss, nnabla_opts):
    func = getattr(F, loss)
    fb = FunctionBenchmark(func, inspecs, [], {}, nnabla_opts.ext,
                           nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #22

File: test_convolution.py Project: zwsong/nnabla

def test_convolution(inputs, func_kwargs, nnabla_opts):
    fb = FunctionBenchmark(
        PF.convolution, inputs, [], func_kwargs,
        nnabla_opts.ext, nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #23

File: test_affine.py Project: stevevista/nbla-go-zero

def test_affine(inspecs, n_outmaps, nnabla_opts):
    fb = FunctionBenchmark(PF.affine, inspecs, [], dict(n_outmaps=n_outmaps),
                           nnabla_opts.ext, nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #24

File: test_activation.py Project: stevevista/nbla-go-zero

def test_activation(inspecs, activation, nnabla_opts):
    func = getattr(F, activation)
    fb = FunctionBenchmark(func, inspecs, [], {}, nnabla_opts.ext,
                           nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #25

File: test_batch_normalization.py Project: zwsong/nnabla

def test_bn(inspecs, batch_stat, nnabla_opts):
    fb = FunctionBenchmark(
        PF.batch_normalization, inspecs, [], dict(batch_stat=batch_stat),
        nnabla_opts.ext, nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #26

File: test_normalization_functions.py Project: sony/nnabla

def test_layer_normalization(inspec_and_axis, nnabla_opts):
    inspec, axis = inspec_and_axis
    fb = FunctionBenchmark(PF.layer_normalization, inspec, [], dict(),
                           nnabla_opts.ext, nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #27

File: test_prelu.py Project: zwsong/nnabla

def test_activation(inspecs, shared, nnabla_opts):
    fb = FunctionBenchmark(
        PF.prelu, inspecs, [1], {},
        nnabla_opts.ext, nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #28

File: test_pad.py Project: aswifi/nnabla

def test_pad(inspecs, nnabla_opts):
    fb = FunctionBenchmark(F.pad, inspecs, [(10, 10, 10, 10), 'constant', 0.0],
                           {}, nnabla_opts.ext, nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #29

File: test_prelu.py Project: aswifi/nnabla

def test_activation(inspecs, shared, nnabla_opts):
    fb = FunctionBenchmark(PF.prelu, inspecs, [1], {}, nnabla_opts.ext,
                           nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #30

File: test_binary_operation.py Project: sony/nnabla

def test_mul2_with_broadcast(inspecs, op, nnabla_opts):
    func = getattr(F, op)
    fb = FunctionBenchmark(func, inspecs, [], {}, nnabla_opts.ext,
                           nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)

Example #31

File: test_transpose.py Project: sony/nnabla

def test_transpose(inspecs, axis, nnabla_opts):
    fb = FunctionBenchmark(F.transpose, inspecs, [axis], dict(),
                           nnabla_opts.ext, nnabla_opts.ext_kwargs)
    fb.benchmark()
    fb.write(writer=nnabla_opts.function_benchmark_writer)