Lines Matching full:ones
65 torch::Tensor c = torch::ones({1, 2, 3}); in TEST()
71 torch::Tensor tmp = torch::ones({1, 2, 3}).set_requires_grad(true); in TEST()
75 tmp = torch::ones({1, 2, 3}).set_requires_grad(false); in TEST()
82 torch::Tensor s = torch::ones({1, 2, 3}).set_requires_grad(true); in TEST()
101 torch::Tensor c = torch::ones({1, 2, 3}).set_requires_grad(requires_grad); in TEST()
112 torch::Tensor c = torch::ones({1, 2, 3}).set_requires_grad(requires_grad); in TEST()
123 torch::Tensor c = torch::ones({1, 2, 3}).set_requires_grad(requires_grad); in TEST()
139 torch::ones({1, 2, 3}).set_requires_grad(requires_grad); in TEST()
160 torch::ones({1, 2, 3}).set_requires_grad(requires_grad); in TEST()
175 torch::ones({1, 2, 3}).set_requires_grad(requires_grad); in TEST()
188 torch::Tensor s = torch::ones({1, 2, 3}).set_requires_grad(requires_grad); in TEST()
214 torch::Tensor s = torch::ones({1, 2, 3}).set_requires_grad(requires_grad); in TEST()
242 torch::Tensor s = torch::ones({1, 2, 3}).set_requires_grad(requires_grad); in TEST()
287 torch::Tensor s = torch::ones({1, 2, 3}).set_requires_grad(requires_grad); in TEST()
321 torch::Tensor s = torch::ones({1, 2, 3}).set_requires_grad(requires_grad); in TEST()
325 c = torch::ones({1, 2, 3}).set_requires_grad(requires_grad); in TEST()
362 torch::Tensor s = torch::ones({1, 2, 3}).set_requires_grad(requires_grad); in TEST()
367 c = torch::ones({1, 2, 3}); in TEST()
394 torch::Tensor s = torch::ones({1, 2, 3}).set_requires_grad(requires_grad); in TEST()
398 c = torch::ones({1, 2, 3}); in TEST()
420 torch::Tensor s = torch::ones({1, 2, 3}).set_requires_grad(requires_grad); in TEST()
439 torch::Tensor s = torch::ones({1, 2, 3}).set_requires_grad(requires_grad); in TEST()
458 torch::Tensor s = torch::ones({1, 2, 3}).set_requires_grad(true); in TEST()
477 torch::Tensor s = torch::ones({2, 2, 3}).set_requires_grad(true); in TEST()
503 torch::Tensor s = torch::ones({1, 2, 3}).set_requires_grad(requires_grad); in TEST()
507 t = torch::ones({1, 2, 3}); in TEST()
523 t = torch::ones({1, 2, 3}); in TEST()
535 t = torch::ones({1, 2, 3}); in TEST()
556 torch::Tensor s = torch::ones({1, 2, 3}); in TEST()
560 t = torch::ones({1, 2, 3}); in TEST()
579 torch::Tensor s = torch::ones({3, 3, 2}); in TEST()
590 torch::Tensor e = torch::ones({3, 3, 2}); in TEST()
602 s = torch::ones({3, 3, 2}); in TEST()
639 torch::Tensor var1 = torch::ones({3, 3}).set_requires_grad(true); in TEST()