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
280 Index output_rows = *output_rows_host.data(); in ComputeAsync() local
Dwhere_op_gpu.cu.h54 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.cc131 OpKernelContext* ctx, const GPUDevice& d, 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 int64 kernel_channels, int64 kernel_filters, int64 output_rows, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
47 int64 kernel_filters, int64 output_rows, int64 output_cols, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
Druntime_conv2d.cc32 int64 kernel_filters, int64 output_rows, int64 output_cols, in __xla_cpu_runtime_EigenConvF32()
51 int64 kernel_channels, int64 kernel_filters, int64 output_rows, in __xla_cpu_runtime_EigenConvF16()
Druntime_conv2d_mkl.cc56 int64 output_rows, int64 output_cols, int64 row_stride, in MKLConvImpl()
156 int64 kernel_filters, int64 output_rows, int64 output_cols, in __xla_cpu_runtime_MKLConvF32()
Druntime_conv2d_impl.h33 Eigen::Index kernel_filters, Eigen::Index output_rows, in EigenConvImpl()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dfully_connected.h56 const int output_rows = output_shape.Dims(output_dim_count - 1); in FullyConnected() local
Dconv.h88 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.cc232 DimensionHandle output_rows = transpose_a ? c->Dim(a, 1) : c->Dim(a, 0); in MatMulShape() local
429 DimensionHandle output_rows = c->Dim(a_shape, adj_x ? -1 : -2); in BatchMatMulV2Shape() local
466 DimensionHandle output_rows = c->Dim(a_shape, adj_x ? -1 : -2); in BatchMatMulShape() local
789 DimensionHandle output_rows, output_cols; in Conv2DShapeImpl() local
913 DimensionHandle output_planes, output_rows, output_cols; in Conv3DShape() local
992 DimensionHandle output_rows, output_cols; in DepthwiseConv2DNativeShape() local
1066 DimensionHandle output_rows, output_cols; in AvgPoolShape() local
1495 DimensionHandle output_rows, output_cols, output_depth; in MaxPoolShape() local
1597 DimensionHandle output_rows, output_cols, output_depth; in MaxPoolV2Shape() local
1674 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 __anond96a6a670702() local
426 auto output_rows = c->Dim(a_shape, transpose_a ? -1 : -2); in __anond96a6a670a02() local
Dnn_ops.cc489 DimensionHandle output_rows, output_cols; in CommonFusedConvCalculations() local
1011 int64 output_rows, output_cols; in __anon87c0335a1d02() local
Darray_ops.cc2595 int64 output_rows, output_cols; in __anonf6523ebd3802() local
2701 int64 output_planes, output_rows, output_cols; in __anonf6523ebd3902() 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.h451 const int output_rows = output_shape.Dims(1); variable
552 const int output_rows = output_shape.Dims(1); variable
1053 const int output_rows = output_shape.Dims(output_dim_count - 1); in FullyConnectedAsGEMV() local
1129 const int output_rows = output_shape.Dims(output_dim_count - 1); in FullyConnected() local
2064 const int output_rows = output_shape.Dims(output_dim_count - 1); in LegacyInt8FullyConnectedAsGEMV() local
2146 const int output_rows = output_shape.Dims(output_dim_count - 1); in FullyConnected() local
2689 const int output_rows = output_shape.Dims(3); in Conv() local
2882 const int output_rows = output_dims.sizes[0]; in ConvAsGemm() local
Doptimized_ops.h335 const int output_rows = output_shape.Dims(output_dim_count - 1); in FullyConnected() local
1300 const int output_rows = FlatSizeSkipDim(output_shape, 3); in HybridConv() local
1375 const int output_rows = FlatSizeSkipDim(output_shape, 3); in HybridConvPerChannel() local
1471 const int output_rows = output_shape.Dims(3); in Conv() local

12