Home
last modified time | relevance | path

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

/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.cc93 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
Deigen_benchmark.h126 Eigen::Index filter_rows = filter_dims[0]; in SpatialConvolutionBackwardKernel() local
261 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.cc357 int input_cols, int in_depth, int filter_rows, in Run()
373 int input_cols, int in_depth, int filter_rows, in Run()
414 int input_cols, int in_depth, int filter_rows, in Run()
428 int input_cols, int in_depth, int filter_rows, in Run()
579 const int filter_rows = static_cast<int>(filter.dim_size(0)); in ComputeConv2DDimension() local
Ddepthwise_conv_op.cc362 const int32_t filter_rows = filter.dim_size(0); in Compute() local
Dconv_ops_3d.cc243 const int64_t 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.cc534 const int64_t 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_ops_fused_image_transform.cc797 const int filter_rows = static_cast<int>(filter.dim_size(0)); in Compute() local
Dconv_grad_filter_ops.cc219 auto filter_rows = filter.dimension(0); in operator ()() local
Ddepthwise_conv_grad_op.cc200 const int64_t 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/kernels/mkl/
Dmkl_conv_ops.h216 int filter_rows = in GetFilterSizeInMklOrder() local
272 int filter_rows = in GetFilterSizeInMklOrder() local
362 int filter_planes, filter_rows, filter_cols; variable
/external/tensorflow/tensorflow/core/util/
Duse_cudnn.cc95 bool IsCudnnSupportedFilterSize(const int32_t filter_rows, in IsCudnnSupportedFilterSize()
/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/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 __anon97b5e01b1702() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_test.py1860 filter_rows, argument
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h1126 const int filter_rows = filter_shape.Dims(filter_dim_count - 2); in FullyConnected() local
2143 const int filter_rows = filter_shape.Dims(filter_dim_count - 2); in FullyConnected() local
2684 const int filter_rows = filter_shape.Dims(0); in Conv() local
2880 const int filter_rows = filter_dims.sizes[3]; in ConvAsGemm() local
Doptimized_ops.h390 const int filter_rows = filter_shape.Dims(filter_dim_count - 2); in FullyConnected() local
1385 const int filter_rows = filter_shape.Dims(0); in HybridConv() local
1471 const int filter_rows = filter_shape.Dims(0); in HybridConvPerChannel() local
1597 const int filter_rows = filter_shape.Dims(0); in Conv() local