Home
last modified time | relevance | path

Searched defs:filter_cols (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()
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.cc94 const int filter_cols = filter.dim_size(1); in ParseSizes() local
176 const int filter_cols = filter.dimension(1); in operator ()() local
291 const int filter_cols = filter.dimension(1); in operator ()() local
419 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
Ddepthwise_conv_op.h32 int filter_cols; member
Dconv_ops.cc369 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run()
385 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run()
426 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run()
440 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run()
598 const int filter_cols = static_cast<int>(filter.dim_size(1)); in ComputeConv2DDimension() local
Ddeep_conv2d.h74 int filter_cols; member
Ddepthwise_conv_op.cc390 const int32_t filter_cols = filter.dim_size(1); in Compute() local
Dconv_ops_3d.cc248 const int64_t 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.cc555 const int64_t filter_cols = filter.dim_size(1); in Compute() local
Ddeep_conv2d.cc75 static int64_t GetDirectConvCost(int filter_rows, int filter_cols, int in_depth, in GetDirectConvCost()
99 int filter_cols, int in_depth, int out_depth, in CanUseDeepConv2D()
Dnn_ops_test.cc111 int filter_rows, int filter_cols, CONV_OP op, in BM_ConvFloat()
521 int filter_rows, int filter_cols, in BM_ConvFloatDepthwise()
Dconv_grad_filter_ops.cc221 auto filter_cols = filter.dimension(1); in operator ()() local
Dconv_ops_fused_image_transform.cc808 const int filter_cols = static_cast<int>(filter.dim_size(1)); in Compute() local
Ddepthwise_conv_grad_op.cc202 const int64_t 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/lite/kernels/internal/optimized/integer_ops/
Dfully_connected.h55 const int filter_cols = filter_shape.Dims(filter_dim_count - 1); in FullyConnectedPerChannel() local
127 const int filter_cols = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local
Dconv.h87 const int filter_cols = FlatSizeSkipDim(filter_shape, 0); in ConvPerChannel() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.h214 int filter_cols = in GetFilterSizeInMklOrder() local
267 int filter_cols = in GetFilterSizeInMklOrder() local
375 int filter_planes, filter_rows, filter_cols; variable
/external/tensorflow/tensorflow/core/util/
Duse_cudnn.cc117 const int32_t filter_cols, in ShouldCudnnGroupedConvolutionBeUsed()
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils_test.cc69 int filter_cols = 3; in TEST() local
/external/tensorflow/tensorflow/core/ops/
Dnn_ops.cc1067 auto filter_cols = c->Value(filter_cols_dim); in __anoncb4947e51702() local
/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
Dconv_ops_test.py1884 filter_cols, argument
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h1129 const int filter_cols = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local
2146 const int filter_cols = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local
2689 const int filter_cols = in Conv() local
2883 const int filter_cols = FlatSizeSkipDim(filter_dims, 3); in ConvAsGemm() local
Doptimized_ops.h329 const int filter_cols = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local
1057 const int filter_cols = FlatSizeSkipDim(filter_shape, 0); in HybridConv() local
1143 const int filter_cols = FlatSizeSkipDim(filter_shape, 0); in HybridConvPerChannel() local
1271 const int filter_cols = in Conv() local