Home
last modified time | relevance | path

Searched defs:DepthToSpaceOpFunctor (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddepthtospace_op_gpu.cu.cc148 struct DepthToSpaceOpFunctor<GPUDevice, T, FORMAT_NHWC> { struct
149 void operator()(const GPUDevice& d, typename TTypes<T, 4>::ConstTensor input, in operator ()()
171 void operator()(const GPUDevice& d, typename TTypes<T, 5>::ConstTensor input, in operator ()()
178 struct DepthToSpaceOpFunctor<GPUDevice, T, FORMAT_NCHW> { struct
179 void operator()(const GPUDevice& d, typename TTypes<T, 4>::ConstTensor input, in operator ()()
235 void operator()(const GPUDevice& d, typename TTypes<T, 5>::ConstTensor input, in operator ()()
243 template struct functor::DepthToSpaceOpFunctor<GPUDevice, float, FORMAT_NCHW>; member in tensorflow::functor
244 template struct functor::DepthToSpaceOpFunctor<GPUDevice, float, FORMAT_NHWC>; member in tensorflow::functor
247 template struct functor::DepthToSpaceOpFunctor<GPUDevice, Eigen::half, member in tensorflow::functor
249 template struct functor::DepthToSpaceOpFunctor<GPUDevice, Eigen::half, member in tensorflow::functor
[all …]
Ddepthtospace_op.h44 struct DepthToSpaceOpFunctor { struct
Ddepthtospace_op.cc150 struct DepthToSpaceOpFunctor<CPUDevice, T, FORMAT_NHWC> { struct
151 void operator()(const CPUDevice& d, typename TTypes<T, 4>::ConstTensor input, in operator ()()