Searched defs:bottom_diff (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | maxpooling_op_gpu.cu.cc | 207 dtype* __restrict__ bottom_diff, in MaxPoolBackward()
236 const dtype* __restrict__ top_diff, dtype* __restrict__ bottom_diff) { in MaxPoolGradBackwardNoMaskNCHW()
277 const dtype* __restrict__ top_diff, dtype* __restrict__ bottom_diff) { in MaxPoolGradBackwardNoMaskNHWC()
337 dtype* __restrict__ bottom_diff, in MaxPoolGradBackward()
408 T* bottom_diff, const Eigen::GpuDevice& d, in operator ()()
429 const int pad_l, const T* top_diff, T* bottom_diff, in operator ()()
457 T* bottom_diff, const Eigen::GpuDevice& d, in operator ()()
|
| H A D | pooling_ops_3d_gpu.cu.cc | 38 dtype* __restrict__ bottom_diff) { in MaxPoolGradBackwardNoMaskNCDHW()
89 dtype* __restrict__ bottom_diff) { in MaxPoolGradBackwardNoMaskNDHWC()
144 const T* top_diff, T* bottom_diff, const Eigen::GpuDevice& d) { in operator ()()
|
| H A D | avgpooling_op_gpu.cu.cc | 88 const int pad_l, T* const bottom_diff, in RunAvePoolBackwardNHWC()
|
| H A D | maxpooling_op.cc | 570 void SpatialMaxPoolGradGrad(OpKernelContext* context, Tensor* bottom_diff, in SpatialMaxPoolGradGrad()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/ |
| H A D | resize_nearest_neighbor_op_gpu.cu.cc | 99 T* __restrict__ bottom_diff) { in ResizeNearestNeighborBackwardNHWC()
130 T* __restrict__ bottom_diff) { in LegacyResizeNearestNeighborBackwardNHWC()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | AveragePool2d.cu | 136 scalar_t* const bottom_diff, const int divisor_override, in avg_pool2d_backward_out_cuda_frame()
193 scalar_t* const bottom_diff, const int divisor_override, in avg_pool2d_backward_out_cuda_frame_nhwc()
|
| H A D | DilatedMaxPool2d.cu | 183 scalar_t* bottom_diff) { in max_pool_backward_nchw()
|