Searched defs:InflatePadAndShuffle (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_gpu_2.cu.cc | 28 template struct functor::InflatePadAndShuffle<GPUDevice, double, 4, int>; member in tensorflow::functor
29 template struct functor::InflatePadAndShuffle<GPUDevice, double, 4, member in tensorflow::functor
31 template struct functor::InflatePadAndShuffle<GPUDevice, float, 4, int>; member in tensorflow::functor
32 template struct functor::InflatePadAndShuffle<GPUDevice, float, 4, member in tensorflow::functor
34 template struct functor::InflatePadAndShuffle<GPUDevice, Eigen::half, 4, int>; member in tensorflow::functor
35 template struct functor::InflatePadAndShuffle<GPUDevice, Eigen::half, 4, member in tensorflow::functor
|
| D | conv_2d.h | 43 struct InflatePadAndShuffle { struct
44 void operator()( in operator()
|