Searched defs:SpaceToDepthOpFunctor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | spacetodepth_op_gpu.cu.cc | 144 struct SpaceToDepthOpFunctor<GPUDevice, T, FORMAT_NHWC> { struct
145 void operator()(const GPUDevice& d, typename TTypes<T, 4>::ConstTensor input, in operator ()()
167 void operator()(const GPUDevice& d, typename TTypes<T, 5>::ConstTensor input, in operator ()()
174 struct SpaceToDepthOpFunctor<GPUDevice, T, FORMAT_NCHW> { struct
175 void operator()(const GPUDevice& d, typename TTypes<T, 4>::ConstTensor input, in operator ()()
231 void operator()(const GPUDevice& d, typename TTypes<T, 5>::ConstTensor input, in operator ()()
239 template struct functor::SpaceToDepthOpFunctor<GPUDevice, float, FORMAT_NCHW>; member in tensorflow::functor
240 template struct functor::SpaceToDepthOpFunctor<GPUDevice, float, FORMAT_NHWC>; member in tensorflow::functor
243 template struct functor::SpaceToDepthOpFunctor<GPUDevice, Eigen::half, member in tensorflow::functor
245 template struct functor::SpaceToDepthOpFunctor<GPUDevice, Eigen::half, member in tensorflow::functor
[all …]
|
| D | spacetodepth_op.h | 45 struct SpaceToDepthOpFunctor { struct
|
| D | spacetodepth_op.cc | 151 struct SpaceToDepthOpFunctor<CPUDevice, T, FORMAT_NHWC> { struct
152 void operator()(const CPUDevice& d, typename TTypes<T, 4>::ConstTensor input, in operator ()()
|