Searched refs:SpaceToBatchFunctor (Results 1 - 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| H A D | spacetobatch_functor.cc | 16 // Specialization of SpaceToBatchFunctor for a CPUDevice.
88 struct SpaceToBatchFunctor<CPUDevice, T, NUM_BLOCK_DIMS, B2S> { struct in namespace:tensorflow::functor
155 template struct SpaceToBatchFunctor<CPUDevice, T, NUM_BLOCK_DIMS, false>; \
156 template struct SpaceToBatchFunctor<CPUDevice, T, NUM_BLOCK_DIMS, true>; \
|
| H A D | spacetobatch_functor_gpu.cu.cc | 16 // Specialization of SpaceToBatchFunctor for a GPUDevice.
102 struct SpaceToBatchFunctor<GPUDevice, T, NUM_BLOCK_DIMS, B2S> { struct in namespace:tensorflow::functor
154 template struct SpaceToBatchFunctor<GPUDevice, T, NUM_BLOCK_DIMS, false>; \
155 template struct SpaceToBatchFunctor<GPUDevice, T, NUM_BLOCK_DIMS, true>; \
|
| H A D | spacetobatch_functor.h | 86 struct SpaceToBatchFunctor { struct in namespace:tensorflow::functor
|
| H A D | batchtospace_op.cc | 189 (functor::SpaceToBatchFunctor<Device, T, NUM_BLOCK_DIMS, true>()( \
|
| H A D | spacetobatch_op.cc | 188 (functor::SpaceToBatchFunctor<Device, T, NUM_BLOCK_DIMS, false>()( \
|
Completed in 365 milliseconds