Home
last modified time | relevance | path

Searched refs:ResizeNearestNeighbor (Results 1 – 25 of 34) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/image/
Dresize_op_benchmark_test.cc58 BM_ResizeDev(cpu, ResizeNearestNeighbor, 10, 499, 499, 1, 250, 250);
59 BM_ResizeDev(cpu, ResizeNearestNeighbor, 10, 499, 499, 3, 250, 250);
60 BM_ResizeDev(cpu, ResizeNearestNeighbor, 10, 499, 499, 1, 998, 998);
61 BM_ResizeDev(cpu, ResizeNearestNeighbor, 10, 499, 499, 3, 998, 998);
68 BM_ResizeDev(gpu, ResizeNearestNeighbor, 10, 499, 499, 3, 998, 998);
Dresize_nearest_neighbor_op_gpu.cu.cc160 struct ResizeNearestNeighbor<GPUDevice, T, half_pixel_centers, align_corners> { struct
193 template struct ResizeNearestNeighbor<GPUDevice, T, false, false>; \
194 template struct ResizeNearestNeighbor<GPUDevice, T, false, true>; \
195 template struct ResizeNearestNeighbor<GPUDevice, T, true, false>; \
196 template struct ResizeNearestNeighbor<GPUDevice, T, true, true>;
Dresize_nearest_neighbor_op.cc68 status = functor::ResizeNearestNeighbor<Device, T, in Compute()
74 status = functor::ResizeNearestNeighbor<Device, T, in Compute()
82 status = functor::ResizeNearestNeighbor<Device, T, in Compute()
88 status = functor::ResizeNearestNeighbor<Device, T, in Compute()
184 struct ResizeNearestNeighbor<CPUDevice, T, half_pixel_centers, align_corners> { struct
Dresize_nearest_neighbor_op.h27 struct ResizeNearestNeighbor { struct
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DResizeNearestNeighbor.pbtxt2 name: "ResizeNearestNeighbor"
40 name: "ResizeNearestNeighbor"
79 name: "ResizeNearestNeighbor"
125 name: "ResizeNearestNeighbor"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DResizeNearestNeighbor.pbtxt2 name: "ResizeNearestNeighbor"
40 name: "ResizeNearestNeighbor"
79 name: "ResizeNearestNeighbor"
125 name: "ResizeNearestNeighbor"
/external/tensorflow/tensorflow/lite/kernels/
Dresize_nearest_neighbor.cc110 reference_ops::ResizeNearestNeighbor( in Eval()
116 reference_ops::ResizeNearestNeighbor( in Eval()
122 optimized_ops::ResizeNearestNeighbor( in Eval()
128 reference_ops::ResizeNearestNeighbor( in Eval()
133 reference_ops::ResizeNearestNeighbor( in Eval()
/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_ResizeNearestNeighbor.pbtxt2 graph_op_name: "ResizeNearestNeighbor"
4 name: "image.ResizeNearestNeighbor"
/external/tensorflow/tensorflow/lite/kernels/internal/
Dresize_nearest_neighbor_test.cc39 reference_ops::ResizeNearestNeighbor( in TestReferenceResizeNearestNeighbor()
367 reference_ops::ResizeNearestNeighbor( in TestOptimizedResizeNearestNeighbor()
370 optimized_ops::ResizeNearestNeighbor( in TestOptimizedResizeNearestNeighbor()
376 reference_ops::ResizeNearestNeighbor( in TestOptimizedResizeNearestNeighbor()
379 optimized_ops::ResizeNearestNeighbor( in TestOptimizedResizeNearestNeighbor()
386 reference_ops::ResizeNearestNeighbor( in TestOptimizedResizeNearestNeighbor()
389 optimized_ops::ResizeNearestNeighbor( in TestOptimizedResizeNearestNeighbor()
396 reference_ops::ResizeNearestNeighbor( in TestOptimizedResizeNearestNeighbor()
399 optimized_ops::ResizeNearestNeighbor( in TestOptimizedResizeNearestNeighbor()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dresize_nearest_neighbor.cc74 reference_ops::ResizeNearestNeighbor( in Eval()
82 reference_ops::ResizeNearestNeighbor( in Eval()
90 reference_ops::ResizeNearestNeighbor( in Eval()
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_ResizeNearestNeighbor.pbtxt2 graph_op_name: "ResizeNearestNeighbor"
/external/tensorflow/tensorflow/cc/gradients/
Dimage_grad_test.cc33 using ops::ResizeNearestNeighbor;
60 *y = ResizeNearestNeighbor( in MakeOp()
62 ResizeNearestNeighbor::AlignCorners(align_corners)); in MakeOp()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_ResizeNearestNeighbor.pbtxt2 graph_op_name: "ResizeNearestNeighbor"
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dresize_nearest_neighbor.h49 inline void ResizeNearestNeighbor( in ResizeNearestNeighbor() function
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dop_select_allowlist.md394 * `raw_ops.ResizeNearestNeighbor`
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dlower_tf.mlir840 …%resize = "tf.ResizeNearestNeighbor"(%arg0, %shape) {align_corners = false, device = "", half_pixe…
884 …%resize = "tf.ResizeNearestNeighbor"(%arg0, %shape) {align_corners = false, device = "", half_pixe…
938 …%resize = "tf.ResizeNearestNeighbor"(%arg0, %arg1) {align_corners = false, device = "", half_pixel…
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc1130 class ResizeNearestNeighbor class
1913 ops.push_back(MakeUnique<ResizeNearestNeighbor>( in BuildOperatorList()
Doperator_test.cc436 TEST_F(OperatorTest, ResizeNearestNeighbor) { in TEST_F() argument
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md2515 ### tf.ResizeNearestNeighbor
2522 %output = tf.ResizeNearestNeighbor(%images, %size) {align_corners, half_pixel_centers}
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc118 DECL_CONVERT_OP(ResizeNearestNeighbor);
Dlegalize_tfl.cc123 DECL_CONVERT_OP(ResizeNearestNeighbor);
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc441 HANDLE_OPERATORTYPENAME_CASE(ResizeNearestNeighbor) in OperatorTypeName()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dlegalize-tf.mlir1424 …%0 = "tf.ResizeNearestNeighbor"(%arg0, %arg1) {align_corners = true} : (tensor<1x100x100x3xf32>, t…
1431 …%0 = "tf.ResizeNearestNeighbor"(%arg0, %arg1) {align_corners = false, half_pixel_centers = true} :…
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h6023 inline void ResizeNearestNeighbor( in ResizeNearestNeighbor() function
6031 reference_ops::ResizeNearestNeighbor( in ResizeNearestNeighbor()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc7094 TestConvertResize<ops::ResizeNearestNeighbor>(this, p); in TEST_P()

12