Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/
Dresize_nearest_neighbor_op_gpu.cu.cc35 template <typename T, bool align_corners>
52 min((align_corners) ? static_cast<int>(roundf(out_y * height_scale)) in ResizeNearestNeighborNHWC()
56 min((align_corners) ? static_cast<int>(roundf(out_x * width_scale)) in ResizeNearestNeighborNHWC()
64 template <typename T, bool align_corners>
81 min((align_corners) ? static_cast<int>(roundf(in_y * height_scale)) in ResizeNearestNeighborBackwardNHWC()
85 min((align_corners) ? static_cast<int>(roundf(in_x * width_scale)) in ResizeNearestNeighborBackwardNHWC()
98 template <typename T, bool align_corners>
99 struct ResizeNearestNeighbor<GPUDevice, T, align_corners> {
115 ResizeNearestNeighborNHWC<T, align_corners> in operator ()()
132 template <typename T, bool align_corners>
[all …]
Dresize_nearest_neighbor_op.cc86 template <typename T, bool align_corners>
87 struct ResizeNearestNeighbor<CPUDevice, T, align_corners> {
102 (align_corners) ? static_cast<int64>(roundf(y * height_scale)) in operator ()()
107 (align_corners) ? static_cast<int64>(roundf(x * width_scale)) in operator ()()
197 template <typename T, bool align_corners>
198 struct ResizeNearestNeighborGrad<CPUDevice, T, align_corners> {
214 (align_corners) ? static_cast<int64>(roundf(y * height_scale)) in operator ()()
219 (align_corners) ? static_cast<int64>(roundf(x * width_scale)) in operator ()()
Dimage_resizer_state.h42 bool align_corners) { in CalculateResizeScale() argument
43 return (align_corners && out_size > 1) in CalculateResizeScale()
49 explicit ImageResizerState(bool align_corners) in ImageResizerState()
50 : align_corners_(align_corners) {} in ImageResizerState()
133 explicit ImageResizerGradientState(bool align_corners) in ImageResizerGradientState()
134 : align_corners_(align_corners) {} in ImageResizerGradientState()
Dresize_nearest_neighbor_op.h25 template <typename Device, typename T, bool align_corners>
32 template <typename Device, typename T, 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()
146 const int channels, const bool align_corners, in CheckTensorValue() argument
151 CalculateResizeScale(in_height, out_height, align_corners); in CheckTensorValue()
153 CalculateResizeScale(in_width, out_width, align_corners); in CheckTensorValue()
/external/tensorflow/tensorflow/python/ops/
Dimage_grad_test.py86 for align_corners in [True, False]:
90 input_tensor, out_shape[1:3], align_corners=align_corners)
97 input_tensor, out_shape[1:3], align_corners=align_corners)
151 for align_corners in [True, False]:
157 input_tensor, out_shape[1:3], align_corners=align_corners)
192 for align_corners in [True, False]:
196 align_corners=align_corners)
208 for align_corners in [True, False]:
212 align_corners=align_corners)
223 for align_corners in [True, False]:
[all …]
Dimage_grad.py48 align_corners=op.get_attr("align_corners"))
66 grad, op.inputs[0], align_corners=op.get_attr("align_corners"))
87 grad, op.inputs[0], align_corners=op.get_attr("align_corners"))
Dimage_ops_impl.py840 align_corners=False): argument
921 images, size, align_corners=align_corners)
924 images, size, align_corners=align_corners)
927 images, size, align_corners=align_corners)
930 images, size, align_corners=align_corners)
Dimage_ops_test.py655 img, [299, 299], align_corners=False)
703 img, [299, 299], align_corners=False)
760 resize_op = image_ops.resize_area(img, [299, 299], align_corners=False)
2278 image, [target_height, target_width], opt, align_corners=False)
2314 image, [target_height, target_width], opt, align_corners=True)
2376 for align_corners in [True, False]:
2386 align_corners=align_corners)
2395 align_corners=align_corners)
2405 for align_corners in [True, False]:
2417 align_corners=align_corners)
/external/tensorflow/tensorflow/tools/api/golden/
Dtensorflow.image.pbtxt145 …argspec: "args=[\'images\', \'size\', \'align_corners\', \'name\'], varargs=None, keywords=None, d…
149 …argspec: "args=[\'images\', \'size\', \'align_corners\', \'name\'], varargs=None, keywords=None, d…
153 …argspec: "args=[\'images\', \'size\', \'align_corners\', \'name\'], varargs=None, keywords=None, d…
161 …argspec: "args=[\'images\', \'size\', \'method\', \'align_corners\'], varargs=None, keywords=None,…
165 …argspec: "args=[\'images\', \'size\', \'align_corners\', \'name\'], varargs=None, keywords=None, d…
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_ResizeNearestNeighbor.pbtxt24 name: "align_corners"
Dapi_def_ResizeNearestNeighborGrad.pbtxt25 name: "align_corners"
Dapi_def_QuantizedResizeBilinear.pbtxt24 name: "align_corners"
Dapi_def_ResizeBicubicGrad.pbtxt26 name: "align_corners"
Dapi_def_ResizeBilinearGrad.pbtxt26 name: "align_corners"
Dapi_def_ResizeBicubic.pbtxt24 name: "align_corners"
Dapi_def_ResizeBilinear.pbtxt24 name: "align_corners"
Dapi_def_ResizeArea.pbtxt24 name: "align_corners"
Dapi_def_CropAndResize.pbtxt72 with `align_corners=True`.
/external/tensorflow/tensorflow/contrib/lite/examples/label_image/
Dbitmap_helpers_impl.h69 params->align_corners = false; in resize()
/external/tensorflow/tensorflow/contrib/lite/
Dbuiltin_op_data.h154 bool align_corners; member
Dmodel.cc474 params->align_corners = schema_params->align_corners(); in ParseOpData()
/external/tensorflow/tensorflow/contrib/lite/toco/tflite/
Doperator_test.cc335 op.align_corners = true; in TEST_F()
338 EXPECT_EQ(op.align_corners, output_toco_op->align_corners); in TEST_F()
/external/tensorflow/tensorflow/contrib/lite/kernels/
Dresize_bilinear.cc95 params->align_corners) in Eval()
/external/tensorflow/tensorflow/python/tools/
Doptimize_for_inference_test.py233 input_op, [12, 4], align_corners=False)
264 input_op, [12, 4], align_corners=False)

12