Searched defs:SpaceToBatchFunctor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | spacetobatch_functor.h | 86 struct SpaceToBatchFunctor { struct
87 using InputT = typename std::conditional<B2S, T, const T>::type;
88 using OutputT = typename std::conditional<B2S, const T, T>::type;
|
| D | spacetobatch_functor.cc | 88 struct SpaceToBatchFunctor<CPUDevice, T, NUM_BLOCK_DIMS, B2S> { struct
89 using SpaceT = typename std::conditional<B2S, T, const T>::type;
90 using BatchT = typename std::conditional<B2S, const T, T>::type;
91 Status operator()( in operator ()()
|
| D | spacetobatch_functor_gpu.cu.cc | 102 struct SpaceToBatchFunctor<GPUDevice, T, NUM_BLOCK_DIMS, B2S> { struct
103 using SpaceT = typename std::conditional<B2S, T, const T>::type;
104 using BatchT = typename std::conditional<B2S, const T, T>::type;
105 Status operator()( in operator ()()
|