Home
last modified time | relevance | path

Searched refs:top_y_index (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/image/
Dresize_bilinear_op_gpu.cu.cc54 const int top_y_index = in_y > 0.0 ? floorf(in_y) : 0; in ResizeBilinearKernel_faster() local
75 ((float4*)images)[(((b * in_height + top_y_index) * in_width + in ResizeBilinearKernel_faster()
81 ((float4*)images)[(((b * in_height + top_y_index) * in_width + in ResizeBilinearKernel_faster()
133 const int top_y_index = in_y > 0.0 ? floorf(in_y) : 0; in ResizeBilinearKernel() local
145 images[((b * in_height + top_y_index) * in_width + left_x_index) * in ResizeBilinearKernel()
149 images[((b * in_height + top_y_index) * in_width + right_x_index) * in ResizeBilinearKernel()
187 const int top_y_index = original_y > 0.0 ? floorf(original_y) : 0; in ResizeBilinearGradKernel() local
204 ((b * original_height + top_y_index) * original_width + in ResizeBilinearGradKernel()
210 ((b * original_height + top_y_index) * original_width + in ResizeBilinearGradKernel()
301 const int top_y_index = floorf(in_y); in LegacyResizeBilinearKernel() local
[all …]
Dcrop_and_resize_op_gpu.cu.cc89 const int top_y_index = floorf(in_y); in CropAndResizeKernel() local
91 const float y_lerp = in_y - top_y_index; in CropAndResizeKernel()
98 image_ptr[((b_in * image_height + top_y_index) * image_width + in CropAndResizeKernel()
103 image_ptr[((b_in * image_height + top_y_index) * image_width + in CropAndResizeKernel()
180 const int top_y_index = floorf(in_y); in CropAndResizeBackpropImageKernel() local
182 const float y_lerp = in_y - top_y_index; in CropAndResizeBackpropImageKernel()
190 ((b_in * image_height + top_y_index) * image_width + in CropAndResizeBackpropImageKernel()
196 ((b_in * image_height + top_y_index) * image_width + in CropAndResizeBackpropImageKernel()
281 const int top_y_index = floorf(in_y); in CropAndResizeBackpropBoxesKernel() local
283 const float y_lerp = in_y - top_y_index; in CropAndResizeBackpropBoxesKernel()
[all …]
Dcrop_and_resize_op.cc270 const int top_y_index = floorf(in_y); in operator ()() local
272 const float y_lerp = in_y - top_y_index; in operator ()()
290 image(b_in, top_y_index, left_x_index, d))); in operator ()()
292 image(b_in, top_y_index, right_x_index, d))); in operator ()()
490 const int top_y_index = floorf(in_y); in operator ()() local
492 const float y_lerp = in_y - top_y_index; in operator ()()
509 grads_image(b_in, top_y_index, left_x_index, d) += in operator ()()
511 grads_image(b_in, top_y_index, right_x_index, d) += in operator ()()
693 const int top_y_index = floorf(in_y); in operator ()() local
695 const float y_lerp = in_y - top_y_index; in operator ()()
[all …]
Dresize_bilinear_op.cc297 const Eigen::Index top_y_index = in ResizeGradCore() local
315 output_grad(b, top_y_index, left_x_index, c) += in ResizeGradCore()
317 output_grad(b, top_y_index, right_x_index, c) += in ResizeGradCore()
Dresize_bilinear_op_test.cc103 const int64 top_y_index = in ResizeBilinearBaseline() local
119 const float top_left = images(b, top_y_index, left_x_index, c); in ResizeBilinearBaseline()
120 const float top_right = images(b, top_y_index, right_x_index, c); in ResizeBilinearBaseline()
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_fused_image_transform.cc196 const int64 top_y_index = static_cast<int64>(std::floor(in_y)); in CalculatePerCacheLineParameters() local
200 result.y_lerp = static_cast<T1>(in_y - top_y_index); in CalculatePerCacheLineParameters()
203 input_batch_start + (top_y_index * input_width * input_depth); in CalculatePerCacheLineParameters()