| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | dilation_ops_gpu.cu.cc | 41 int depth, int filter_rows, int filter_cols, int output_rows, in DilationKernel()
80 int batch, int input_rows, int input_cols, int depth, int filter_rows, in DilationBackpropInputKernel()
130 int batch, int input_rows, int input_cols, int depth, int filter_rows, in DilationBackpropFilterKernel()
190 const int filter_rows = filter.dimension(0); in operator ()() local
221 const int filter_rows = filter.dimension(0); in operator ()() local
263 const int filter_rows = filter.dimension(0); in operator ()() local
|
| H A D | dilation_ops.cc | 93 const int filter_rows = filter.dim_size(0); in ParseSizes() local
175 const int filter_rows = filter.dimension(0); in operator ()() local
290 const int filter_rows = filter.dimension(0); in operator ()() local
418 const int filter_rows = filter.dimension(0); in operator ()() local
|
| H A D | conv_ops.cc | 368 int input_cols, int in_depth, int filter_rows, in Run()
384 int input_cols, int in_depth, int filter_rows, in Run()
425 int input_cols, int in_depth, int filter_rows, in Run()
439 int input_cols, int in_depth, int filter_rows, in Run()
590 const int filter_rows = static_cast<int>(filter.dim_size(0)); in ComputeConv2DDimension() local
|
| H A D | eigen_benchmark.h | 126 Eigen::Index filter_rows = filter_dims[0]; in SpatialConvolutionBackwardKernel() local
261 Eigen::Index filter_rows = filter_dims[0]; in CuboidConvolutionBackwardKernel() local
|
| H A D | depthwise_conv_op.h | 31 int filter_rows; member
|
| H A D | deep_conv2d.h | 73 int filter_rows; member
|
| H A D | depthwise_conv_op.cc | 382 const int32_t filter_rows = filter.dim_size(0); in Compute() local
|
| H A D | conv_ops_3d.cc | 247 const int64_t filter_rows = filter.dim_size(1); in launch() local
|
| H A D | deep_conv2d.cc | 75 static int64_t GetDirectConvCost(int filter_rows, int filter_cols, int in_depth, in GetDirectConvCost()
98 bool CanUseDeepConv2D(int stride_rows, int stride_cols, int filter_rows, in CanUseDeepConv2D()
|
| H A D | nn_ops_test.cc | 111 int filter_rows, int filter_cols, CONV_OP op, in BM_ConvFloat()
521 int filter_rows, int filter_cols, in BM_ConvFloatDepthwise()
|
| H A D | conv_ops_using_gemm.cc | 498 const int filter_rows = static_cast<int>(filter.dim_size(0)); in Compute() local
|
| H A D | quantized_conv_ops.cc | 550 const int64_t filter_rows = filter.dim_size(0); in Compute() local
|
| H A D | conv_grad_filter_ops.cc | 220 auto filter_rows = filter.dimension(0); in operator ()() local
|
| H A D | conv_ops_fused_image_transform.cc | 797 const int filter_rows = static_cast<int>(filter.dim_size(0)); in Compute() local
|
| H A D | eigen_spatial_convolutions_test.cc | 1384 int filter_count, int filter_cols, int filter_rows, in PackRhsHelper()
1582 int filter_count, int filter_cols, int filter_rows, in PackLhsHelper()
|
| H A D | depthwise_conv_grad_op.cc | 201 const int64_t filter_rows = args.filter_rows; in CopyOutputBackpropRegion() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_conv_ops.h | 212 int filter_rows = in GetFilterSizeInMklOrder() local
265 int filter_rows = in GetFilterSizeInMklOrder() local
375 int filter_planes, filter_rows, filter_cols; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| H A D | fully_connected.h | 54 const int filter_rows = filter_shape.Dims(filter_dim_count - 2); in FullyConnectedPerChannel() local
126 const int filter_rows = filter_shape.Dims(filter_dim_count - 2); in FullyConnected() local
|
| H A D | conv.h | 86 const int filter_rows = filter_shape.Dims(0); in ConvPerChannel() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | use_cudnn.cc | 116 bool ShouldCudnnGroupedConvolutionBeUsed(const int32_t filter_rows, in ShouldCudnnGroupedConvolutionBeUsed()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | utils_test.cc | 68 int filter_rows = 3; in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | nn_ops.cc | 1066 auto filter_rows = c->Value(filter_rows_dim); in __anon84cbdd651702() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| H A D | conv_ops_test.py | 1883 filter_rows, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | legacy_optimized_ops.h | 1128 const int filter_rows = filter_shape.Dims(filter_dim_count - 2); in FullyConnected() local
2145 const int filter_rows = filter_shape.Dims(filter_dim_count - 2); in FullyConnected() local
2686 const int filter_rows = filter_shape.Dims(0); in Conv() local
2882 const int filter_rows = filter_dims.sizes[3]; in ConvAsGemm() local
|
| H A D | optimized_ops.h | 328 const int filter_rows = filter_shape.Dims(filter_dim_count - 2); in FullyConnected() local
1056 const int filter_rows = filter_shape.Dims(0); in HybridConv() local
1142 const int filter_rows = filter_shape.Dims(0); in HybridConvPerChannel() local
1268 const int filter_rows = filter_shape.Dims(0); in Conv() local
|