Home
last modified time | relevance | path

Searched defs:filter_cols (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Ddilation_ops_gpu.cu.cc42 int filter_cols, int output_rows, in DilationKernel()
82 int depth, int filter_rows, int filter_cols, int output_rows, in DilationBackpropInputKernel()
131 int depth, int filter_rows, int filter_cols, int output_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
Dmkl_conv_ops.h216 int filter_cols = in GetFilterSizeInMklOrder() local
256 int filter_cols = in GetFilterSizeInMklOrder() local
343 int filter_planes, filter_rows, filter_cols; variable
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.h130 Eigen::Index filter_cols = filter_dims[1]; in SpatialConvolutionBackwardKernel() local
269 Eigen::Index filter_cols = filter_dims[1]; in CuboidConvolutionBackwardKernel() local
Ddepthwise_conv_op.h32 int filter_cols; member
Ddeep_conv2d.h74 int filter_cols; member
Dconv_ops.cc157 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run()
173 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run()
214 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run()
228 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run()
383 const int filter_cols = static_cast<int>(filter.dim_size(1)); in ComputeConv2DDimension() local
Dconv_ops_3d.cc221 const int64 filter_cols = filter.dim_size(2); in launch() local
Ddepthwise_conv_op.cc343 const int32 filter_cols = filter.dim_size(1); in Compute() local
Dconv_ops_using_gemm.cc506 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.cc106 int out_depth, int filter_rows, int filter_cols, in BM_ConvFloat()
501 int filter_cols, DEPTHWISE_CONV_OP op, in BM_ConvFloatDepthwise()
Dconv_grad_filter_ops.cc143 auto filter_cols = filter.dimension(1); in operator ()() local
Dconv_grad_input_ops.cc150 auto filter_cols = kernel.dimension(1); in operator ()() local
Dconv_ops_fused_image_transform.cc810 const int filter_cols = static_cast<int>(filter.dim_size(1)); in Compute() local
Ddepthwise_conv_grad_op.cc189 const int64 filter_cols = args.filter_cols; in CopyOutputBackpropRegion() local
Deigen_spatial_convolutions_test.cc1383 int filter_count, int filter_cols, int filter_rows, in PackRhsHelper()
1556 int filter_count, int filter_cols, int filter_rows, in PackLhsHelper()
Dmkl_conv_ops.cc508 const int filter_cols = static_cast<int>(filter.dim_size(1)); in Compute() 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/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc204 const int32 filter_cols = GetFilterDim(filter, filter_format_, 'W'); in Compute() local
480 const int filter_cols = GetFilterDim(filter_param, filter_format, 'W'); in launch() local
/external/tensorflow/tensorflow/core/ops/
Dnn_ops.cc939 auto filter_cols = c->Value(filter_cols_dim); in __anon3e672dd81d02() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_test.py1639 filter_cols, in_depth, out_depth, stride_rows, argument
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h1185 const int filter_cols = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local
2245 const int filter_cols = FlatSizeSkipDim(filter_shape, 0); in HybridConv() local
2425 const int filter_cols = in Conv() local
2612 const int filter_cols = FlatSizeSkipDim(filter_dims, 3); in ConvAsGemm() local

12