Searched defs:top_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 | 204 const dtype* __restrict__ top_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()
333 const dtype* __restrict__ top_diff, in MaxPoolGradBackward()
406 const int output_size, const int input_size, const T* top_diff, in operator ()()
429 const int pad_l, const T* top_diff, T* bottom_diff, in operator ()()
455 const int output_size, const int input_size, const T* top_diff, in operator ()()
|
| H A D | pooling_ops_3d_gpu.cu.cc | 37 const int pad_l, const dtype* __restrict__ top_diff, in MaxPoolGradBackwardNoMaskNCDHW()
88 const int pad_l, const dtype* __restrict__ top_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 | 82 bool RunAvePoolBackwardNHWC(const T* const top_diff, const int num, in RunAvePoolBackwardNHWC()
|
| H A D | maxpooling_op.cc | 572 const Tensor& top_diff, in SpatialMaxPoolGradGrad()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/ |
| H A D | resize_nearest_neighbor_op_gpu.cu.cc | 96 const int nthreads, const T* __restrict__ top_diff, const int in_height, in ResizeNearestNeighborBackwardNHWC()
127 const int nthreads, const T* __restrict__ top_diff, const int in_height, in LegacyResizeNearestNeighborBackwardNHWC()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | AveragePool2d.cu | 131 …l__ void avg_pool2d_backward_out_cuda_frame(const index_t nthreads, const scalar_t* const top_diff, in avg_pool2d_backward_out_cuda_frame()
188 const scalar_t* const top_diff, in avg_pool2d_backward_out_cuda_frame_nhwc()
|
| H A D | DilatedMaxPool2d.cu | 177 __global__ void max_pool_backward_nchw(const scalar_t* top_diff, in max_pool_backward_nchw()
|