Searched refs:create_symbolic_tensor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_dynamic_shapes.py | 164 def create_symbolic_tensor(name, arg, shape_env, source=None, dynamic_dims=None): function
267 x = create_symbolic_tensor("x", torch.randn(5, 4, 3), shape_env)
292 y = create_symbolic_tensor("y", torch.randn(5, 4, 3)[1:], shape_env)
298 x = create_symbolic_tensor("x", torch.randn(5, 4, 3), shape_env)
299 y = create_symbolic_tensor("y", torch.randn(5, 4, 3), shape_env)
307 y = create_symbolic_tensor("y2", torch.randn(1, 4, 1), shape_env)
315 x = create_symbolic_tensor("x", torch.randn(5, 4, 3), shape_env)
316 y = create_symbolic_tensor("y", torch.randn(5, 4, 1), shape_env)
331 x = create_symbolic_tensor("x", torch.randn(5, 4, 3), shape_env)
332 y = create_symbolic_tensor("y", torch.randn(1, 4, 1), shape_env)
[all …]
|