OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:ResizeNearestNeighbor
(Results
1 - 4
of
4
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
resize_nearest_neighbor_op.h
26
struct
ResizeNearestNeighbor
{
resize_op_benchmark_test.cc
51
BM_ResizeDev(cpu,
ResizeNearestNeighbor
, 10, 499, 499);
52
BM_ResizeDev(gpu,
ResizeNearestNeighbor
, 10, 499, 499);
resize_nearest_neighbor_op_gpu.cu.cc
97
// Partial specialization of
ResizeNearestNeighbor
functor for a GPUDevice.
99
struct
ResizeNearestNeighbor
<GPUDevice, T, align_corners> {
124
template struct
ResizeNearestNeighbor
<GPUDevice, T, false>; \
125
template struct
ResizeNearestNeighbor
<GPUDevice, T, true>;
resize_nearest_neighbor_op.cc
65
functor::
ResizeNearestNeighbor
<Device, T, /*align_corners=*/true>()(
70
functor::
ResizeNearestNeighbor
<Device, T, /*align_corners=*/false>()(
76
errors::Internal("Failed launching
ResizeNearestNeighbor
"));
84
// Partial specialization of
ResizeNearestNeighbor
functor for a CPUDevice.
87
struct
ResizeNearestNeighbor
<CPUDevice, T, align_corners> {
235
REGISTER_KERNEL_BUILDER(Name("
ResizeNearestNeighbor
") \
253
REGISTER_KERNEL_BUILDER(Name("
ResizeNearestNeighbor
") \
Completed in 321 milliseconds