Home
last modified time | relevance | path

Searched refs:out_width (Results 1 – 25 of 79) sorted by relevance

1234

/external/tensorflow/tensorflow/core/kernels/
Dresize_nearest_neighbor_op_gpu.cu.cc39 const int out_width, const float height_scale, const float width_scale, in ResizeNearestNeighborNHWC() argument
45 int out_x = n % out_width; in ResizeNearestNeighborNHWC()
46 n /= out_width; in ResizeNearestNeighborNHWC()
68 const int out_width, const float height_scale, const float width_scale, in ResizeNearestNeighborBackwardNHWC() argument
79 T* bottom_diff_n = bottom_diff + n * channels * out_height * out_width; in ResizeNearestNeighborBackwardNHWC()
87 out_width - 1); in ResizeNearestNeighborBackwardNHWC()
88 const int idx = (out_y * out_width + out_x) * channels + c; in ResizeNearestNeighborBackwardNHWC()
109 const int64 out_width = output.dimension(2); in operator ()() local
111 const int output_size = batch_size * out_height * out_width * channels; in operator ()()
118 out_height, out_width, height_scale, width_scale, output.data()); in operator ()()
[all …]
Deigen_spatial_convolutions_test.cc678 const int out_width = in_cols; in TEST() local
683 Tensor<float, 4> result(kern_filters, out_depth, out_height, out_width); in TEST()
693 EXPECT_EQ(result.dimension(3), out_width); in TEST()
702 for (int k = 0; k < out_width; ++k) { in TEST()
739 const int out_width = in_cols; in TEST() local
744 Tensor<float, 4, RowMajor> result(out_width, out_height, out_depth, in TEST()
755 EXPECT_EQ(result.dimension(0), out_width); in TEST()
764 for (int k = 0; k < out_width; ++k) { in TEST()
801 const int out_width = 3; in TEST() local
806 Tensor<float, 4> result(kern_filters, out_depth, out_height, out_width); in TEST()
[all …]
Dpooling_ops_common.h65 int64 out_width; member
181 params.out_width * params.out_height * params.tensor_in_batch); in SpatialMaxPool()
207 const int32 out_width = params.out_width; in SpatialMaxPool() local
211 const int32 output_image_size = out_height * out_width * params.depth; in SpatialMaxPool()
232 const int32 w_end = std::min(wpad / col_stride + 1, out_width); in SpatialMaxPool()
237 (out_offset_batch + ph) * out_width; in SpatialMaxPool()
275 params.depth, params.out_height, params.out_width, params.window_rows,
428 params.out_width * params.out_height * params.tensor_in_batch);
454 const int32 out_width = params.out_width;
458 const int32 output_image_size = out_height * out_width * params.depth;
[all …]
Deigen_cuboid_convolution.h123 TensorIndex out_width; variable
130 out_width = Eigen::divup(inputCols - kernelCols + 1,
138 out_width = Eigen::divup(inputCols, static_cast<TensorIndex>(strideCols));
143 out_width = 0;
164 pre_contract_dims[1] = out_depth * out_height * out_width;
171 pre_contract_dims[0] = out_depth * out_height * out_width;
192 post_contract_dims[3] = out_width;
200 post_contract_dims[NumDims - 4] = out_width;
Dimage_resizer_state.h72 out_width = internal::SubtleMustCopy(Svec(1)); in ValidateAndCalculateOutputSize()
83 OP_REQUIRES(context, out_height > 0 && out_width > 0, in ValidateAndCalculateOutputSize()
92 width_scale = CalculateResizeScale(in_width, out_width, align_corners_); in ValidateAndCalculateOutputSize()
102 ceilf((out_width - 1) * width_scale) <= static_cast<float>(INT_MAX), in ValidateAndCalculateOutputSize()
114 out_width, input.dim_size(3)}), in ValidateAndCreateOutput()
120 int64 out_width; member
Dquantized_resize_bilinear_op_test.cc110 const int64 out_height, const int64 out_width, in CalcReferenceResizedVal() argument
116 GetReferenceWeight(out_width, in_width, channels, x, width_scale); in CalcReferenceResizedVal()
145 const int64 out_height, const int64 out_width, in CheckTensorValue() argument
149 const int64 out_row_size = out_width * channels; in CheckTensorValue()
153 CalculateResizeScale(in_width, out_width, align_corners); in CheckTensorValue()
157 for (int64 x = 0; x < out_width; ++x) { in CheckTensorValue()
160 in_data, batch_size, in_height, in_width, out_height, out_width, in CheckTensorValue()
290 int out_height, int out_width, int channels, in RunTestResizeBilinearTwoDims() argument
301 {out_height, out_width}, false, 1, min, max, &outputs); in RunTestResizeBilinearTwoDims()
304 in_width, out_height, out_width, channels, in RunTestResizeBilinearTwoDims()
[all …]
Dquantized_resize_bilinear_op.cc464 const int64 out_width, const int channels, in ResizeImageReference() argument
471 BuildLerpCache<float>(out_width, in_width, width_scale, channels, 0); in ResizeImageReference()
477 const int64 out_row_size = out_width * channels; in ResizeImageReference()
487 for (int64 x = 0; x < out_width; ++x) { in ResizeImageReference()
512 const int64 out_width, const int channels, in ResizeImage() argument
517 out_width, channels, height_scale, width_scale, in ResizeImage()
525 const int64 out_width, const int channels, in ResizeImage() argument
536 out_width, in_width, width_scale, channels, RESOLUTION); in ResizeImage()
542 const int64 out_row_size = out_width * channels; in ResizeImage()
559 for (; x < out_width - SIMD_STEP + 1; x += SIMD_STEP) { in ResizeImage()
[all …]
Dresize_bilinear_op.cc106 const int64 out_width, const int channels,
114 const int64 out_width, const int channels, in resize_image() argument
120 const int64 out_row_size = out_width * channels; in resize_image()
132 for (int64 x = 0; x < out_width; ++x) { in resize_image()
177 for (int64 x = 0; x < out_width; ++x) { in resize_image()
213 const int64 out_width = output.dimension(2); in operator ()() local
216 if (out_height == in_height && out_width == in_width) { in operator ()()
222 std::vector<CachedInterpolation> xs(out_width + 1); in operator ()()
227 compute_interpolation_weights(out_width, in_width, width_scale, xs.data()); in operator ()()
236 out_width, channels, xs, ys, output); in operator ()()
Dresize_nearest_neighbor_op.cc97 const int64 out_width = output.dimension(2); in operator ()() local
105 for (int x = 0; x < out_width; ++x) { in operator ()()
153 const int64 out_width = sizes(1); in Compute() local
159 0, TensorShape({batch_size, out_height, out_width, channels}), in Compute()
171 CalculateResizeScale(out_width, in_width, align_corners_); in Compute()
208 const int64 out_width = output.dimension(2); in operator ()() local
221 out_width - 1); in operator ()()
Dresize_bilinear_op_gpu.cu.cc40 int out_width, float* output) { in ResizeBilinearKernel() argument
46 const int x = idx % out_width; in ResizeBilinearKernel()
47 idx /= out_width; in ResizeBilinearKernel()
161 const int out_width = output.dimension(2); in operator ()() local
163 const int total_count = batch * out_height * out_width * channels; in operator ()()
171 out_width, output.data()); in operator ()()
Dmaxpooling_op.cc72 params.out_width * params.out_height * params.tensor_in_batch); in SpatialMaxPoolWithArgMaxHelper()
75 params.out_width * params.out_height * params.tensor_in_batch); in SpatialMaxPoolWithArgMaxHelper()
102 const int32 out_width = params.out_width; in SpatialMaxPoolWithArgMaxHelper() local
106 const int32 output_image_size = out_height * out_width * depth; in SpatialMaxPoolWithArgMaxHelper()
128 const int w_end = std::min(wpad / col_stride + 1, out_width); in SpatialMaxPoolWithArgMaxHelper()
132 const int64 out_index_base = (b * out_height + ph) * out_width; in SpatialMaxPoolWithArgMaxHelper()
168 const int out_size = out_height * out_width * depth; in SpatialMaxPoolWithArgMaxHelper()
326 params.out_height, params.out_width, params.window_rows, in MaxPoolingBackwardCustomKernel()
539 params.out_width * params.out_height * params.tensor_in_batch); in SpatialMaxPoolGradGrad()
545 params.out_width * params.out_height * params.tensor_in_batch); in SpatialMaxPoolGradGrad()
[all …]
Ddepthwise_conv_op_gpu.cu.cc88 const int out_width = args.out_cols; in DepthwiseConv2dGPUKernelNHWC() local
94 const int out_col = (thread_id / out_depth) % out_width; in DepthwiseConv2dGPUKernelNHWC()
95 const int out_row = (thread_id / out_depth / out_width) % out_height; in DepthwiseConv2dGPUKernelNHWC()
96 const int batch = thread_id / out_depth / out_width / out_height; in DepthwiseConv2dGPUKernelNHWC()
324 const int out_width = args.out_cols; in DepthwiseConv2dGPUKernelNCHW() local
335 const int out_col = thread_id % out_width; in DepthwiseConv2dGPUKernelNCHW()
336 const int out_row = (thread_id / out_width) % out_height; in DepthwiseConv2dGPUKernelNCHW()
337 const int out_channel = (thread_id / out_width / out_height) % out_depth; in DepthwiseConv2dGPUKernelNCHW()
338 const int batch = thread_id / out_width / out_height / out_depth; in DepthwiseConv2dGPUKernelNCHW()
778 const int out_width = args.out_cols; in DepthwiseConv2dBackpropInputGPUKernelNHWC() local
[all …]
Dresize_area_op_test.cc71 const int64 out_width = output_data.dimension(2); in ResizeAreaBaseline() local
74 const float width_scale = in_width / static_cast<float>(out_width); in ResizeAreaBaseline()
109 for (int64 x = 0; x < out_width; ++x) { in ResizeAreaBaseline()
Dmkl_pooling_ops_common.cc109 padding, &out_width, &pad_left, &pad_right)); in Init()
119 FastBoundsCheck(out_width, std::numeric_limits<int>::max()), in Init()
162 mkl_params->out_sizes[0] = params.out_width; in ExtractMklOpParams()
Dresize_area_op.cc157 std::vector<CachedInterpolation> x_interps(st.out_width); in Compute()
158 for (int64 x = 0; x < st.out_width; ++x) { in Compute()
244 for (int64 x = 0; x < st.out_width; ++x) { in ComputeLoop()
256 for (int64 x = 0; x < st.out_width; ++x) { in ComputeLoop()
/external/libjpeg-turbo/simd/
Djsimd.h241 (JDIMENSION out_width, JSAMPIMAGE input_buf, JDIMENSION input_row,
244 (JDIMENSION out_width, JSAMPIMAGE input_buf, JDIMENSION input_row,
247 (JDIMENSION out_width, JSAMPIMAGE input_buf, JDIMENSION input_row,
250 (JDIMENSION out_width, JSAMPIMAGE input_buf, JDIMENSION input_row,
253 (JDIMENSION out_width, JSAMPIMAGE input_buf, JDIMENSION input_row,
256 (JDIMENSION out_width, JSAMPIMAGE input_buf, JDIMENSION input_row,
259 (JDIMENSION out_width, JSAMPIMAGE input_buf, JDIMENSION input_row,
264 (JDIMENSION out_width, JSAMPIMAGE input_buf, JDIMENSION input_row,
267 (JDIMENSION out_width, JSAMPIMAGE input_buf, JDIMENSION input_row,
270 (JDIMENSION out_width, JSAMPIMAGE input_buf, JDIMENSION input_row,
[all …]
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Deigen_spatial_convolutions.h146 TensorIndex out_width; variable
151 out_width = numext::ceil((InputCols - kernelColsEff + 1.f) /
156 out_width = numext::ceil(InputCols / static_cast<float>(col_stride));
161 out_width = 0;
172 pre_contract_dims[1] = out_height * out_width;
178 pre_contract_dims[0] = out_height * out_width;
194 post_contract_dims[2] = out_width;
201 post_contract_dims[NumDims - 3] = out_width;
/external/tensorflow/tensorflow/contrib/hvx/hexagon_controller/src_impl/
Dgraph_functions_wrapper.c216 uint32_t* out_width, uint32_t* out_depth, uint8_t* out_vals, in hexagon_controller_ExecuteGraph() argument
245 *out_width = output.width; in hexagon_controller_ExecuteGraph()
252 *out_width, *out_depth, *out_data_byte_size); in hexagon_controller_ExecuteGraph()
259 uint32_t out_batches, out_height, out_width, out_depth; in hexagon_controller_ExecuteInceptionDummyData() local
268 &out_batches, &out_height, &out_width, &out_depth, in hexagon_controller_ExecuteInceptionDummyData()
273 s_output_values, out_batches * out_height * out_width * out_depth, in hexagon_controller_ExecuteInceptionDummyData()
276 out_width, out_depth, out_data_size); in hexagon_controller_ExecuteInceptionDummyData()
279 out_batches * out_height * out_width * out_depth)); in hexagon_controller_ExecuteInceptionDummyData()
/external/webp/src/dec/
Dio_dec.c300 const int out_width = io->scaled_width; in InitYUVRescaler() local
302 const int uv_out_width = (out_width + 1) >> 1; in InitYUVRescaler()
306 const size_t work_size = 2 * out_width; // scratch memory for luma rescaler in InitYUVRescaler()
332 buf->y, out_width, out_height, buf->y_stride, 1, in InitYUVRescaler()
344 buf->a, out_width, out_height, buf->a_stride, 1, in InitYUVRescaler()
484 const int out_width = io->scaled_width; in InitRGBRescaler() local
488 const size_t work_size = 2 * out_width; // scratch memory for one rescaler in InitRGBRescaler()
496 tmp_size2 = 3 * out_width; in InitRGBRescaler()
499 tmp_size2 += out_width; in InitRGBRescaler()
518 tmp + 0 * out_width, out_width, out_height, 0, 1, in InitRGBRescaler()
[all …]
/external/libxcam/cl_kernel/
Dkernel_image_warp.cl33 int out_width = get_image_width (output);
68 warp_x = (s_x * w) / (float)(PIXEL_X_STEP * out_width);
98 int out_width = get_image_width (output);
115 float warp_x = (s_x * w) / (float)out_width;
/external/deqp/external/openglcts/modules/glesext/geometry_shader/
DesextcGeometryShaderRendering.hpp144 …virtual void getRenderTargetSize(unsigned int n_instances, unsigned int* out_width, unsigned int* …
264 …void getRenderTargetSize(unsigned int n_instances, unsigned int* out_width, unsigned int* out_heig…
359 …void getRenderTargetSize(unsigned int n_instances, unsigned int* out_width, unsigned int* out_heig…
515 …void getRenderTargetSize(unsigned int n_instances, unsigned int* out_width, unsigned int* out_heig…
/external/tensorflow/tensorflow/examples/ios/simple/
Dios_image_load.mm28 int* out_width, int* out_height,
59 *out_width = 0;
83 *out_width = width;
/external/tensorflow/tensorflow/examples/ios/camera/
Dios_image_load.mm28 int* out_width, int* out_height,
59 *out_width = 0;
83 *out_width = width;
/external/tensorflow/tensorflow/contrib/lite/examples/ios/simple/
Dios_image_load.mm25 std::vector<uint8_t> LoadImageFromFile(const char* file_name, int* out_width, int* out_height,
53 *out_width = 0;
78 *out_width = width;
/external/tensorflow/tensorflow/examples/ios/benchmark/
Dios_image_load.mm28 int* out_width, int* out_height,
59 *out_width = 0;
83 *out_width = width;

1234