Home
last modified time | relevance | path

Searched defs:output_rows (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions_test.cc37 const int output_rows = input_rows - patch_rows + 1; in TEST() local
86 const int output_rows = input_rows - patch_rows + 1; in TEST() local
138 const int output_rows = input_rows; in TEST() local
186 const int output_rows = input_rows; in TEST() local
238 const int output_rows = input_rows - patch_rows + 1; in TEST() local
293 const int output_rows = input_rows - patch_rows + 1; in TEST() local
348 const int output_rows = divup(input_rows - patch_rows + 1, stride); in test_batched_strided_spatial_convolution_backward_input_valid() local
424 const int output_rows = divup(input_rows - patch_rows + 1, stride); in test_batched_strided_spatial_convolution_backward_input_valid_row_major() local
501 const int output_rows = input_rows - patch_rows + 1; in TEST() local
555 const int output_rows = input_rows - patch_rows + 1; in TEST() local
[all …]
Deigen_pooling_test.cc34 const int output_rows = 2; in TEST() local
82 const int output_rows = 2; in TEST() local
133 const int output_rows = 2; in TEST() local
193 const int output_rows = 2; in TEST() local
253 const int output_rows = 2; in TEST() local
315 const int output_rows = 2; in TEST() local
377 const int output_rows = input_rows; in TEST() local
455 const int output_rows = input_rows; in TEST() local
531 const int output_rows = 2; in TEST() local
579 const int output_rows = 2; in TEST() local
[all …]
Ddilation_ops_gpu.cu.cc41 int depth, int filter_rows, int filter_cols, int output_rows, in DilationKernel()
81 int filter_cols, int output_rows, int output_cols, int stride_rows, in DilationBackpropInputKernel()
131 int filter_cols, int output_rows, int output_cols, int stride_rows, in DilationBackpropFilterKernel()
193 const int output_rows = output.dimension(1); in operator ()() local
224 const int output_rows = out_backprop.dimension(1); in operator ()() local
266 const int output_rows = out_backprop.dimension(1); in operator ()() local
Deigen_backward_cuboid_convolutions_test.cc38 const int output_rows = input_rows - patch_rows + 1; in TEST() local
101 const int output_rows = input_rows - patch_rows + 1; in TEST() local
164 const int output_rows = input_rows; in TEST() local
235 const int output_rows = input_rows; in TEST() local
307 const int output_rows = input_rows - patch_rows + 1; in TEST() local
374 const int output_rows = input_rows - patch_rows + 1; in TEST() local
441 const int output_rows = input_rows - patch_rows + 1; in TEST() local
507 const int output_rows = input_rows - patch_rows + 1; in TEST() local
574 const int output_rows = input_rows - patch_rows + 1; in TEST() local
641 const int output_rows = input_rows - patch_rows + 1; in TEST() local
[all …]
Dsegment_reduction_ops_impl.h106 const Index output_rows = in Compute() local
282 Index output_rows = *output_rows_host.data(); in ComputeAsync() local
Dwhere_op_gpu.cu.h41 const TIndex output_rows, const typename Eigen::array<TIndex, NDIM> strides, in PropagateWhereIndicesKernel()
Ddilation_ops.cc176 const int output_rows = output.dimension(1); in operator ()() local
285 const int output_rows = out_backprop.dimension(1); in operator ()() local
405 const int output_rows = out_backprop.dimension(1); in operator ()() local
Deigen_spatial_convolutions_test.cc36 const int output_rows = input_rows; in TEST() local
81 const int output_rows = input_rows; in TEST() local
201 const int output_rows = input_rows - patch_rows + 1; in TEST() local
256 const int output_rows = 2; in TEST() local
311 const int output_rows = input_rows - patch_rows + 1; in TEST() local
367 const int output_rows = 2; in TEST() local
415 const int output_rows = 2; in TEST() local
463 const int output_rows = 2; in TEST() local
514 const int output_rows = 3; in TEST() local
565 const int output_rows = 3; in TEST() local
[all …]
Dsegment_reduction_ops_gpu.cu.cc136 const Index output_rows, in operator ()()
/external/libopus/src/
Dmapping_matrix.c91 int output_rows, in mapping_matrix_multiply_channel_in_float()
125 int output_rows, in mapping_matrix_multiply_channel_out_float()
162 int output_rows, in mapping_matrix_multiply_channel_in_short()
202 int output_rows, in mapping_matrix_multiply_channel_out_short()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_conv2d.cc29 tensorflow::int64 kernel_filters, tensorflow::int64 output_rows, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
51 tensorflow::int64 output_rows, tensorflow::int64 output_cols, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
Druntime_conv2d.cc32 tensorflow::int64 output_rows, tensorflow::int64 output_cols, in __xla_cpu_runtime_EigenConvF32()
55 tensorflow::int64 kernel_filters, tensorflow::int64 output_rows, in __xla_cpu_runtime_EigenConvF16()
Druntime_conv2d_mkl.cc56 int64 output_rows, int64 output_cols, int64 row_stride, in MKLConvImpl()
158 int64 kernel_filters, int64 output_rows, int64 output_cols, in __xla_cpu_runtime_MKLConvF32()
Druntime_conv2d_impl.h37 Eigen::Index kernel_filters, Eigen::Index output_rows, in EigenConvImpl()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dfully_connected.h58 const int output_rows = output_shape.Dims(output_dim_count - 1); in FullyConnected() local
Dconv.h87 const int output_rows = output_shape.Dims(3); in ConvPerChannel() local
Ddepthwise_conv_hybrid.h65 const int output_rows = output_shape.Dims(1); in DepthwiseConvHybridGeneral() local
406 const int output_rows = output_shape.Dims(1); in DepthwiseConvHybridPerChannel() local
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc120 DimensionHandle output_rows = transpose_a ? c->Dim(a, 1) : c->Dim(a, 0); in MatMulShape() local
320 DimensionHandle output_rows = c->Dim(a_shape, adj_x ? -1 : -2); in BatchMatMulV2Shape() local
357 DimensionHandle output_rows = c->Dim(a_shape, adj_x ? -1 : -2); in BatchMatMulShape() local
701 DimensionHandle output_rows, output_cols; in Conv2DShapeImpl() local
825 DimensionHandle output_planes, output_rows, output_cols; in Conv3DShape() local
1015 DimensionHandle output_rows, output_cols; in DepthwiseConv2DNativeShapeImpl() local
1109 DimensionHandle output_rows, output_cols; in AvgPoolShape() local
1563 DimensionHandle output_rows, output_cols, output_depth; in MaxPoolShapeImpl() local
1688 DimensionHandle output_rows, output_cols, output_depth; in MaxPoolV2Shape() local
1765 DimensionHandle output_planes, output_rows, output_cols; in Pool3DShape() local
/external/tensorflow/tensorflow/core/ops/
Dsparse_csr_matrix_ops.cc269 auto output_rows = c->Dim(a_shape, transpose_a ? -1 : -2); in __anon5631e9dc0702() local
426 auto output_rows = c->Dim(a_shape, transpose_a ? -1 : -2); in __anon5631e9dc0a02() local
Dnn_ops.cc504 DimensionHandle output_rows, output_cols; in CommonFusedConvCalculations() local
1040 int64 output_rows, output_cols; in __anon4650176f1702() local
Darray_ops.cc2647 int64 output_rows, output_cols; in __anondb9326b23902() local
2753 int64 output_planes, output_rows, output_cols; in __anondb9326b23a02() local
/external/libjpeg-turbo/
Djcprepct.c110 int output_rows) in expand_bottom_edge()
/external/tensorflow/tensorflow/core/kernels/sparse/
Dtranspose_op.cc164 const int64 output_rows = output_dense_shape(rank - 2); in operator ()() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h450 const int output_rows = output_shape.Dims(1); variable
552 const int output_rows = output_shape.Dims(1); variable
1054 const int output_rows = output_shape.Dims(output_dim_count - 1); in FullyConnectedAsGEMV() local
1130 const int output_rows = output_shape.Dims(output_dim_count - 1); in FullyConnected() local
2065 const int output_rows = output_shape.Dims(output_dim_count - 1); in LegacyInt8FullyConnectedAsGEMV() local
2147 const int output_rows = output_shape.Dims(output_dim_count - 1); in FullyConnected() local
2690 const int output_rows = output_shape.Dims(3); in Conv() local
2883 const int output_rows = output_dims.sizes[0]; in ConvAsGemm() local
Doptimized_ops.h395 const int output_rows = output_shape.Dims(output_dim_count - 1); in FullyConnected() local
1399 const int output_rows = FlatSizeSkipDim(output_shape, 3); in HybridConv() local
1478 const int output_rows = output_shape.Dims(3); in HybridConvPerChannel() local
1604 const int output_rows = output_shape.Dims(3); in Conv() local

12