Searched defs:grad_output0 (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/benchmarks/static_runtime/ |
| H A D | test_generated_ops.cc | 2000 auto grad_output0 = at::rand({6, 6, 6}); in TEST() local
2504 auto grad_output0 = at::rand({6, 6, 6}); in TEST() local
2664 auto grad_output0 = at::rand({6, 6, 6}); in TEST() local
2850 auto grad_output0 = at::rand({6, 6, 6}); in TEST() local
3078 auto grad_output0 = at::rand({6, 6, 6}); in TEST() local
5732 auto grad_output0 = at::rand({6, 6, 6}); in TEST() local
5836 auto grad_output0 = at::rand({6, 6, 6}); in TEST() local
5932 auto grad_output0 = at::rand({6, 6, 6}); in TEST() local
6034 auto grad_output0 = at::rand({6, 6, 6}); in TEST() local
6102 auto grad_output0 = at::rand({6, 6, 6}); in TEST() local
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | LossNLL2d.cpp | 85 const int64_t grad_output0 = grad_output.size(0); in check_gradout_shape_nll_loss2d() local
|