| /external/tensorflow/tensorflow/core/kernels/ |
| D | image_resizer_state.h | 69 : align_corners_(align_corners), in ImageResizerState()
81 !half_pixel_centers_ || (half_pixel_centers_ && !align_corners_), in ValidateAndCalculateOutputSize()
116 height_scale = CalculateResizeScale(in_height, out_height, align_corners_); in ValidateAndCalculateOutputSize()
117 width_scale = CalculateResizeScale(in_width, out_width, align_corners_); in ValidateAndCalculateOutputSize()
154 bool align_corners_;
161 : align_corners_(align_corners), in ImageResizerGradientState()
168 !half_pixel_centers_ || (half_pixel_centers_ && !align_corners_), in ValidateAndCreateOutput()
201 CalculateResizeScale(original_height, resized_height, align_corners_); in ValidateAndCreateOutput()
203 CalculateResizeScale(original_width, resized_width, align_corners_); in ValidateAndCreateOutput()
223 bool align_corners_;
|
| D | resize_nearest_neighbor_op.cc | 42 OP_REQUIRES_OK(context, context->GetAttr("align_corners", &align_corners_)); in ResizeNearestNeighborOp()
49 ImageResizerState st(align_corners_, half_pixel_centers_); in Compute()
66 if (align_corners_) { in Compute()
80 if (align_corners_) { in Compute()
101 bool align_corners_; member in tensorflow::ResizeNearestNeighborOp
178 OP_REQUIRES_OK(context, context->GetAttr("align_corners", &align_corners_)); in ResizeNearestNeighborOpGrad()
225 CalculateResizeScale(out_height, in_height, align_corners_); in Compute()
227 CalculateResizeScale(out_width, in_width, align_corners_); in Compute()
231 if (align_corners_) { in Compute()
245 if (align_corners_) { in Compute()
[all …]
|
| D | resize_bilinear_op.cc | 41 OP_REQUIRES_OK(context, context->GetAttr("align_corners", &align_corners_)); in ResizeBilinearOp()
48 ImageResizerState st(align_corners_, half_pixel_centers_); in Compute()
65 bool align_corners_; member in tensorflow::ResizeBilinearOp
263 OP_REQUIRES_OK(context, context->GetAttr("align_corners", &align_corners_)); in ResizeBilinearOpGrad()
274 ImageResizerGradientState st(align_corners_, half_pixel_centers_); in Compute()
288 bool align_corners_; member in tensorflow::ResizeBilinearOpGrad
|
| D | resize_bicubic_op.cc | 553 OP_REQUIRES_OK(context, context->GetAttr("align_corners", &align_corners_)); in ResizeBicubicOp()
560 ImageResizerState st(align_corners_, half_pixel_centers_); in Compute()
573 bool align_corners_; member in tensorflow::ResizeBicubicOp
582 OP_REQUIRES_OK(context, context->GetAttr("align_corners", &align_corners_)); in ResizeBicubicOpGrad()
593 ImageResizerGradientState st(align_corners_, half_pixel_centers_); in Compute()
605 bool align_corners_; member in tensorflow::ResizeBicubicOpGrad
|
| D | resize_area_op.cc | 49 OP_REQUIRES_OK(context, context->GetAttr("align_corners", &align_corners_)); in ResizeAreaOp()
151 ImageResizerState st(align_corners_, /*unused half_pixel_centers=*/false); in Compute()
281 bool align_corners_; member in tensorflow::ResizeAreaOp
|
| D | conv_ops_fused_image_transform.cc | 613 context->GetAttr("resize_align_corners", &align_corners_)); in FusedResizeConv2DUsingGemmOp()
654 st = ImageResizerState(align_corners_, false); in Compute()
868 bool align_corners_; member in tensorflow::FusedResizeConv2DUsingGemmOp
|
| D | quantized_resize_bilinear_op.cc | 697 OP_REQUIRES_OK(context, context->GetAttr("align_corners", &align_corners_)); in QuantizedResizeBilinearOp()
707 ImageResizerState st(align_corners_, false); in Compute()
730 bool align_corners_; member in tensorflow::QuantizedResizeBilinearOp
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_resize_ops.cc | 448 void GeneralCompile(XlaOpKernelContext* ctx, bool align_corners_, in GeneralCompile() argument
524 k[0] > 1 && k[1] > 1 && align_corners_) { in GeneralCompile()
529 channels, align_corners_, is_kernel_bilinear); in GeneralCompile()
535 align_corners_, is_kernel_bilinear); in GeneralCompile()
541 align_corners_, is_kernel_bilinear); in GeneralCompile()
553 OP_REQUIRES_OK(ctx, ctx->GetAttr("align_corners", &align_corners_)); in ResizeNearestNeighborOp()
555 ctx, align_corners_ == true, in ResizeNearestNeighborOp()
561 GeneralCompile(ctx, align_corners_, is_kernel_bilinear_); in Compile()
565 bool align_corners_ = true; member in tensorflow::__anon1e51e95b0111::ResizeNearestNeighborOp
575 OP_REQUIRES_OK(ctx, ctx->GetAttr("align_corners", &align_corners_)); in ResizeBilinearOp()
[all …]
|