OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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