Home
last modified time | relevance | path

Searched refs:InflatePadAndShuffle (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_gpu_2.cu.cc28 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
Dconv_2d.h43 struct InflatePadAndShuffle { struct
Dconv_grad_filter_ops.cc918 void InflatePadAndShuffle<GPUDevice, T, 4, int>::operator()( \
924 extern template struct InflatePadAndShuffle<GPUDevice, T, 4, int>; \
Dconv_grad_input_ops.cc1061 void InflatePadAndShuffle<GPUDevice, T, 4, int>::operator()( \
1067 extern template struct InflatePadAndShuffle<GPUDevice, T, 4, int>; \