pytorch

1
#!/usr/bin/env python3
2

3
import caffe2.python.hypothesis_test_util as hu
4
import hypothesis.strategies as st
5
import numpy as np
6
from caffe2.python import core, utils
7
from hypothesis import given
8

9

10
class TestAliasWithNameOp(hu.HypothesisTestCase):
11
    @given(
12
        shape=st.lists(st.integers(0, 5), min_size=1, max_size=3),
13
        dtype=st.sampled_from([np.float32, np.int64]),
14
        **hu.gcs
15
    )
16
    def test_alias_with_name_op(self, shape, dtype, dc, gc):
17
        test_input = (100 * np.random.random(shape)).astype(dtype)
18
        test_inputs = [test_input]
19

20
        alias_op = core.CreateOperator(
21
            "AliasWithName",
22
            ["input"],
23
            ["output"],
24
            device_option=gc,
25
        )
26
        alias_op.arg.add().CopyFrom(utils.MakeArgument("name", "whatever_name"))
27

28
        def reference_func(x):
29
            return (x,)
30

31
        self.assertReferenceChecks(gc, alias_op, test_inputs, reference_func)
32