Home
last modified time | relevance | path

Searched refs:output_cols (Results 1 – 25 of 37) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Deigen_pooling_test.cc35 const int output_cols = 2; in TEST() local
38 Tensor<float, 4> result(depth, output_rows, output_cols, num_batches); in TEST()
50 EXPECT_EQ(result.dimension(2), output_cols); in TEST()
56 for (int j = 0; j < output_cols; ++j) { in TEST()
83 const int output_cols = 2; in TEST() local
86 Tensor<float, 4, RowMajor> result(num_batches, output_cols, output_rows, in TEST()
99 EXPECT_EQ(result.dimension(1), output_cols); in TEST()
105 for (int j = 0; j < output_cols; ++j) { in TEST()
134 const int output_cols = 3; in TEST() local
139 Tensor<float, 5> result(channels, output_planes, output_rows, output_cols, in TEST()
[all …]
Deigen_backward_spatial_convolutions_test.cc38 const int output_cols = input_cols - patch_cols + 1; in TEST() local
42 Tensor<float, 3> output_backward(output_depth, output_rows, output_cols); in TEST()
65 output_j < output_cols) { in TEST()
87 const int output_cols = input_cols - patch_cols + 1; in TEST() local
93 Tensor<float, 3, RowMajor> output_backward(output_cols, output_rows, in TEST()
117 output_j < output_cols) { in TEST()
140 const int output_cols = input_cols - patch_cols + 1; in TEST() local
149 output_cols); in TEST()
177 output_k >= 0 && output_k < output_cols) { in TEST()
203 const int output_cols = input_cols - patch_cols + 1; in TEST() local
[all …]
Ddilation_ops_gpu.cu.cc43 int output_cols, int stride_rows, in DilationKernel() argument
50 const int w_out = out_idx2 % output_cols; in DilationKernel()
51 const int out_idx3 = out_idx2 / output_cols; in DilationKernel()
83 int output_cols, int stride_rows, int stride_cols, int rate_rows, in DilationBackpropInputKernel() argument
89 const int w_out = out_idx2 % output_cols; in DilationBackpropInputKernel()
90 const int out_idx3 = out_idx2 / output_cols; in DilationBackpropInputKernel()
132 int output_cols, int stride_rows, int stride_cols, int rate_rows, in DilationBackpropFilterKernel() argument
138 const int w_out = out_idx2 % output_cols; in DilationBackpropFilterKernel()
139 const int out_idx3 = out_idx2 / output_cols; in DilationBackpropFilterKernel()
194 const int output_cols = output.dimension(2); in operator ()() local
[all …]
Deigen_spatial_convolutions_test.cc37 const int output_cols = input_cols; in TEST() local
41 Tensor<float, 3> result(output_depth, output_rows, output_cols); in TEST()
51 EXPECT_EQ(result.dimension(2), output_cols); in TEST()
55 for (int j = 0; j < output_cols; ++j) { in TEST()
61 c - 1 + j < output_cols) { in TEST()
82 const int output_cols = input_cols; in TEST() local
87 Tensor<float, 3, RowMajor> result(output_cols, output_rows, output_depth); in TEST()
94 EXPECT_EQ(result.dimension(0), output_cols); in TEST()
100 for (int j = 0; j < output_cols; ++j) { in TEST()
106 c - 1 + j < output_cols) { in TEST()
[all …]
Ddilation_ops.cc177 const int output_cols = output.dimension(2); in operator ()() local
184 for (int w_out = 0; w_out < output_cols; ++w_out) { in operator ()()
286 const int output_cols = out_backprop.dimension(2); in operator ()() local
299 for (int w_out = 0; w_out < output_cols; ++w_out) { in operator ()()
406 const int output_cols = out_backprop.dimension(2); in operator ()() local
419 for (int w_out = 0; w_out < output_cols; ++w_out) { in operator ()()
/external/libjpeg-turbo/
Djcsample.c95 JDIMENSION output_cols) in expand_right_edge() argument
101 int numcols = (int)(output_cols - input_cols); in expand_right_edge()
152 JDIMENSION output_cols = compptr->width_in_blocks * DCTSIZE; in int_downsample() local
166 output_cols * h_expand); in int_downsample()
171 for (outcol = 0, outcol_h = 0; outcol < output_cols; in int_downsample()
224 JDIMENSION output_cols = compptr->width_in_blocks * DCTSIZE; in h2v1_downsample() local
233 output_cols * 2); in h2v1_downsample()
239 for (outcol = 0; outcol < output_cols; outcol++) { in h2v1_downsample()
261 JDIMENSION output_cols = compptr->width_in_blocks * DCTSIZE; in h2v2_downsample() local
270 output_cols * 2); in h2v2_downsample()
[all …]
/external/libjpeg-turbo/simd/powerpc/
Djcsample-altivec.c37 JDIMENSION output_cols = width_in_blocks * DCTSIZE; in jsimd_h2v1_downsample_altivec() local
51 output_cols * 2); in jsimd_h2v1_downsample_altivec()
57 for (outcol = output_cols; outcol > 0; in jsimd_h2v1_downsample_altivec()
93 JDIMENSION output_cols = width_in_blocks * DCTSIZE; in jsimd_h2v2_downsample_altivec() local
108 output_cols * 2); in jsimd_h2v2_downsample_altivec()
117 for (outcol = output_cols; outcol > 0; in jsimd_h2v2_downsample_altivec()
Djcsample.h12 JDIMENSION output_cols) in expand_right_edge() argument
18 int numcols = (int)(output_cols - input_cols); in expand_right_edge()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_conv2d.cc30 int64 output_cols, int64 row_stride, int64 col_stride, int64 padding_top, in __xla_cpu_runtime_EigenSingleThreadedConvF16() argument
37 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
47 int64 kernel_filters, int64 output_rows, int64 output_cols, in __xla_cpu_runtime_EigenSingleThreadedConvF32() argument
54 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
Druntime_conv2d_impl.h32 int64 kernel_filters, int64 output_rows, int64 output_cols, in EigenConvImpl() argument
48 output(out, input_batch, output_rows, output_cols, kernel_filters); in EigenConvImpl()
58 pre_contract_dims[0] = output_cols * output_rows * input_batch; in EigenConvImpl()
65 post_contract_dims[2] = output_cols; in EigenConvImpl()
Druntime_conv2d.cc31 int64 kernel_filters, int64 output_rows, int64 output_cols, in __xla_cpu_runtime_EigenConvF32() argument
40 kernel_channels, kernel_filters, output_rows, output_cols, row_stride, in __xla_cpu_runtime_EigenConvF32()
50 int64 output_cols, int64 row_stride, int64 col_stride, int64 padding_top, in __xla_cpu_runtime_EigenConvF16() argument
59 kernel_channels, kernel_filters, output_rows, output_cols, row_stride, in __xla_cpu_runtime_EigenConvF16()
Druntime_conv2d_mkl.cc56 int64 output_rows, int64 output_cols, int64 row_stride, in MKLConvImpl() argument
74 ToInt(output_rows), ToInt(output_cols)}; in MKLConvImpl()
156 int64 kernel_filters, int64 output_rows, int64 output_cols, in __xla_cpu_runtime_MKLConvF32() argument
167 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_MKLConvF32()
173 kernel_filters, output_rows, output_cols, row_stride, in __xla_cpu_runtime_MKLConvF32()
Druntime_single_threaded_conv2d.h31 tensorflow::int64 output_rows, tensorflow::int64 output_cols,
45 tensorflow::int64 output_cols, tensorflow::int64 row_stride,
Druntime_conv2d.h31 tensorflow::int64 output_cols, tensorflow::int64 row_stride,
45 tensorflow::int64 output_rows, tensorflow::int64 output_cols,
Druntime_conv2d_mkl.h31 tensorflow::int64 output_cols, tensorflow::int64 row_stride,
/external/libjpeg-turbo/simd/loongson/
Djcsample-mmi.c43 JDIMENSION output_cols = width_in_blocks * DCTSIZE; in jsimd_h2v2_downsample_mmi() local
48 output_cols * 2); in jsimd_h2v2_downsample_mmi()
62 for (outcol = output_cols; outcol > 0; in jsimd_h2v2_downsample_mmi()
Djcsample.h12 JDIMENSION output_cols) in expand_right_edge() argument
18 int numcols = (int)(output_cols - input_cols); in expand_right_edge()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc224 DimensionHandle output_cols = transpose_b ? c->Dim(b, 0) : c->Dim(b, 1); in MatMulShape() local
232 c->set_output(0, c->Matrix(output_rows, output_cols)); in MatMulShape()
518 DimensionHandle output_rows, output_cols; in Conv2DShapeImpl() local
532 padding, pad_cols_before, pad_cols_after, &output_cols)); in Conv2DShapeImpl()
536 ShapeFromDimensions(batch_size_dim, {output_rows, output_cols}, in Conv2DShapeImpl()
622 DimensionHandle output_planes, output_rows, output_cols; in Conv3DShape() local
632 -1, &output_cols)); in Conv3DShape()
637 output_planes, output_rows, output_cols}); in Conv3DShape()
640 output_cols, output_depth_dim}); in Conv3DShape()
701 DimensionHandle output_rows, output_cols; in DepthwiseConv2DNativeShape() local
[all …]
/external/libjpeg-turbo/simd/x86_64/
Djcsample-sse2.asm53 shl rcx, 3 ; imul rcx,DCTSIZE (rcx = output_cols)
61 shl rcx, 1 ; output_cols * 2
89 pop rcx ; output_cols
198 shl rcx, 3 ; imul rcx,DCTSIZE (rcx = output_cols)
206 shl rcx, 1 ; output_cols * 2
234 pop rcx ; output_cols
Djcsample-avx2.asm54 shl rcx, 3 ; imul rcx,DCTSIZE (rcx = output_cols)
62 shl rcx, 1 ; output_cols * 2
90 pop rcx ; output_cols
216 shl rcx, 3 ; imul rcx,DCTSIZE (rcx = output_cols)
224 shl rcx, 1 ; output_cols * 2
252 pop rcx ; output_cols
/external/libjpeg-turbo/simd/i386/
Djcsample-mmx.asm56 shl ecx, 3 ; imul ecx,DCTSIZE (ecx = output_cols)
64 shl ecx, 1 ; output_cols * 2
93 pop ecx ; output_cols
199 shl ecx, 3 ; imul ecx,DCTSIZE (ecx = output_cols)
207 shl ecx, 1 ; output_cols * 2
236 pop ecx ; output_cols
Djcsample-sse2.asm56 shl ecx, 3 ; imul ecx,DCTSIZE (ecx = output_cols)
64 shl ecx, 1 ; output_cols * 2
93 pop ecx ; output_cols
212 shl ecx, 3 ; imul ecx,DCTSIZE (ecx = output_cols)
220 shl ecx, 1 ; output_cols * 2
249 pop ecx ; output_cols
Djcsample-avx2.asm57 shl ecx, 3 ; imul ecx,DCTSIZE (ecx = output_cols)
65 shl ecx, 1 ; output_cols * 2
94 pop ecx ; output_cols
230 shl ecx, 3 ; imul ecx,DCTSIZE (ecx = output_cols)
238 shl ecx, 1 ; output_cols * 2
267 pop ecx ; output_cols
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc211 int64 output_rows = 0, output_cols = 0, pad_rows = 0, pad_cols = 0; in Compute() local
217 &output_cols, &pad_cols)); in Compute()
220 data_format_, batch_size, output_rows, output_cols, output_depth); in Compute()
231 << ", output_cols = " << output_cols in Compute()
477 const int output_cols = GetTensorDim(*output_param, data_format, 'W'); in launch() local
498 0, (output_cols - 1) * col_stride + filter_cols - conv_input_cols); in launch()
558 ctx, side_input_param, batch_size, output_rows, output_cols, in launch()
566 FORMAT_NCHW, batch_size, output_rows, output_cols, output_depth); in launch()
596 .set_width(output_cols) in launch()
608 .set_width(output_cols) in launch()
/external/toybox/toys/pending/
Dstty.c47 unsigned output_cols;
141 if (TT.output_cols == 0) { in out()
142 TT.output_cols = 80; in out()
143 terminal_size(&TT.output_cols, NULL); in out()
147 else if (TT.col + 1 + len >= TT.output_cols) { in out()

12