HomeSort by relevance Sort by last modified time
    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