Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/
Ddilation_ops_gpu.cu.cc41 int input_cols, int depth, int filter_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()
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
Dmkl_conv_ops.h214 int filter_rows = in GetFilterSizeInMklOrder() local
254 int filter_rows = in GetFilterSizeInMklOrder() local
343 int filter_planes, filter_rows, filter_cols; variable
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.h129 Eigen::Index filter_rows = filter_dims[0]; in SpatialConvolutionBackwardKernel() local
268 Eigen::Index filter_rows = filter_dims[0]; in CuboidConvolutionBackwardKernel() local
Ddepthwise_conv_op.h31 int filter_rows; member
Ddeep_conv2d.h73 int filter_rows; member
Dconv_ops.cc156 int input_cols, int in_depth, int filter_rows, in Run()
172 int input_cols, int in_depth, int filter_rows, in Run()
213 int input_cols, int in_depth, int filter_rows, in Run()
227 int input_cols, int in_depth, int filter_rows, in Run()
375 const int filter_rows = static_cast<int>(filter.dim_size(0)); in ComputeConv2DDimension() local
Dconv_ops_3d.cc220 const int64 filter_rows = filter.dim_size(1); in launch() local
Ddepthwise_conv_op.cc335 const int32 filter_rows = filter.dim_size(0); in Compute() local
Dconv_ops_using_gemm.cc496 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.cc106 int out_depth, int filter_rows, int filter_cols, in BM_ConvFloat()
500 int out_depth, int filter_rows, in BM_ConvFloatDepthwise()
Dconv_grad_filter_ops.cc142 auto filter_rows = filter.dimension(0); in operator ()() local
Dconv_grad_input_ops.cc149 auto filter_rows = kernel.dimension(0); in operator ()() local
Dconv_ops_fused_image_transform.cc799 const int filter_rows = static_cast<int>(filter.dim_size(0)); in Compute() local
Ddepthwise_conv_grad_op.cc188 const int64 filter_rows = args.filter_rows; 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.cc496 const int filter_rows = static_cast<int>(filter.dim_size(0)); in Compute() 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/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc203 const int32 filter_rows = GetFilterDim(filter, filter_format_, 'H'); in Compute() local
479 const int filter_rows = GetFilterDim(filter_param, filter_format, 'H'); in launch() local
/external/tensorflow/tensorflow/core/ops/
Dnn_ops.cc938 auto filter_rows = c->Value(filter_rows_dim); in __anon3e672dd81d02() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_test.py1638 def ConstructAndTestGradient(self, batch, input_rows, input_cols, filter_rows, argument
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h1184 const int filter_rows = filter_shape.Dims(filter_dim_count - 2); in FullyConnected() local
2244 const int filter_rows = filter_shape.Dims(0); in HybridConv() local
2422 const int filter_rows = filter_shape.Dims(0); in Conv() local
2611 const int filter_rows = filter_dims.sizes[3]; in ConvAsGemm() local

12