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

  /external/tensorflow/tensorflow/core/kernels/
depthtospace_op.h 44 struct DepthToSpaceOpFunctor {
depthtospace_op_gpu.cu.cc 144 // Specialization of DepthToSpaceOpFunctor for a GPUDevice.
148 struct DepthToSpaceOpFunctor<GPUDevice, T, FORMAT_NHWC> {
174 struct DepthToSpaceOpFunctor<GPUDevice, T, FORMAT_NCHW> {
229 template struct functor::DepthToSpaceOpFunctor<GPUDevice, float, FORMAT_NCHW>;
230 template struct functor::DepthToSpaceOpFunctor<GPUDevice, float, FORMAT_NHWC>;
233 template struct functor::DepthToSpaceOpFunctor<GPUDevice, int32, FORMAT_NCHW>;
depthtospace_op.cc 120 functor::DepthToSpaceOpFunctor<GPUDevice, int32, FORMAT_NCHW> functor;
125 functor::DepthToSpaceOpFunctor<GPUDevice, T, FORMAT_NCHW> functor;
136 functor::DepthToSpaceOpFunctor<Device, T, FORMAT_NHWC> functor;
146 // Partial specialization of DepthToSpaceOpFunctor for a CPUDevice
150 struct DepthToSpaceOpFunctor<CPUDevice, T, FORMAT_NHWC> {

Completed in 69 milliseconds