Home
last modified time | relevance | path

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

12345

/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/python/ops/
Dimage_grad_test_base.py98 for align_corners in [True, False]:
100 def resize_nn(t, shape=out_shape, align_corners=align_corners): argument
102 t, shape[1:3], align_corners=align_corners)
129 for align_corners, half_pixel_centers in options:
131 yield in_shape, out_shape, align_corners, half_pixel_centers
136 align_corners=False, argument
150 align_corners=align_corners,
194 for in_shape, out_shape, align_corners, half_pixel_centers in \
196 jacob_a, jacob_n = self._getJacobians(in_shape, out_shape, align_corners,
237 def _gpuVsCpuCase(self, in_shape, out_shape, align_corners, argument
[all …]
Dimage_grad_deterministic_test.py70 def testDeterministicGradients(self, align_corners, half_pixel_centers, argument
72 if not align_corners and test_util.is_xla_enabled():
78 hash(align_corners) % 256 + hash(half_pixel_centers) % 256 +
95 align_corners=align_corners,
111 align_corners=align_corners,
/external/tensorflow/tensorflow/core/kernels/image/
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> {
182 : LegacyResizeNearestNeighborNHWC<T, align_corners>; in operator ()()
203 template <typename T, bool half_pixel_centers, bool align_corners>
[all …]
Dresize_nearest_neighbor_op.cc130 template <bool half_pixel_centers, bool align_corners>
136 (align_corners) ? static_cast<Eigen::Index>(roundf(scaler(i, scale))) in compute_indices()
147 template <typename T, bool half_pixel_centers, bool align_corners>
157 compute_indices<half_pixel_centers, align_corners>( in ResizeNearestNeighborGenerator()
159 compute_indices<half_pixel_centers, align_corners>( in ResizeNearestNeighborGenerator()
183 template <typename T, bool half_pixel_centers, bool align_corners>
184 struct ResizeNearestNeighbor<CPUDevice, T, half_pixel_centers, align_corners> {
191 align_corners> in operator ()()
301 template <typename T, bool half_pixel_centers, bool align_corners>
303 align_corners> {
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dresize_nearest_neighbor.h30 const bool align_corners, in GetNearestNeighbor() argument
33 (align_corners && output_size > 1) in GetNearestNeighbor()
38 align_corners in GetNearestNeighbor()
82 op_params.align_corners, in ResizeNearestNeighbor()
87 op_params.align_corners, in ResizeNearestNeighbor()
/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()
144 attr.align_corners = false; in TEST()
170 attr.align_corners = false; in TEST()
195 attr.align_corners = true; in TEST()
221 attr.align_corners = false; in TEST()
/external/XNNPACK/test/
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
224 if (align_corners()) { in TestNHWCxF32()
240 const float offset = (tf_legacy_mode() || align_corners()) ? 0.0f : 0.5f; in TestNHWCxF32()
270 …(align_corners() ? XNN_FLAG_ALIGN_CORNERS : 0) | (tf_legacy_mode() ? XNN_FLAG_TENSORFLOW_LEGACY_MO… in TestNHWCxF32()
305 if (align_corners()) { in TestNCHWxF32()
321 const float offset = (tf_legacy_mode() || align_corners()) ? 0.0f : 0.5f; in TestNCHWxF32()
[all …]
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()
Dresize-bilinear-nchw.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()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dresize_nearest_neighbor_builder.cc55 int align_corners = params->align_corners ? 1 : 0; in PopulateSubGraph() local
57 kScalarShape, reinterpret_cast<char*>(&align_corners), in PopulateSubGraph()
58 sizeof(align_corners)); in PopulateSubGraph()
Dresize_bilinear_builder.cc55 int align_corners = params->align_corners ? 1 : 0; in PopulateSubGraph() local
57 kScalarShape, reinterpret_cast<char*>(&align_corners), in PopulateSubGraph()
58 sizeof(align_corners)); in PopulateSubGraph()
/external/tensorflow/tensorflow/core/util/
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()
/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"
199 name: "align_corners"
DResizeBicubic.pbtxt32 name: "align_corners"
71 name: "align_corners"
110 name: "align_corners"
157 name: "align_corners"
DResizeNearestNeighbor.pbtxt32 name: "align_corners"
71 name: "align_corners"
110 name: "align_corners"
157 name: "align_corners"
DResizeBilinearGrad.pbtxt27 name: "align_corners"
61 name: "align_corners"
95 name: "align_corners"
DResizeArea.pbtxt32 name: "align_corners"
71 name: "align_corners"
111 name: "align_corners"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DResizeBilinear.pbtxt32 name: "align_corners"
71 name: "align_corners"
111 name: "align_corners"
151 name: "align_corners"
199 name: "align_corners"
DResizeBicubic.pbtxt32 name: "align_corners"
71 name: "align_corners"
110 name: "align_corners"
157 name: "align_corners"
DResizeNearestNeighbor.pbtxt32 name: "align_corners"
71 name: "align_corners"
110 name: "align_corners"
157 name: "align_corners"
DResizeBilinearGrad.pbtxt27 name: "align_corners"
61 name: "align_corners"
95 name: "align_corners"
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dresize_test_util.cc36 attr.align_corners = true; in ResizeBilinearAlignedTest()
68 attr.align_corners = false; in ResizeBilinearNonAlignedTest()
100 attr.align_corners = false; in ResizeBilinearWithoutHalfPixelTest()
132 attr.align_corners = false; in ResizeBilinearWithHalfPixelTest()
162 attr.align_corners = false; in ResizeNearestTest()
194 attr.align_corners = true; in ResizeNearestAlignCornersTest()
226 attr.align_corners = false; in ResizeNearestHalfPixelCentersTest()
/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()
203 op_params.align_corners = false; in TEST()
247 op_params.align_corners = false; in TEST()
298 op_params.align_corners = false; in TestResizeBilinearHalfPixelCenters_2x2to4x6()

12345