/external/tensorflow/tensorflow/core/kernels/image/ |
D | resize_bilinear_op_gpu.cu.cc | 63 const float x_lerp = in_x - left_x_index; in ResizeBilinearKernel_faster() local 102 (top_right_reg[unroll] - top_left_reg[unroll]) * x_lerp; in ResizeBilinearKernel_faster() 105 (bottom_right_reg[unroll] - bottom_left_reg[unroll]) * x_lerp; in ResizeBilinearKernel_faster() 142 const float x_lerp = in_x - left_x_index; in ResizeBilinearKernel() local 161 const float top = top_left + (top_right - top_left) * x_lerp; in ResizeBilinearKernel() 162 const float bottom = bottom_left + (bottom_right - bottom_left) * x_lerp; in ResizeBilinearKernel() 200 const float x_lerp = original_x - floorf(original_x); in ResizeBilinearGradKernel() local 208 static_cast<T>((1 - x_lerp) * dtop)); in ResizeBilinearGradKernel() 214 static_cast<T>(x_lerp * dtop)); in ResizeBilinearGradKernel() 222 static_cast<T>((1 - x_lerp) * dbottom)); in ResizeBilinearGradKernel() [all …]
|
D | crop_and_resize_op_gpu.cu.cc | 95 const float x_lerp = in_x - left_x_index; in CropAndResizeKernel() local 117 const float top = top_left + (top_right - top_left) * x_lerp; in CropAndResizeKernel() 118 const float bottom = bottom_left + (bottom_right - bottom_left) * x_lerp; in CropAndResizeKernel() 186 const float x_lerp = in_x - left_x_index; in CropAndResizeBackpropImageKernel() local 194 static_cast<T>((1 - x_lerp) * dtop)); in CropAndResizeBackpropImageKernel() 200 static_cast<T>(x_lerp * dtop)); in CropAndResizeBackpropImageKernel() 208 static_cast<T>((1 - x_lerp) * dbottom)); in CropAndResizeBackpropImageKernel() 214 static_cast<T>(x_lerp * dbottom)); in CropAndResizeBackpropImageKernel() 287 const float x_lerp = in_x - left_x_index; in CropAndResizeBackpropBoxesKernel() local 311 float image_grad_y = (1 - x_lerp) * (bottom_left - top_left) + in CropAndResizeBackpropBoxesKernel() [all …]
|
D | resize_bilinear_op.cc | 110 const float x_lerp, const float y_lerp) { in compute_lerp() argument 111 const float top = top_left + (top_right - top_left) * x_lerp; in compute_lerp() 112 const float bottom = bottom_left + (bottom_right - bottom_left) * x_lerp; in compute_lerp() 311 const float x_lerp = in_x - floorf(in_x); in ResizeGradCore() local 312 const float inverse_x_lerp = (1.0f - x_lerp); in ResizeGradCore() 318 T(input_grad(b, y, x, c) * inverse_y_lerp * x_lerp); in ResizeGradCore() 322 T(input_grad(b, y, x, c) * y_lerp * x_lerp); in ResizeGradCore()
|
D | crop_and_resize_op.cc | 286 const float x_lerp = in_x - left_x_index; in operator ()() local 297 const float top = top_left + (top_right - top_left) * x_lerp; in operator ()() 299 bottom_left + (bottom_right - bottom_left) * x_lerp; in operator ()() 505 const float x_lerp = in_x - left_x_index; in operator ()() local 510 static_cast<T>((1 - x_lerp) * dtop); in operator ()() 512 static_cast<T>(x_lerp * dtop); in operator ()() 515 static_cast<T>((1 - x_lerp) * dbottom); in operator ()() 517 static_cast<T>(x_lerp * dbottom); in operator ()() 706 const float x_lerp = in_x - left_x_index; in operator ()() local 718 float image_grad_y = (1 - x_lerp) * (bottom_left - top_left) + in operator ()() [all …]
|
D | resize_bilinear_op_test.cc | 117 const float x_lerp = in_x - std::floor(in_x); in ResizeBilinearBaseline() local 125 const float top = top_left + (top_right - top_left) * x_lerp; in ResizeBilinearBaseline() 127 bottom_left + (bottom_right - bottom_left) * x_lerp; in ResizeBilinearBaseline()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | quantized_resize_bilinear_op_test.cc | 83 const float x_lerp, const float y_lerp, const float min, in ComputeLerpReference() argument 89 const float top = top_left + (top_right - top_left) * x_lerp; in ComputeLerpReference() 90 const float bottom = bottom_left + (bottom_right - bottom_left) * x_lerp; in ComputeLerpReference() 100 const float x_lerp, const float y_lerp, in ComputeLerpReference() argument 102 const float top = in_top_left + (in_top_right - in_top_left) * x_lerp; in ComputeLerpReference() 104 in_bottom_left + (in_bottom_right - in_bottom_left) * x_lerp; in ComputeLerpReference()
|
D | conv_ops_fused_image_transform.cc | 152 EIGEN_ALWAYS_INLINE T1 BilinearSample(int channel, T1 x_lerp, in BilinearSample() 155 top_left[channel] + (top_right[channel] - top_left[channel]) * x_lerp; in BilinearSample() 157 (bottom_right[channel] - bottom_left[channel]) * x_lerp; in BilinearSample() 216 x_lerp(other.x_lerp) {} in PerCachePixelParameters() 221 T1 x_lerp; member 250 result.x_lerp = static_cast<T1>(in_x - result.left_x_index); in CalculatePerCachePixelParameters() 532 pixel_params.x_lerp, in operator ()()
|
D | quantization_utils_test.cc | 827 T_CALC bottom_right, T_CALC x_lerp, T_CALC y_lerp) { in ComputeRefLerp() argument 830 top_left * RESOLUTION_POW + (top_right - top_left) * x_lerp; in ComputeRefLerp() 832 bottom_left * RESOLUTION_POW + (bottom_right - bottom_left) * x_lerp; in ComputeRefLerp() 839 uint8 bottom_right, int16 x_lerp, int16 y_lerp) { in TestComputeLerp8x8() argument 844 int16x8_t x_lerp16x8 = To16x8(x_lerp); in TestComputeLerp8x8() 855 static_cast<int16>(bottom_left), static_cast<int16>(bottom_right), x_lerp, in TestComputeLerp8x8() 865 << static_cast<int>(bottom_right) << ", " << x_lerp << ", " << y_lerp in TestComputeLerp8x8() 871 int32 bottom_right, int32 x_lerp, int32 y_lerp) { in TestComputeLerp32x2() argument 876 int32x2_t x_lerp32x2 = To32x2(x_lerp); in TestComputeLerp32x2() 886 static_cast<int64>(x_lerp), static_cast<int64>(y_lerp)); in TestComputeLerp32x2() [all …]
|
D | quantized_resize_bilinear_op.cc | 103 const float x_lerp, const float y_lerp, in ComputeLerpReference() argument 109 const float top = top_left + (top_right - top_left) * x_lerp; in ComputeLerpReference() 110 const float bottom = bottom_left + (bottom_right - bottom_left) * x_lerp; in ComputeLerpReference() 123 const T bottom_right, const T_SCALE x_lerp, in ComputeLerp() argument 127 MulOffset<T, T_SCALE, T_CALC>(top_right, top_left, x_lerp); in ComputeLerp() 130 MulOffset<T, T_SCALE, T_CALC>(bottom_right, bottom_left, x_lerp); in ComputeLerp() 182 const int32* x_lerp, const int32x2_t y_lerpsx) { in ComputeLerpx2() argument 184 X_LERP_SAME ? vld1_dup_s32(reinterpret_cast<const int32*>(x_lerp)) in ComputeLerpx2() 185 : vld1_s32(reinterpret_cast<const int32*>(x_lerp)); in ComputeLerpx2()
|
D | quantization_utils.h | 483 const int32x2_t x_lerp, in ComputeLerp32x2() argument 495 vmlal_s32(top_left_x_res, top_right_sub_top_left, x_lerp); in ComputeLerp32x2() 499 vmlal_s32(bottom_left_x_res, bottom_right_sub_bottom_left, x_lerp); in ComputeLerp32x2() 519 const int16x8_t x_lerp, in ComputeLerp8x8() argument 533 vmlaq_s16(top_left_x_res, top_right_sub_top_left, x_lerp); in ComputeLerp8x8() 538 vmlaq_s16(bottom_left_x_res, bottom_right_sub_bottom_left, x_lerp); in ComputeLerp8x8()
|