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

  /external/tensorflow/tensorflow/core/kernels/
spacetobatch_functor.h 86 struct SpaceToBatchFunctor {
spacetobatch_functor.cc 16 // Specialization of SpaceToBatchFunctor for a CPUDevice.
88 struct SpaceToBatchFunctor<CPUDevice, T, NUM_BLOCK_DIMS, B2S> {
155 template struct SpaceToBatchFunctor<CPUDevice, T, NUM_BLOCK_DIMS, false>; \
156 template struct SpaceToBatchFunctor<CPUDevice, T, NUM_BLOCK_DIMS, true>; \
spacetobatch_functor_gpu.cu.cc 16 // Specialization of SpaceToBatchFunctor for a GPUDevice.
102 struct SpaceToBatchFunctor<GPUDevice, T, NUM_BLOCK_DIMS, B2S> {
154 template struct SpaceToBatchFunctor<GPUDevice, T, NUM_BLOCK_DIMS, false>; \
155 template struct SpaceToBatchFunctor<GPUDevice, T, NUM_BLOCK_DIMS, true>; \

Completed in 113 milliseconds