Searched refs:resized_height (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | resize_bilinear_op_gpu.cu.cc | 91 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 ()()
|
D | image_resizer_state.h | 188 resized_height = input.dim_size(1); in ValidateAndCreateOutput() 201 CalculateResizeScale(original_height, resized_height, align_corners_); in ValidateAndCreateOutput() 214 int64 resized_height; member
|
D | conv_ops_fused_image_transform.cc | 114 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 ()()
|
D | resize_bicubic_op.cc | 483 const int64 resized_height = resizer_state.resized_height; in ResizeBicubicGrad() local 490 for (int64 y = 0; y < resized_height; ++y) { in ResizeBicubicGrad()
|
D | resize_bilinear_op.cc | 307 const Eigen::Index resized_height = input_grad.dimension(1); in ResizeGradCore() local 320 for (Eigen::Index y = 0; y < resized_height; ++y) { in ResizeGradCore()
|
D | conv_ops_test.cc | 134 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/ |
D | image_ops_impl.py | 924 _, 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])
|