Home
last modified time | relevance | path

Searched refs:resized_height (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dresize_bilinear_op_gpu.cu.cc91 int channels, int resized_height, int resized_width, T* output_grad) { in ResizeBilinearGradKernel() argument
99 const int y = idx % resized_height; in ResizeBilinearGradKernel()
100 const int b = idx / resized_height; in ResizeBilinearGradKernel()
205 int channels, int resized_height, int resized_width, T* output_grad) { in LegacyResizeBilinearGradKernel() argument
213 const int y = idx % resized_height; in LegacyResizeBilinearGradKernel()
214 const int b = idx / resized_height; in LegacyResizeBilinearGradKernel()
312 const int resized_height = input_grad.dimension(1); in operator ()() local
327 total_count = batch * resized_height * resized_width * channels; in operator ()()
334 original_width, channels, resized_height, resized_width, in operator ()()
341 original_width, channels, resized_height, resized_width, in operator ()()
Dimage_resizer_state.h188 resized_height = input.dim_size(1); in ValidateAndCreateOutput()
201 CalculateResizeScale(original_height, resized_height, align_corners_); in ValidateAndCreateOutput()
214 int64 resized_height; member
Dconv_ops_fused_image_transform.cc114 int64 resized_height; member
171 int64 resized_height, const ImageResizerState& st, in CalculatePerCacheLineParameters() argument
190 } else if (in_y >= resized_height) { in CalculatePerCacheLineParameters()
191 in_y = (resized_height * 2.0f) - (in_y + 1.0f + pad_offset); in CalculatePerCacheLineParameters()
260 int input_batches, int resized_height, int resized_width, in operator ()() argument
314 task_params.resized_height = resized_height; in operator ()()
475 task_params.pad_offset, task_params.resized_height, in operator ()()
Dresize_bicubic_op.cc483 const int64 resized_height = resizer_state.resized_height; in ResizeBicubicGrad() local
490 for (int64 y = 0; y < resized_height; ++y) { in ResizeBicubicGrad()
Dresize_bilinear_op.cc307 const Eigen::Index resized_height = input_grad.dimension(1); in ResizeGradCore() local
320 for (Eigen::Index y = 0; y < resized_height; ++y) { in ResizeGradCore()
Dconv_ops_test.cc134 const int resized_height = image_height; in HandwrittenConv() local
142 AddInputFromArray<int32>(TensorShape({2}), {resized_height, resized_width}); in HandwrittenConv()
/external/tensorflow/tensorflow/python/ops/
Dimage_ops_impl.py924 _, resized_height, resized_width, _ = _ImageDimensions(resized, rank=4)
928 equal_(resized_height, target_height), ValueError,
1299 resized_height = math_ops.cast(
1312 resized = resize_fn(image, [resized_height, resized_width])