Home
last modified time | relevance | path

Searched defs:filter_cols (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()
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()
191 const int filter_cols = filter.dimension(1); in operator ()() local
222 const int filter_cols = filter.dimension(1); in operator ()() local
264 const int filter_cols = filter.dimension(1); in operator ()() local
Ddilation_ops.cc93 const int filter_cols = filter.dim_size(1); in ParseSizes() local
174 const int filter_cols = filter.dimension(1); in operator ()() local
283 const int filter_cols = filter.dimension(1); in operator ()() local
403 const int filter_cols = filter.dimension(1); in operator ()() local
Deigen_benchmark.h127 Eigen::Index filter_cols = filter_dims[1]; in SpatialConvolutionBackwardKernel() local
262 Eigen::Index filter_cols = filter_dims[1]; in CuboidConvolutionBackwardKernel() local
Dconv_ops.cc224 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run()
240 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run()
281 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run()
295 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run()
450 const int filter_cols = static_cast<int>(filter.dim_size(1)); in ComputeConv2DDimension() local
Ddepthwise_conv_op.h32 int filter_cols; member
Ddeep_conv2d.h74 int filter_cols; member
Ddepthwise_conv_op.cc369 const int32 filter_cols = filter.dim_size(1); in Compute() local
Dconv_ops_3d.cc230 const int64 filter_cols = filter.dim_size(2); in launch() local
Dconv_ops_using_gemm.cc508 const int filter_cols = static_cast<int>(filter.dim_size(1)); in Compute() local
Dquantized_conv_ops.cc530 const int64 filter_cols = filter.dim_size(1); in Compute() local
Ddeep_conv2d.cc74 static int64 GetDirectConvCost(int filter_rows, int filter_cols, int in_depth, in GetDirectConvCost()
98 int filter_cols, int in_depth, int out_depth, 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.cc217 auto filter_cols = filter.dimension(1); in operator ()() local
Dconv_ops_fused_image_transform.cc807 const int filter_cols = static_cast<int>(filter.dim_size(1)); in Compute() local
Ddepthwise_conv_grad_op.cc201 const int64 filter_cols = args.filter_cols; 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.cc74 const int32 filter_cols, const int32 in_depth, in IsCudnnSupportedFilterSize()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.h222 int filter_cols = in GetFilterSizeInMklOrder() local
262 int filter_cols = in GetFilterSizeInMklOrder() local
349 int filter_planes, filter_rows, filter_cols; variable
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dfully_connected.h56 const int filter_cols = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local
Dconv.h86 const int filter_cols = FlatSizeSkipDim(filter_shape, 0); in ConvPerChannel() local
/external/tensorflow/tensorflow/core/kernels/neon/
Dneon_depthwise_conv_op.cc83 const int32 filter_cols = filter.dim_size(1); in Compute() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils_test.cc69 int filter_cols = 3; in TEST() local
/external/tensorflow/tensorflow/core/ops/
Dnn_ops.cc1033 auto filter_cols = c->Value(filter_cols_dim); in __anon4650176f1702() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_test.py1855 filter_cols, argument
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h1128 const int filter_cols = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local
2145 const int filter_cols = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local
2688 const int filter_cols = in Conv() local
2882 const int filter_cols = FlatSizeSkipDim(filter_dims, 3); in ConvAsGemm() local

12