Searched defs:grad_output1 (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/benchmarks/static_runtime/ |
| H A D | test_generated_ops.cc | 2013 auto grad_output1 = at::rand({22, 22, 22}); in TEST() local
2516 auto grad_output1 = at::rand({22, 22, 22}); in TEST() local
2675 auto grad_output1 = at::rand({22, 22, 22}); in TEST() local
2863 auto grad_output1 = at::rand({22, 22, 22}); in TEST() local
3090 auto grad_output1 = at::rand({22, 22, 22}); in TEST() local
5748 auto grad_output1 = at::rand({22, 22, 22}); in TEST() local
5847 auto grad_output1 = at::rand({22, 22, 22}); in TEST() local
5946 auto grad_output1 = at::rand({22, 22, 22}); in TEST() local
6047 auto grad_output1 = at::rand({22, 22, 22}); in TEST() local
6114 auto grad_output1 = at::rand({22, 22, 22}); in TEST() local
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | LossNLL2d.cpp | 86 const int64_t grad_output1 = grad_output.size(1); in check_gradout_shape_nll_loss2d() local
|