Home
last modified time | relevance | path

Searched refs:output_cols (Results 1 – 25 of 35) 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()
139 const int output_cols = input_cols; in TEST() local
143 Tensor<float, 3> output_backward(output_depth, output_rows, output_cols); in TEST()
165 output_j < output_cols) { in TEST()
187 const int output_cols = input_cols; in TEST() local
[all …]
Ddilation_ops_gpu.cu.cc42 int output_cols, int stride_rows, int stride_cols, int rate_rows, in DilationKernel() argument
48 const int w_out = out_idx2 % output_cols; in DilationKernel()
49 const int out_idx3 = out_idx2 / output_cols; in DilationKernel()
81 int filter_cols, int output_rows, int output_cols, int stride_rows, in DilationBackpropInputKernel() argument
88 const int w_out = out_idx2 % output_cols; in DilationBackpropInputKernel()
89 const int out_idx3 = out_idx2 / output_cols; in DilationBackpropInputKernel()
131 int filter_cols, int output_rows, int output_cols, int stride_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_backward_cuboid_convolutions_test.cc39 const int output_cols = input_cols - patch_cols + 1; in TEST() local
48 output_cols); in TEST()
76 output_k >= 0 && output_k < output_cols) { in TEST()
102 const int output_cols = input_cols - patch_cols + 1; in TEST() local
110 Tensor<float, 4, RowMajor> output_backward(output_cols, output_rows, in TEST()
139 output_k >= 0 && output_k < output_cols) { in TEST()
165 const int output_cols = input_cols; in TEST() local
174 output_cols); in TEST()
210 output_k >= 0 && output_k < output_cols) { in TEST()
236 const int output_cols = input_cols; in TEST() 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()
260 JDIMENSION output_cols = compptr->width_in_blocks * DCTSIZE; in h2v2_downsample() local
269 output_cols * 2); in h2v2_downsample()
[all …]
/external/angle/src/libANGLE/
Dgen_overlay_fonts.py158 output_cols = 32
159 assert (len(chars) == output_rows * output_cols)
189 font_array_width = output_cols * glyph_width // 32
195 base_x = (charIndex % output_cols) * glyph_width
196 base_y = (charIndex // output_cols) * glyph_height
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_conv2d_impl.h38 Eigen::Index output_cols, Eigen::Index row_stride, in EigenConvImpl() argument
54 output(out, input_batch, output_rows, output_cols, kernel_filters); in EigenConvImpl()
64 pre_contract_dims[0] = output_cols * output_rows * input_batch; in EigenConvImpl()
71 post_contract_dims[2] = output_cols; in EigenConvImpl()
Druntime_single_threaded_conv2d.cc30 tensorflow::int64 output_cols, tensorflow::int64 row_stride, in __xla_cpu_runtime_EigenSingleThreadedConvF16() argument
39 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
51 tensorflow::int64 output_rows, tensorflow::int64 output_cols, in __xla_cpu_runtime_EigenSingleThreadedConvF32() argument
60 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
Druntime_conv2d.cc32 tensorflow::int64 output_rows, tensorflow::int64 output_cols, in __xla_cpu_runtime_EigenConvF32() argument
44 kernel_channels, kernel_filters, output_rows, output_cols, row_stride, in __xla_cpu_runtime_EigenConvF32()
56 tensorflow::int64 output_cols, tensorflow::int64 row_stride, in __xla_cpu_runtime_EigenConvF16() argument
68 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()
158 int64 kernel_filters, int64 output_rows, int64 output_cols, in __xla_cpu_runtime_MKLConvF32() argument
169 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_MKLConvF32()
175 kernel_filters, output_rows, output_cols, row_stride, in __xla_cpu_runtime_MKLConvF32()
Druntime_conv2d.h31 tensorflow::int64 output_cols, tensorflow::int64 row_stride,
45 tensorflow::int64 output_rows, tensorflow::int64 output_cols,
Druntime_single_threaded_conv2d.h31 tensorflow::int64 output_rows, tensorflow::int64 output_cols,
45 tensorflow::int64 output_cols, tensorflow::int64 row_stride,
Druntime_conv2d_mkl.h31 tensorflow::int64 output_cols, tensorflow::int64 row_stride,
/external/libjpeg-turbo/simd/x86_64/
Djcsample-sse2.asm52 shl rcx, 3 ; imul rcx,DCTSIZE (rcx = output_cols)
60 shl rcx, 1 ; output_cols * 2
88 pop rcx ; output_cols
197 shl rcx, 3 ; imul rcx,DCTSIZE (rcx = output_cols)
205 shl rcx, 1 ; output_cols * 2
233 pop rcx ; output_cols
Djcsample-avx2.asm53 shl rcx, 3 ; imul rcx,DCTSIZE (rcx = output_cols)
61 shl rcx, 1 ; output_cols * 2
89 pop rcx ; output_cols
215 shl rcx, 3 ; imul rcx,DCTSIZE (rcx = output_cols)
223 shl rcx, 1 ; output_cols * 2
251 pop rcx ; output_cols
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dconv.h90 const int output_cols = in ConvPerChannel() local
93 TFLITE_DCHECK_EQ(output_cols, gemm_input_cols); in ConvPerChannel()
109 dst_params.cols = output_cols; in ConvPerChannel()
/external/libjpeg-turbo/simd/i386/
Djcsample-mmx.asm54 shl ecx, 3 ; imul ecx,DCTSIZE (ecx = output_cols)
62 shl ecx, 1 ; output_cols * 2
91 pop ecx ; output_cols
197 shl ecx, 3 ; imul ecx,DCTSIZE (ecx = output_cols)
205 shl ecx, 1 ; output_cols * 2
234 pop ecx ; output_cols
Djcsample-sse2.asm54 shl ecx, 3 ; imul ecx,DCTSIZE (ecx = output_cols)
62 shl ecx, 1 ; output_cols * 2
91 pop ecx ; output_cols
210 shl ecx, 3 ; imul ecx,DCTSIZE (ecx = output_cols)
218 shl ecx, 1 ; output_cols * 2
247 pop ecx ; output_cols
Djcsample-avx2.asm55 shl ecx, 3 ; imul ecx,DCTSIZE (ecx = output_cols)
63 shl ecx, 1 ; output_cols * 2
92 pop ecx ; output_cols
228 shl ecx, 3 ; imul ecx,DCTSIZE (ecx = output_cols)
236 shl ecx, 1 ; output_cols * 2
265 pop ecx ; output_cols
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc121 DimensionHandle output_cols = transpose_b ? c->Dim(b, 0) : c->Dim(b, 1); in MatMulShape() local
129 c->set_output(0, c->Matrix(output_rows, output_cols)); in MatMulShape()
321 DimensionHandle output_cols = c->Dim(b_shape, adj_y ? -2 : -1); in BatchMatMulV2Shape() local
340 output_batch_shape, c->Matrix(output_rows, output_cols), &output_shape)); in BatchMatMulV2Shape()
358 DimensionHandle output_cols = c->Dim(b_shape, adj_y ? -2 : -1); in BatchMatMulShape() local
375 c->Concatenate(batch_dims, c->Matrix(output_rows, output_cols), &out)); in BatchMatMulShape()
701 DimensionHandle output_rows, output_cols; in Conv2DShapeImpl() local
715 padding, pad_cols_before, pad_cols_after, &output_cols)); in Conv2DShapeImpl()
719 ShapeFromDimensions(batch_size_dim, {output_rows, output_cols}, in Conv2DShapeImpl()
825 DimensionHandle output_planes, output_rows, output_cols; in Conv3DShape() local
[all …]
/external/toybox/toys/pending/
Dstty.c47 unsigned output_cols;
144 if (TT.output_cols == 0) { in out()
145 TT.output_cols = 80; in out()
146 terminal_size(&TT.output_cols, NULL); in out()
150 else if (TT.col + 1 + len >= TT.output_cols) { in out()
/external/tensorflow/tensorflow/core/ops/
Dsparse_csr_matrix_ops.cc270 auto output_cols = c->Dim(b_shape, transpose_b ? -2 : -1); in __anon59a9d6c30702() local
272 std::tie(output_rows, output_cols) = in __anon59a9d6c30702()
273 std::make_tuple(output_cols, output_rows); in __anon59a9d6c30702()
292 batch_dims, c->Matrix(output_rows, output_cols), &out)); in __anon59a9d6c30702()
427 auto output_cols = c->Dim(b_shape, transpose_b ? -2 : -1); in __anon59a9d6c30a02() local
445 batch_dims, c->Matrix(output_rows, output_cols), &out)); in __anon59a9d6c30a02()
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_3d_test.py485 output_cols = int(
490 output_cols = int(math.ceil(float(input_cols) / strides[3]))
491 output_shape = [batch, output_planes, output_rows, output_cols, out_depth]

12