Searched refs:InflatePadAndShuffle (Results 1 – 4 of 4) 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
|
D | conv_grad_filter_ops.cc | 918 void InflatePadAndShuffle<GPUDevice, T, 4, int>::operator()( \ 924 extern template struct InflatePadAndShuffle<GPUDevice, T, 4, int>; \
|
D | conv_grad_input_ops.cc | 1061 void InflatePadAndShuffle<GPUDevice, T, 4, int>::operator()( \ 1067 extern template struct InflatePadAndShuffle<GPUDevice, T, 4, int>; \
|