Home
last modified time | relevance | path

Searched refs:align_corners (Results 1 – 25 of 81) sorted by relevance

1234

/external/tensorflow/tensorflow/cc/gradients/
Dimage_grad_test.cc55 const bool align_corners, const bool half_pixel_centers, in MakeOp() argument
62 ResizeNearestNeighbor::AlignCorners(align_corners)); in MakeOp()
66 ResizeBilinear::AlignCorners(align_corners) in MakeOp()
71 ResizeBicubic::AlignCorners(align_corners) in MakeOp()
79 void TestResizedShapeForType(const OpType op_type, const bool align_corners, in TestResizedShapeForType() argument
84 MakeOp<T>(op_type, x_data, {4, 6}, align_corners, half_pixel_centers, &x, in TestResizedShapeForType()
96 for (const bool align_corners : {true, false}) { in TestResizedShape()
97 if (half_pixel_centers && align_corners) { in TestResizedShape()
100 TestResizedShapeForType<Eigen::half>(op_type, align_corners, in TestResizedShape()
102 TestResizedShapeForType<float>(op_type, align_corners, in TestResizedShape()
[all …]
Dimage_grad.cc29 bool align_corners; in ResizeNearestNeighborGradHelper() local
31 GetNodeAttr(op.node()->attrs(), "align_corners", &align_corners)); in ResizeNearestNeighborGradHelper()
41 internal::ResizeNearestNeighborGrad::AlignCorners(align_corners) in ResizeNearestNeighborGradHelper()
51 bool align_corners; in ResizeBilinearGradHelper() local
53 GetNodeAttr(op.node()->attrs(), "align_corners", &align_corners)); in ResizeBilinearGradHelper()
59 internal::ResizeBilinearGrad::AlignCorners(align_corners) in ResizeBilinearGradHelper()
69 bool align_corners; in ResizeBicubicGradHelper() local
71 GetNodeAttr(op.node()->attrs(), "align_corners", &align_corners)); in ResizeBicubicGradHelper()
78 internal::ResizeBicubicGrad::AlignCorners(align_corners) in ResizeBicubicGradHelper()
/external/tensorflow/tensorflow/core/kernels/
Dresize_nearest_neighbor_op_gpu.cu.cc65 template <typename T, bool align_corners>
82 min((align_corners) ? static_cast<int>(roundf(out_y * height_scale)) in LegacyResizeNearestNeighborNHWC()
86 min((align_corners) ? static_cast<int>(roundf(out_x * width_scale)) in LegacyResizeNearestNeighborNHWC()
125 template <typename T, bool align_corners>
142 min((align_corners) ? static_cast<int>(roundf(in_y * height_scale)) in LegacyResizeNearestNeighborBackwardNHWC()
146 min((align_corners) ? static_cast<int>(roundf(in_x * width_scale)) in LegacyResizeNearestNeighborBackwardNHWC()
159 template <typename T, bool half_pixel_centers, bool align_corners>
160 struct ResizeNearestNeighbor<GPUDevice, T, half_pixel_centers, align_corners> {
185 LegacyResizeNearestNeighborNHWC<T, align_corners>, config.block_count, in operator ()()
205 template <typename T, bool half_pixel_centers, bool align_corners>
[all …]
Dimage_resizer_state.h42 bool align_corners) { in CalculateResizeScale() argument
43 return (align_corners && out_size > 1) in CalculateResizeScale()
70 explicit ImageResizerState(bool align_corners, bool half_pixel_centers) in ImageResizerState()
71 : align_corners_(align_corners), in ImageResizerState()
161 explicit ImageResizerGradientState(bool align_corners, in ImageResizerGradientState()
163 : align_corners_(align_corners), in ImageResizerGradientState()
Dresize_nearest_neighbor_op.cc132 template <typename T, bool half_pixel_centers, bool align_corners>
133 struct ResizeNearestNeighbor<CPUDevice, T, half_pixel_centers, align_corners> {
152 (align_corners) in operator ()()
161 (align_corners) in operator ()()
179 (align_corners) in operator ()()
187 (align_corners) in operator ()()
313 template <typename T, bool half_pixel_centers, bool align_corners>
315 align_corners> {
332 (align_corners) in operator ()()
338 (align_corners) in operator ()()
Dresize_nearest_neighbor_op.h26 bool align_corners>
34 bool align_corners>
/external/XNNPACK/test/
Dresize-bilinear-nhwc.cc168 .align_corners(true) in TEST()
182 .align_corners(true) in TEST()
196 .align_corners(true) in TEST()
249 .align_corners(true) in TEST()
264 .align_corners(true) in TEST()
280 .align_corners(true) in TEST()
297 .align_corners(true) in TEST()
314 .align_corners(true) in TEST()
494 .align_corners(true) in TEST()
509 .align_corners(true) in TEST()
[all …]
Dresize-bilinear-operator-tester.h81 if (align_corners() && output_height() > 1) { in height_scale()
89 if (align_corners() && output_width() > 1) { in width_scale()
196 inline ResizeBilinearOperatorTester& align_corners(bool align_corners) { in align_corners() argument
197 this->align_corners_ = align_corners; in align_corners()
201 inline bool align_corners() const { in align_corners() function
266 …(align_corners() ? XNN_FLAG_ALIGN_CORNERS : 0) | (tf_legacy_mode() ? XNN_FLAG_TENSORFLOW_LEGACY_MO… in TestF32()
/external/tensorflow/tensorflow/python/ops/
Dimage_grad_test.py96 for align_corners in [True, False]:
100 input_tensor, out_shape[1:3], align_corners=align_corners)
107 input_tensor, out_shape[1:3], align_corners=align_corners)
168 for align_corners in align_corner_options:
174 input_tensor, out_shape[1:3], align_corners=align_corners)
210 for align_corners in [True, False]:
213 input_tensor, out_shape[1:3], align_corners=align_corners)
226 for align_corners in [True, False]:
230 align_corners=align_corners)
242 for align_corners in [True, False]:
[all …]
/external/tensorflow/tensorflow/lite/experimental/delegates/hexagon/builders/
Dresize_bilinear_builder.cc68 int align_corners = params->align_corners ? 1 : 0; in PopulateSubGraph() local
70 quant_bound_shape, reinterpret_cast<char*>(&align_corners), in PopulateSubGraph()
71 sizeof(align_corners)); in PopulateSubGraph()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DResizeBilinear.pbtxt32 name: "align_corners"
71 name: "align_corners"
111 name: "align_corners"
151 name: "align_corners"
DResizeBilinearGrad.pbtxt27 name: "align_corners"
61 name: "align_corners"
95 name: "align_corners"
DResizeBicubic.pbtxt32 name: "align_corners"
71 name: "align_corners"
110 name: "align_corners"
DResizeNearestNeighbor.pbtxt32 name: "align_corners"
71 name: "align_corners"
110 name: "align_corners"
DResizeBicubicGrad.pbtxt26 name: "align_corners"
58 name: "align_corners"
DResizeNearestNeighborGrad.pbtxt30 name: "align_corners"
66 name: "align_corners"
DResizeArea.pbtxt32 name: "align_corners"
71 name: "align_corners"
DQuantizedResizeBilinear.pbtxt43 name: "align_corners"
92 name: "align_corners"
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dresize_test.cc43 attr.align_corners = true; in TEST()
68 attr.align_corners = false; in TEST()
92 attr.align_corners = false; in TEST()
118 attr.align_corners = false; in TEST()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dresize_bilinear_test.cc114 if (op_params.align_corners) { in TEST_P()
139 if (op_params.align_corners) { in TEST_P()
164 if (op_params.align_corners) { in TEST_P()
205 op_params.align_corners = false; in TEST()
255 op_params.align_corners = false; in TEST()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dresize_test.cc42 attr.align_corners = true; in TEST_F()
73 attr.align_corners = false; in TEST_F()
102 attr.align_corners = false; in TEST_F()
/external/tensorflow/tensorflow/lite/kernels/
Dresize_bilinear.cc80 if (params->half_pixel_centers && params->align_corners) { in Prepare()
106 op_params.align_corners = params->align_corners; \ in Eval()
/external/tensorflow/tensorflow/compiler/tests/
Dimage_ops_test.py417 align_corners=True): argument
423 image, target_shape, align_corners=align_corners)
546 align_corners=True): argument
552 image, target_shape, align_corners=align_corners)
636 align_corners=True)
707 align_corners=True): argument
713 image, target_shape, align_corners=align_corners)
730 align_corners=False)
740 align_corners=False)
758 image, [6, 4], align_corners=False)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.image.pbtxt193 …argspec: "args=[\'images\', \'size\', \'method\', \'align_corners\', \'preserve_aspect_ratio\', \'…
197 …argspec: "args=[\'images\', \'size\', \'align_corners\', \'name\'], varargs=None, keywords=None, d…
201 …argspec: "args=[\'images\', \'size\', \'align_corners\', \'name\', \'half_pixel_centers\'], vararg…
205 …argspec: "args=[\'images\', \'size\', \'align_corners\', \'name\', \'half_pixel_centers\'], vararg…
213 …argspec: "args=[\'image\', \'target_height\', \'target_width\', \'method\', \'align_corners\'], va…
217 …argspec: "args=[\'images\', \'size\', \'method\', \'align_corners\', \'preserve_aspect_ratio\', \'…
221 …argspec: "args=[\'images\', \'size\', \'align_corners\', \'name\', \'half_pixel_centers\'], vararg…
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
Dresize_test.mm61 attr.align_corners = true;
85 attr.align_corners = false;
109 attr.align_corners = false;
135 attr.align_corners = false;

1234