Home
last modified time | relevance | path

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

123

/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.cc66 template <typename T, bool align_corners>
83 min((align_corners) ? static_cast<int>(roundf(out_y * height_scale)) in LegacyResizeNearestNeighborNHWC()
87 min((align_corners) ? static_cast<int>(roundf(out_x * width_scale)) in LegacyResizeNearestNeighborNHWC()
126 template <typename T, bool align_corners>
143 min((align_corners) ? static_cast<int>(roundf(in_y * height_scale)) in LegacyResizeNearestNeighborBackwardNHWC()
147 min((align_corners) ? static_cast<int>(roundf(in_x * width_scale)) in LegacyResizeNearestNeighborBackwardNHWC()
160 template <typename T, bool half_pixel_centers, bool align_corners>
161 struct ResizeNearestNeighbor<GPUDevice, T, half_pixel_centers, align_corners> {
186 LegacyResizeNearestNeighborNHWC<T, align_corners>, config.block_count, in operator ()()
206 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()
68 explicit ImageResizerState(bool align_corners, bool half_pixel_centers) in ImageResizerState()
69 : align_corners_(align_corners), in ImageResizerState()
159 explicit ImageResizerGradientState(bool align_corners, in ImageResizerGradientState()
161 : align_corners_(align_corners), in ImageResizerGradientState()
Dresize_nearest_neighbor_op.cc131 template <typename T, bool half_pixel_centers, bool align_corners>
132 struct ResizeNearestNeighbor<CPUDevice, T, half_pixel_centers, align_corners> {
148 (align_corners) in operator ()()
157 (align_corners) in operator ()()
274 template <typename T, bool half_pixel_centers, bool align_corners>
276 align_corners> {
293 (align_corners) in operator ()()
299 (align_corners) in operator ()()
Dresize_nearest_neighbor_op.h26 bool align_corners>
34 bool align_corners>
Dquantized_resize_bilinear_op_test.cc62 float CalculateResizeScale(int64 in_size, int64 out_size, bool align_corners) { in CalculateResizeScale() argument
63 return (align_corners && out_size > 1) in CalculateResizeScale()
147 const int channels, const bool align_corners, in CheckTensorValue() argument
153 CalculateResizeScale(in_height, out_height, align_corners); in CheckTensorValue()
155 CalculateResizeScale(in_width, out_width, align_corners); in CheckTensorValue()
/external/tensorflow/tensorflow/python/ops/
Dimage_grad_test.py92 for align_corners in [True, False]:
96 input_tensor, out_shape[1:3], align_corners=align_corners)
103 input_tensor, out_shape[1:3], align_corners=align_corners)
165 for align_corners in align_corner_options:
171 input_tensor, out_shape[1:3], align_corners=align_corners)
207 for align_corners in [True, False]:
211 align_corners=align_corners)
224 for align_corners in [True, False]:
228 align_corners=align_corners)
240 for align_corners in [True, False]:
[all …]
Dimage_grad.py47 align_corners=op.get_attr("align_corners"),
66 align_corners=op.get_attr("align_corners"),
110 align_corners=op.get_attr("align_corners"),
Dimage_ops_impl.py1043 align_corners=False, argument
1100 images_t, new_size, align_corners=align_corners)
1104 images_t, new_size, align_corners=align_corners)
1107 images_t, new_size, align_corners=align_corners)
1110 images_t, new_size, align_corners=align_corners)
1333 align_corners=False): argument
1364 return resize_images(im, new_size, method, align_corners=align_corners)
3195 align_corners=False, argument
3201 align_corners=align_corners,
3208 align_corners=False, argument
[all …]
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.image.pbtxt181 …argspec: "args=[\'images\', \'size\', \'method\', \'align_corners\', \'preserve_aspect_ratio\', \'…
185 …argspec: "args=[\'images\', \'size\', \'align_corners\', \'name\'], varargs=None, keywords=None, d…
189 …argspec: "args=[\'images\', \'size\', \'align_corners\', \'name\', \'half_pixel_centers\'], vararg…
193 …argspec: "args=[\'images\', \'size\', \'align_corners\', \'name\', \'half_pixel_centers\'], vararg…
201 …argspec: "args=[\'image\', \'target_height\', \'target_width\', \'method\', \'align_corners\'], va…
205 …argspec: "args=[\'images\', \'size\', \'method\', \'align_corners\', \'preserve_aspect_ratio\', \'…
209 …argspec: "args=[\'images\', \'size\', \'align_corners\', \'name\', \'half_pixel_centers\'], vararg…
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dimage_resize_ops.cc83 bool align_corners) { in ComputeResizeConvolutionParameters() argument
100 const int64 in_size_factor = align_corners ? in_size[i] - 1 : in_size[i]; in ComputeResizeConvolutionParameters()
102 align_corners ? out_size[i] - 1 : out_size[i]; in ComputeResizeConvolutionParameters()
230 const bool align_corners, bool is_kernel_bilinear) { in ResizeUsingDilationAndConvolution() argument
255 ComputeResizeConvolutionParameters(in_size, out_size, align_corners); in ResizeUsingDilationAndConvolution()
268 if (!align_corners) { in ResizeUsingDilationAndConvolution()
358 const bool align_corners, bool is_kernel_bilinear) { in ResizeUsingDilationAndConvolutionGradOp() argument
360 ComputeResizeConvolutionParameters(in_size, grad_size, align_corners); in ResizeUsingDilationAndConvolutionGradOp()
/external/tensorflow/tensorflow/compiler/tests/
Dimage_ops_test.py416 align_corners=True): argument
422 image, target_shape, align_corners=align_corners)
524 align_corners=True): argument
530 image, target_shape, align_corners=align_corners)
553 align_corners=True)
700 align_corners=False)
710 align_corners=False)
728 image, [6, 4], align_corners=False)
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h232 bool align_corners; member
236 bool align_corners; member
/external/tensorflow/tensorflow/lite/kernels/
Dresize_bilinear.cc93 op_params.align_corners = params->align_corners; \ in Eval()
Dresize_nearest_neighbor.cc89 op_params.align_corners = params->align_corners; in Eval()
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc436 params->align_corners = schema_params->align_corners(); in ParseOpData()
450 params->align_corners = schema_params->align_corners(); in ParseOpData()
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator_test.cc418 op.align_corners = true; in TEST_F()
421 EXPECT_EQ(op.align_corners, output_toco_op->align_corners); in TEST_F()
426 op.align_corners = true; in TEST_F()
431 EXPECT_EQ(op.align_corners, output_toco_op->align_corners); in TEST_F()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_ResizeNearestNeighborGrad.pbtxt25 name: "align_corners"
Dapi_def_ResizeNearestNeighbor.pbtxt24 name: "align_corners"
Dapi_def_ResizeBilinearGrad.pbtxt26 name: "align_corners"
Dapi_def_ResizeBicubicGrad.pbtxt26 name: "align_corners"
Dapi_def_QuantizedResizeBilinear.pbtxt24 name: "align_corners"
Dapi_def_ResizeBilinear.pbtxt24 name: "align_corners"
Dapi_def_ResizeBicubic.pbtxt24 name: "align_corners"

123