HomeSort by relevance Sort by last modified time
    Searched refs:InflatePadAndShuffle (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
conv_ops_gpu_2.cu.cc 28 template struct functor::InflatePadAndShuffle<GPUDevice, float, 4, int>;
29 template struct functor::InflatePadAndShuffle<GPUDevice, float, 4,
31 template struct functor::InflatePadAndShuffle<GPUDevice, Eigen::half, 4, int>;
32 template struct functor::InflatePadAndShuffle<GPUDevice, Eigen::half, 4,
conv_2d.h 43 struct InflatePadAndShuffle {
conv_grad_filter_ops.cc     [all...]
conv_grad_input_ops.cc     [all...]
conv_grad_ops_3d.cc 199 functor::InflatePadAndShuffle<Device, T, 5, Eigen::DenseIndex>()(
321 functor::InflatePadAndShuffle<Device, T, 5, Eigen::DenseIndex>()(
    [all...]

Completed in 687 milliseconds