Searched refs:SpaceToBatchFunctor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | spacetobatch_functor.cc | 88 struct SpaceToBatchFunctor<CPUDevice, T, NUM_BLOCK_DIMS, B2S> { struct
155 template struct SpaceToBatchFunctor<CPUDevice, T, NUM_BLOCK_DIMS, false>; \
156 template struct SpaceToBatchFunctor<CPUDevice, T, NUM_BLOCK_DIMS, true>; \
|
| D | spacetobatch_functor_gpu.cu.cc | 102 struct SpaceToBatchFunctor<GPUDevice, T, NUM_BLOCK_DIMS, B2S> { struct
154 template struct SpaceToBatchFunctor<GPUDevice, T, NUM_BLOCK_DIMS, false>; \
155 template struct SpaceToBatchFunctor<GPUDevice, T, NUM_BLOCK_DIMS, true>; \
|
| D | spacetobatch_functor.h | 86 struct SpaceToBatchFunctor { struct
|
| D | spacetobatch_op.cc | 190 functor::SpaceToBatchFunctor<Device, T, NUM_BLOCK_DIMS, false>()( \ in SpaceToBatchOpCompute()
|
| D | batchtospace_op.cc | 189 (functor::SpaceToBatchFunctor<Device, T, NUM_BLOCK_DIMS, true>()( \ in BatchToSpaceOpCompute()
|