Searched refs:exp_tensor (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/aten/src/ATen/test/ |
| D | atest.cpp | 127 auto exp_tensor = tensor({0, 1, 0, 1, 0}); in TEST_F() local
129 logical_and_out, x_logical, y_logical, exp_tensor, INTBOOL); in TEST_F()
133 auto exp_tensor = tensor({1, 1, 0, 1, 1}); in TEST_F() local
135 logical_or_out, x_logical, y_logical, exp_tensor, INTBOOL); in TEST_F()
139 auto exp_tensor = tensor({1, 0, 0, 0, 1}); in TEST_F() local
141 logical_xor_out, x_logical, y_logical, exp_tensor, INTBOOL); in TEST_F()
145 auto exp_tensor = tensor({0, 0, 0, 0, 1}); in TEST_F() local
148 lt_out, x_logical, y_logical, exp_tensor, INTBOOL); in TEST_F()
152 auto exp_tensor = tensor({0, 1, 1, 1, 1}); in TEST_F() local
155 le_out, x_logical, y_logical, exp_tensor, INTBOOL); in TEST_F()
[all …]
|
| /external/pytorch/test/distributed/_tensor/ |
| D | test_init.py | 83 exp_tensor = init_op(tensor_size, *args, **kwargs)
84 eq_op(exp_tensor, dist_tensor.to_local())
|
| /external/pytorch/test/ |
| D | test_binary_ufuncs.py | 1525 exp_tensor = make_tensor(
1529 exp_tensor = make_tensor(
1533 self._test_pow(base_tensor, exp_tensor)
1550 exp_tensor = make_tensor(
1559 exp_tensor = make_tensor(
1568 self._test_pow(base_tensor, exp_tensor)
|