Searched refs:resized_width (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
97 const int x = idx % resized_width; in ResizeBilinearGradKernel()
98 idx /= resized_width; in ResizeBilinearGradKernel()
205 int channels, int resized_height, int resized_width, T* output_grad) { in LegacyResizeBilinearGradKernel() argument
211 const int x = idx % resized_width; in LegacyResizeBilinearGradKernel()
212 idx /= resized_width; in LegacyResizeBilinearGradKernel()
313 const int resized_width = input_grad.dimension(2); 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 | 189 resized_width = input.dim_size(2); in ValidateAndCreateOutput()
203 CalculateResizeScale(original_width, resized_width, align_corners_); in ValidateAndCreateOutput()
215 int64 resized_width; member
|
| D | conv_ops_fused_image_transform.cc | 120 int64 resized_width; member
229 int64 resized_width, in CalculatePerCachePixelParameters() argument
239 } else if (in_x >= resized_width) { in CalculatePerCachePixelParameters()
240 in_x = (resized_width * 2.0f) - (in_x + 1.0f + pad_offset); in CalculatePerCachePixelParameters()
260 int input_batches, int resized_height, int resized_width, in operator ()() argument
317 task_params.resized_width = resized_width; in operator ()()
487 task_params.pad_offset, task_params.resized_width, in operator ()()
|
| D | resize_bicubic_op.cc | 243 for (int64 x = 0; x < resizer_state.resized_width; ++x) { in ComputeGradientXWeightsAndIndices()
253 for (int64 x = 0; x < resizer_state.resized_width; ++x) { in ComputeGradientXWeightsAndIndices()
482 const int64 resized_width = resizer_state.resized_width; in ResizeBicubicGrad() local
487 std::vector<WeightsAndIndices> x_wais(resizer_state.resized_width); in ResizeBicubicGrad()
499 for (int64 x = 0; x < resized_width; ++x) { in ResizeBicubicGrad()
|
| D | resize_bilinear_op.cc | 308 const Eigen::Index resized_width = input_grad.dimension(2); in ResizeGradCore() local
329 for (Eigen::Index x = 0; x < resized_width; ++x) { in ResizeGradCore()
|
| D | conv_ops_test.cc | 133 const int resized_width = image_width; 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)
931 equal_(resized_width, target_width), ValueError,
1301 resized_width = math_ops.cast(
1312 resized = resize_fn(image, [resized_height, resized_width])
|