Home
last modified time | relevance | path

Searched refs:align_corners_ (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dimage_resizer_state.h71 : align_corners_(align_corners), in ImageResizerState()
83 !half_pixel_centers_ || (half_pixel_centers_ && !align_corners_), in ValidateAndCalculateOutputSize()
118 height_scale = CalculateResizeScale(in_height, out_height, align_corners_); in ValidateAndCalculateOutputSize()
119 width_scale = CalculateResizeScale(in_width, out_width, align_corners_); in ValidateAndCalculateOutputSize()
156 bool align_corners_;
163 : align_corners_(align_corners), in ImageResizerGradientState()
170 !half_pixel_centers_ || (half_pixel_centers_ && !align_corners_), in ValidateAndCreateOutput()
217 CalculateResizeScale(original_height, resized_height, align_corners_); in ValidateAndCreateOutput()
219 CalculateResizeScale(original_width, resized_width, align_corners_); in ValidateAndCreateOutput()
239 bool align_corners_;
/external/tensorflow/tensorflow/core/kernels/image/
Dresize_nearest_neighbor_op.cc43 OP_REQUIRES_OK(context, context->GetAttr("align_corners", &align_corners_)); in ResizeNearestNeighborOp()
50 ImageResizerState st(align_corners_, half_pixel_centers_); in Compute()
67 if (align_corners_) { in Compute()
81 if (align_corners_) { in Compute()
102 bool align_corners_; member in tensorflow::ResizeNearestNeighborOp
205 OP_REQUIRES_OK(context, context->GetAttr("align_corners", &align_corners_)); in ResizeNearestNeighborOpGrad()
252 CalculateResizeScale(out_height, in_height, align_corners_); in Compute()
254 CalculateResizeScale(out_width, in_width, align_corners_); in Compute()
258 if (align_corners_) { in Compute()
272 if (align_corners_) { in Compute()
[all …]
Dresize_nearest_neighbor_op_test.cc41 : align_corners_(false), half_pixel_centers_(half_pixel_centers) {} in ResizeNearestNeighborOpTestBase()
53 .Attr("align_corners", align_corners_) in SetUp()
58 bool align_corners_; member in tensorflow::ResizeNearestNeighborOpTestBase
80 ResizeNearestNeighborOpAlignCornersTest() : align_corners_(true) {} in ResizeNearestNeighborOpAlignCornersTest()
92 .Attr("align_corners", align_corners_) in SetUp()
96 bool align_corners_; member in tensorflow::ResizeNearestNeighborOpAlignCornersTest
Dresize_bilinear_op.cc47 OP_REQUIRES_OK(context, context->GetAttr("align_corners", &align_corners_)); in ResizeBilinearOp()
54 ImageResizerState st(align_corners_, half_pixel_centers_); in Compute()
71 bool align_corners_; member in tensorflow::ResizeBilinearOp
216 OP_REQUIRES_OK(context, context->GetAttr("align_corners", &align_corners_)); in ResizeBilinearOpGrad()
227 ImageResizerGradientState st(align_corners_, half_pixel_centers_); in Compute()
260 bool align_corners_; member in tensorflow::ResizeBilinearOpGrad
Dresize_bilinear_op_test.cc42 : align_corners_(false), half_pixel_centers_(false) {} in ResizeBilinearOpTestBase()
54 .Attr("align_corners", align_corners_) in SetUp()
165 bool align_corners_; member in tensorflow::ResizeBilinearOpTestBase
181 ResizeBilinearOpAlignCornersTest() { align_corners_ = true; } in ResizeBilinearOpAlignCornersTest()
556 .Attr("align_corners", align_corners_) in SetUpBenchmark()
Dresize_area_op.cc50 OP_REQUIRES_OK(context, context->GetAttr("align_corners", &align_corners_)); in ResizeAreaOp()
152 ImageResizerState st(align_corners_, /*unused half_pixel_centers=*/false); in Compute()
282 bool align_corners_; member in tensorflow::ResizeAreaOp
Dresize_bicubic_op.cc554 OP_REQUIRES_OK(context, context->GetAttr("align_corners", &align_corners_)); in ResizeBicubicOp()
561 ImageResizerState st(align_corners_, half_pixel_centers_); in Compute()
574 bool align_corners_; member in tensorflow::ResizeBicubicOp
583 OP_REQUIRES_OK(context, context->GetAttr("align_corners", &align_corners_)); in ResizeBicubicOpGrad()
594 ImageResizerGradientState st(align_corners_, half_pixel_centers_); in Compute()
606 bool align_corners_; member in tensorflow::ResizeBicubicOpGrad
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dimage_resize_ops.h29 bool align_corners_ = true;
41 bool align_corners_ = true;
53 bool align_corners_;
Dimage_resize_ops.cc471 void GeneralCompile(XlaOpKernelContext* ctx, bool align_corners_, in GeneralCompile() argument
538 align_corners_, is_kernel_bilinear); in GeneralCompile()
546 auto pairwise_distance = [align_corners_](int64 src_size, int64 dst_size) { in GeneralCompile()
548 align_corners_); in GeneralCompile()
571 align_corners_, is_kernel_bilinear); in GeneralCompile()
586 OP_REQUIRES_OK(ctx, ctx->GetAttr("align_corners", &align_corners_)); in ResizeNearestNeighborOp()
588 ctx, align_corners_ == true, in ResizeNearestNeighborOp()
599 GeneralCompile(ctx, align_corners_, is_kernel_bilinear_); in Compile()
607 OP_REQUIRES_OK(ctx, ctx->GetAttr("align_corners", &align_corners_)); in ResizeBilinearOp()
616 GeneralCompile(ctx, align_corners_, is_kernel_bilinear_); in Compile()
[all …]
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dresize_bilinear_tester.h84 align_corners_ = align_corners; in AlignCorners()
88 bool AlignCorners() const { return align_corners_; } in AlignCorners()
108 bool align_corners_ = false; variable
/external/tensorflow/tensorflow/core/tpu/kernels/
Dimage_resize_ops.cc34 OP_REQUIRES_OK(ctx, ctx->GetAttr("align_corners", &align_corners_)); in TpuCustomResizeOp()
51 return absl::StrCat("\"", align_corners_, half_pixel_centers_, "\""); in OpaqueField()
95 bool align_corners_; member in tensorflow::TpuCustomResizeOp
/external/XNNPACK/test/
Dresize-bilinear-operator-tester.h197 this->align_corners_ = align_corners; in align_corners()
202 return this->align_corners_; in align_corners()
558 bool align_corners_{false};
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_fused_image_transform.cc614 context->GetAttr("resize_align_corners", &align_corners_)); in FusedResizeConv2DUsingGemmOp()
655 st = ImageResizerState(align_corners_, false); in Compute()
865 bool align_corners_; member in tensorflow::FusedResizeConv2DUsingGemmOp
Dquantized_resize_bilinear_op.cc697 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