Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/
Ddilation_ops_gpu.cu.cc41 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
Ddilation_ops.cc92 const int filter_rows = filter.dim_size(0); in ParseSizes() local
173 const int filter_rows = filter.dimension(0); in operator ()() local
282 const int filter_rows = filter.dimension(0); in operator ()() local
402 const int filter_rows = filter.dimension(0); in operator ()() local
Deigen_benchmark.h126 Eigen::Index filter_rows = filter_dims[0]; in SpatialConvolutionBackwardKernel() local
261 Eigen::Index filter_rows = filter_dims[0]; in CuboidConvolutionBackwardKernel() local
Dconv_ops.cc223 int input_cols, int in_depth, int filter_rows, in Run()
239 int input_cols, int in_depth, int filter_rows, in Run()
280 int input_cols, int in_depth, int filter_rows, in Run()
294 int input_cols, int in_depth, int filter_rows, in Run()
442 const int filter_rows = static_cast<int>(filter.dim_size(0)); in ComputeConv2DDimension() local
Ddepthwise_conv_op.h31 int filter_rows; member
Ddeep_conv2d.h73 int filter_rows; member
Ddepthwise_conv_op.cc361 const int32 filter_rows = filter.dim_size(0); in Compute() local
Dconv_ops_3d.cc229 const int64 filter_rows = filter.dim_size(1); in launch() local
Dconv_ops_using_gemm.cc498 const int filter_rows = static_cast<int>(filter.dim_size(0)); in Compute() local
Dquantized_conv_ops.cc525 const int64 filter_rows = filter.dim_size(0); in Compute() local
Ddeep_conv2d.cc74 static int64 GetDirectConvCost(int filter_rows, int filter_cols, int in_depth, in GetDirectConvCost()
97 bool CanUseDeepConv2D(int stride_rows, int stride_cols, int filter_rows, in CanUseDeepConv2D()
Dnn_ops_test.cc108 int filter_rows, int filter_cols, CONV_OP op, in BM_ConvFloat()
517 int filter_rows, int filter_cols, in BM_ConvFloatDepthwise()
Dconv_grad_filter_ops.cc216 auto filter_rows = filter.dimension(0); in operator ()() local
Dconv_ops_fused_image_transform.cc796 const int filter_rows = static_cast<int>(filter.dim_size(0)); in Compute() local
Ddepthwise_conv_grad_op.cc200 const int64 filter_rows = args.filter_rows; in CopyOutputBackpropRegion() local
Deigen_spatial_convolutions_test.cc1384 int filter_count, int filter_cols, int filter_rows, in PackRhsHelper()
1582 int filter_count, int filter_cols, int filter_rows, in PackLhsHelper()
/external/tensorflow/tensorflow/core/util/
Duse_cudnn.cc73 bool IsCudnnSupportedFilterSize(const int32 filter_rows, in IsCudnnSupportedFilterSize()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.h220 int filter_rows = in GetFilterSizeInMklOrder() local
260 int filter_rows = in GetFilterSizeInMklOrder() local
349 int filter_planes, filter_rows, filter_cols; variable
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dfully_connected.h55 const int filter_rows = filter_shape.Dims(filter_dim_count - 2); in FullyConnected() local
Dconv.h85 const int filter_rows = filter_shape.Dims(0); in ConvPerChannel() local
/external/tensorflow/tensorflow/core/kernels/neon/
Dneon_depthwise_conv_op.cc82 const int32 filter_rows = filter.dim_size(0); in Compute() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils_test.cc68 int filter_rows = 3; in TEST() local
/external/tensorflow/tensorflow/core/ops/
Dnn_ops.cc1032 auto filter_rows = c->Value(filter_rows_dim); in __anon4650176f1702() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_test.py1854 filter_rows, argument
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h1127 const int filter_rows = filter_shape.Dims(filter_dim_count - 2); in FullyConnected() local
2144 const int filter_rows = filter_shape.Dims(filter_dim_count - 2); in FullyConnected() local
2685 const int filter_rows = filter_shape.Dims(0); in Conv() local
2881 const int filter_rows = filter_dims.sizes[3]; in ConvAsGemm() local

12