Home
last modified time | relevance | path

Searched refs:output_rows (Results 1 – 25 of 34) sorted by relevance

12

/external/libopus/src/
Dmapping_matrix.c91 int output_rows, in mapping_matrix_multiply_channel_in_float() argument
98 celt_assert(input_rows <= matrix->cols && output_rows <= matrix->rows); in mapping_matrix_multiply_channel_in_float()
112 output[output_rows * i] = FLOAT2INT16((1/32768.f)*tmp); in mapping_matrix_multiply_channel_in_float()
114 output[output_rows * i] = (1/32768.f)*tmp; in mapping_matrix_multiply_channel_in_float()
125 int output_rows, in mapping_matrix_multiply_channel_out_float() argument
134 celt_assert(input_rows <= matrix->cols && output_rows <= matrix->rows); in mapping_matrix_multiply_channel_out_float()
145 for (row = 0; row < output_rows; row++) in mapping_matrix_multiply_channel_out_float()
150 output[MATRIX_INDEX(output_rows, row, i)] += tmp; in mapping_matrix_multiply_channel_out_float()
162 int output_rows, in mapping_matrix_multiply_channel_in_short() argument
169 celt_assert(input_rows <= matrix->cols && output_rows <= matrix->rows); in mapping_matrix_multiply_channel_in_short()
[all …]
Dmapping_matrix.h71 int output_rows,
81 int output_rows,
92 int output_rows,
102 int output_rows,
/external/tensorflow/tensorflow/core/kernels/
Deigen_pooling_test.cc34 const int output_rows = 2; in TEST() local
38 Tensor<float, 4> result(depth, output_rows, output_cols, num_batches); in TEST()
49 EXPECT_EQ(result.dimension(1), output_rows); in TEST()
55 for (int i = 0; i < output_rows; ++i) { in TEST()
82 const int output_rows = 2; in TEST() local
86 Tensor<float, 4, RowMajor> result(num_batches, output_cols, output_rows, in TEST()
98 EXPECT_EQ(result.dimension(2), output_rows); in TEST()
104 for (int i = 0; i < output_rows; ++i) { in TEST()
133 const int output_rows = 2; in TEST() local
139 Tensor<float, 5> result(channels, output_planes, output_rows, output_cols, in TEST()
[all …]
Deigen_backward_spatial_convolutions_test.cc37 const int output_rows = input_rows - patch_rows + 1; in TEST() local
42 Tensor<float, 3> output_backward(output_depth, output_rows, output_cols); in TEST()
64 if (output_i >= 0 && output_i < output_rows && output_j >= 0 && in TEST()
86 const int output_rows = input_rows - patch_rows + 1; in TEST() local
93 Tensor<float, 3, RowMajor> output_backward(output_cols, output_rows, in TEST()
116 if (output_i >= 0 && output_i < output_rows && output_j >= 0 && in TEST()
139 const int output_rows = input_rows - patch_rows + 1; in TEST() local
148 Tensor<float, 4> output_backward(output_depth, output_planes, output_rows, in TEST()
176 output_j >= 0 && output_j < output_rows && in TEST()
202 const int output_rows = input_rows - patch_rows + 1; in TEST() local
[all …]
Ddilation_ops_gpu.cu.cc42 int filter_cols, int output_rows, in DilationKernel() argument
52 const int h_out = out_idx3 % output_rows; in DilationKernel()
53 const int b = out_idx3 / output_rows; in DilationKernel()
82 int depth, int filter_rows, int filter_cols, int output_rows, in DilationBackpropInputKernel() argument
91 const int h_out = out_idx3 % output_rows; in DilationBackpropInputKernel()
92 const int b = out_idx3 / output_rows; in DilationBackpropInputKernel()
131 int depth, int filter_rows, int filter_cols, int output_rows, in DilationBackpropFilterKernel() argument
140 const int h_out = out_idx3 % output_rows; in DilationBackpropFilterKernel()
141 const int b = out_idx3 / output_rows; in DilationBackpropFilterKernel()
193 const int output_rows = output.dimension(1); in operator ()() local
[all …]
Dsegment_reduction_ops.cc97 const Index output_rows = in Compute() local
101 OP_REQUIRES(context, output_rows >= 0, in Compute()
105 output_shape.set_dim(0, output_rows); in Compute()
112 OP_REQUIRES(context, output_rows > 0, in Compute()
153 context, FastBoundsCheck(out_index, output_rows), in Compute()
155 "Segment id ", out_index, " out of range [0, ", output_rows, in Compute()
271 Index output_rows = *output_rows_host.data(); in ComputeAsync() local
272 output_rows++; in ComputeAsync()
273 OP_REQUIRES_ASYNC(context, output_rows > 0, in ComputeAsync()
278 output_shape.set_dim(0, output_rows); in ComputeAsync()
[all …]
Deigen_spatial_convolutions_test.cc36 const int output_rows = input_rows; in TEST() local
41 Tensor<float, 3> result(output_depth, output_rows, output_cols); in TEST()
50 EXPECT_EQ(result.dimension(1), output_rows); in TEST()
54 for (int i = 0; i < output_rows; ++i) { in TEST()
60 if (r - 1 + i >= 0 && c - 1 + j >= 0 && r - 1 + i < output_rows && in TEST()
81 const int output_rows = input_rows; in TEST() local
87 Tensor<float, 3, RowMajor> result(output_cols, output_rows, output_depth); in TEST()
95 EXPECT_EQ(result.dimension(1), output_rows); in TEST()
99 for (int i = 0; i < output_rows; ++i) { in TEST()
105 if (r - 1 + i >= 0 && c - 1 + j >= 0 && r - 1 + i < output_rows && in TEST()
[all …]
Dwhere_op_gpu.cu.h44 const TIndex output_rows, const typename Eigen::array<TIndex, NDIM> strides, in PropagateWhereIndicesKernel() argument
49 CUDA_1D_KERNEL_LOOP(i, output_rows) { in PropagateWhereIndicesKernel()
325 const TIndex output_rows = output.dimension(0);
326 CudaLaunchConfig config = GetCudaLaunchConfig(output_rows, d);
329 d.stream(), output_rows, strides,
Ddilation_ops.cc176 const int output_rows = output.dimension(1); in operator ()() local
182 for (int h_out = 0; h_out < output_rows; ++h_out) { in operator ()()
285 const int output_rows = out_backprop.dimension(1); in operator ()() local
297 for (int h_out = 0; h_out < output_rows; ++h_out) { in operator ()()
405 const int output_rows = out_backprop.dimension(1); in operator ()() local
417 for (int h_out = 0; h_out < output_rows; ++h_out) { in operator ()()
Dsegment_reduction_ops_gpu.cu.cc132 OpKernelContext* ctx, const GPUDevice& d, const Index output_rows, in operator ()() argument
169 input_outer_dim_size, input_inner_dim_size, output_rows, in operator ()()
Dsegment_reduction_ops.h65 const Index output_rows, const TensorShape& segment_ids_shape,
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_conv2d.cc29 int64 kernel_channels, int64 kernel_filters, int64 output_rows, 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.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()
49 int64 kernel_channels, int64 kernel_filters, int64 output_rows, in __xla_cpu_runtime_EigenConvF16() argument
59 kernel_channels, kernel_filters, output_rows, output_cols, row_stride, in __xla_cpu_runtime_EigenConvF16()
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()
64 post_contract_dims[1] = output_rows; in EigenConvImpl()
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,
44 tensorflow::int64 kernel_filters, tensorflow::int64 output_rows,
Druntime_conv2d.h30 tensorflow::int64 kernel_filters, tensorflow::int64 output_rows,
45 tensorflow::int64 output_rows, tensorflow::int64 output_cols,
Druntime_conv2d_mkl.h30 tensorflow::int64 kernel_filters, tensorflow::int64 output_rows,
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc223 DimensionHandle output_rows = transpose_a ? c->Dim(a, 1) : c->Dim(a, 0); 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
529 padding, pad_rows_before, pad_rows_after, &output_rows)); in Conv2DShapeImpl()
536 ShapeFromDimensions(batch_size_dim, {output_rows, output_cols}, in Conv2DShapeImpl()
622 DimensionHandle output_planes, output_rows, output_cols; in Conv3DShape() local
629 -1, &output_rows)); in Conv3DShape()
637 output_planes, output_rows, output_cols}); in Conv3DShape()
639 output_shape = c->MakeShape({batch_size_dim, output_planes, output_rows, in Conv3DShape()
701 DimensionHandle output_rows, output_cols; in DepthwiseConv2DNativeShape() local
[all …]
/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
214 &output_rows, &pad_rows)); in Compute()
220 data_format_, batch_size, output_rows, output_cols, output_depth); in Compute()
232 << ", output_rows = " << output_rows in Compute()
476 const int output_rows = GetTensorDim(*output_param, data_format, 'H'); in launch() local
496 0, (output_rows - 1) * row_stride + filter_rows - conv_input_rows); 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()
595 .set_height(output_rows) in launch()
607 .set_height(output_rows) in launch()
/external/tensorflow/tensorflow/python/ops/
Ddata_flow_grad.py61 output_rows = output_shape[0]
62 grad = math_ops.unsorted_segment_sum(grad.values, grad.indices, output_rows)
/external/libjpeg-turbo/
Djcprepct.c110 int output_rows) in expand_bottom_edge() argument
114 for (row = input_rows; row < output_rows; row++) { in expand_bottom_edge()
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_3d_test.py387 output_rows = int(
393 output_rows = int(math.ceil(float(input_rows) / strides[2]))
395 output_shape = [batch, output_planes, output_rows, output_cols, out_depth]
/external/tensorflow/tensorflow/contrib/bigtable/python/kernel_tests/
Dbigtable_ops_test.py50 output_rows = dataset_ops.Dataset.from_tensor_slices(self.COMMON_ROW_KEYS)
52 return dataset_ops.Dataset.zip((output_rows, output_values))
/external/tensorflow/tensorflow/python/framework/
Dcommon_shapes.py110 output_rows = a_shape[1] if transpose_a else a_shape[0]
115 return [tensor_shape.TensorShape([output_rows, output_cols])]

12