OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:DepthToSpaceOpFunctor
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
depthtospace_op.h
44
struct
DepthToSpaceOpFunctor
{
depthtospace_op_gpu.cu.cc
144
// Specialization of
DepthToSpaceOpFunctor
for a GPUDevice.
148
struct
DepthToSpaceOpFunctor
<GPUDevice, T, FORMAT_NHWC> {
174
struct
DepthToSpaceOpFunctor
<GPUDevice, T, FORMAT_NCHW> {
229
template struct functor::
DepthToSpaceOpFunctor
<GPUDevice, float, FORMAT_NCHW>;
230
template struct functor::
DepthToSpaceOpFunctor
<GPUDevice, float, FORMAT_NHWC>;
233
template struct functor::
DepthToSpaceOpFunctor
<GPUDevice, int32, FORMAT_NCHW>;
depthtospace_op.cc
120
functor::
DepthToSpaceOpFunctor
<GPUDevice, int32, FORMAT_NCHW> functor;
125
functor::
DepthToSpaceOpFunctor
<GPUDevice, T, FORMAT_NCHW> functor;
136
functor::
DepthToSpaceOpFunctor
<Device, T, FORMAT_NHWC> functor;
146
// Partial specialization of
DepthToSpaceOpFunctor
for a CPUDevice
150
struct
DepthToSpaceOpFunctor
<CPUDevice, T, FORMAT_NHWC> {
Completed in 69 milliseconds