Searched defs:bottom_diff (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| 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 ()()
|
| 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 ()()
|
| D | avgpooling_op_gpu.cu.cc | 88 const int pad_l, T* const bottom_diff, in RunAvePoolBackwardNHWC()
|
| D | maxpooling_op.cc | 546 void SpatialMaxPoolGradGrad(OpKernelContext* context, Tensor* bottom_diff, in SpatialMaxPoolGradGrad()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | resize_nearest_neighbor_op_gpu.cu.cc | 99 T* __restrict__ bottom_diff) { in ResizeNearestNeighborBackwardNHWC()
130 T* __restrict__ bottom_diff) { in LegacyResizeNearestNeighborBackwardNHWC()
|