Searched defs:gout (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | UpSampleMoreKernel.cpp | 20 void inline nearest_channels_last_acc(acc_t* gin, scalar_t* gout, int64_t size) { in nearest_channels_last_acc()
37 void inline nearest_channels_last_acc(acc_t* gin, scalar_t* gout, int64_t size) { in nearest_channels_last_acc()
57 void inline linear_channels_last_acc(acc_t* gin, const scalar_t* gout, acc_t w, int64_t size) { in linear_channels_last_acc()
74 void inline linear_channels_last_acc(acc_t* gin, const scalar_t* gout, acc_t w, int64_t size) { in linear_channels_last_acc()
|
| D | AdaptiveAvgPoolKernel.cpp | 341 const scalar_t* gout = grad_output_ptr + oh * output_width * channels + ow * channels; in cpu_adaptive_avg_pool2d_backward_channels_last() local
781 …scalar_t* gout = grad_output_ptr + od * output_depth * channels + oh * output_width * channels + o… in cpu_adaptive_avg_pool3d_backward_channels_last() local
|
| D | AdaptiveMaxPoolKernel.cpp | 419 const scalar_t* gout = grad_output_ptr + oh * output_width * channels + ow * channels; in cpu_adaptive_max_pool2d_backward_channels_last() local
916 …scalar_t* gout = grad_output_ptr + od * output_height * output_width * channels + oh * output_widt… in cpu_adaptive_max_pool3d_backward_channels_last() local
|
| D | AvgPoolKernel.cpp | 469 const scalar_t* gout = grad_output_ptr + oh * output_width * channels + ow * channels; in cpu_avg_pool2d_backward_channels_last() local
1045 …scalar_t* gout = grad_output_ptr + od * output_height * output_width * channels + oh * output_widt… in cpu_avg_pool3d_backward_channels_last() local
|
| D | MaxPoolKernel.cpp | 579 …const scalar_t* gout = grad_output_ptr + (od * output_height * output_width + oh * output_width + … in cpu_max_pool_backward_channels_last() local
|
| /external/pytorch/test/functorch/ |
| D | test_vmap.py | 5067 def push_vjp(leaf, gout): argument
|