| /external/tensorflow/tensorflow/core/kernels/ |
| deep_conv2d.h | 74 int filter_cols; member in struct:tensorflow::Conv2DArgs
89 filter_cols(0),
101 int filter_cols, int in_depth, int out_depth,
|
| dilation_ops_gpu.cu.cc | 42 int filter_cols, int output_rows,
60 for (int w = 0; w < filter_cols; ++w) {
66 filter_ptr[d + depth * (w + filter_cols * h)];
82 int depth, int filter_rows, int filter_cols, int output_rows,
99 // last branch, i.e., the one with largest value of `h * filter_cols + w`,
104 for (int w = 0; w < filter_cols; ++w) {
110 filter_ptr[d + depth * (w + filter_cols * h)];
131 int depth, int filter_rows, int filter_cols, int output_rows,
148 // last branch, i.e., the one with largest value of `h * filter_cols + w`,
153 for (int w = 0; w < filter_cols; ++w)
191 const int filter_cols = filter.dimension(1); local
222 const int filter_cols = filter.dimension(1); local
262 const int filter_cols = filter.dimension(1); local
[all...] |
| depthwise_conv_op.cc | 89 const int64 filter_spatial_size = args.filter_rows * args.filter_cols;
170 const int64 filter_spatial_size = args.filter_rows * args.filter_cols;
193 const int64 filter_spatial_size = args.filter_rows * args.filter_cols;
298 // [ filter_rows, filter_cols, in_depth, depth_multiplier]
336 const int32 filter_cols = filter.dim_size(1); variable
346 GetWindowedOutputSize(input_cols, filter_cols, stride_,
363 << filter_cols << ", " << in_depth << ", " << depth_multiplier
390 args.filter_cols = filter_cols;
|
| depthwise_conv_op.h | 32 int filter_cols; member in struct:tensorflow::DepthwiseArgs
49 filter_cols(0),
114 // [filter_rows, filter_cols, in_depth, depth_multiplier]
145 const int64 filter_spatial_size = args.filter_rows * args.filter_cols;
227 for (int64 f_c = 0; f_c < args.filter_cols; ++f_c) {
|
| dilation_ops.cc | 87 // [ filter_rows, filter_cols, depth ]
93 const int filter_cols = filter.dim_size(1); local
104 filter_cols + (filter_cols - 1) * (*rate_cols - 1);
174 const int filter_cols = filter.dimension(1); local
191 for (int w = 0; w < filter_cols; ++w) {
283 const int filter_cols = filter.dimension(1); local
294 // last branch, i.e., the one with largest value of `h * filter_cols + w`,
308 for (int w = 0; w < filter_cols; ++w) {
403 const int filter_cols = filter.dimension(1) local
[all...] |
| conv_ops_3d.cc | 183 const int64 filter_cols = filter.dim_size(2); local
197 0, (out_cols - 1) * strides[2] + filter_cols - in_cols);
201 if (filter_planes == 1 && filter_rows == 1 && filter_cols == 1 &&
228 filter_cols == in_cols && padding == Padding::VALID &&
327 filter_desc.set_spatial_dim(DimIndex::X, filter_cols)
344 filter_rows, filter_cols}),
379 {{filter_planes, filter_rows, filter_cols}},
|
| conv_ops_using_gemm.cc | 459 // [ filter_rows, filter_cols, in_depth, out_depth]
506 const int filter_cols = static_cast<int>(filter.dim_size(1)); variable
525 GetWindowedOutputSize(input_cols, filter_cols, stride_cols,
537 << ", filter_cols = " << filter_cols
550 in_depth, filter.flat<T>().data(), filter_rows, filter_cols,
|
| conv_ops.cc | 138 int filter_cols, int pad_rows, int pad_cols, int out_rows,
154 int filter_cols, int pad_rows, int pad_cols, int out_rows,
160 !CanUseDeepConv2D(stride_rows, stride_cols, filter_rows, filter_cols,
171 args.filter_cols = filter_cols;
195 int filter_cols, int pad_rows, int pad_cols, int out_rows,
209 int filter_cols, int pad_rows, int pad_cols, int out_rows,
223 desc.S = filter_cols;
308 // [ filter_rows, filter_cols, in_depth, out_depth]
355 const int filter_cols = static_cast<int>(filter.dim_size(1)) variable
[all...] |
| depthwise_conv_grad_op.cc | 80 const int64 filter_cols = filter_shape.dim_size(1); \
117 GetWindowedOutputSize(input_cols, filter_cols, stride, \
135 args.filter_cols = filter_cols; \
145 << "]; Filter: [" << filter_rows << ", " << filter_cols << ", " \
186 const int64 filter_cols = args.filter_cols; local
197 static_cast<int64>(0), (in_c - filter_cols + pad_cols + stride) / stride);
201 const int64 filter_spatial_size = args.filter_rows * args.filter_cols;
203 (out_c_end - out_c_start + 1) < args.filter_cols) {
[all...] |
| mkl_conv_ops.h | 166 int filter_cols = static_cast<int>(filter_shape.dim_size(1)); local
176 mkldnn_sizes[MklDnnDims::Dim_W] = filter_cols;
230 int filter_cols = filter_shape.dim_size(1); local
248 input_cols, filter_cols, stride_cols, padding_,
|
| quantized_conv_ops.cc | 488 // [ filter_rows, filter_cols, in_depth, out_depth]
530 const int64 filter_cols = filter.dim_size(1); variable
545 GetWindowedOutputSize(input_cols, filter_cols, stride,
563 filter_rows, filter_cols, out_depth, offset_filter, stride,
|
| conv_grad_filter_ops.cc | 135 auto filter_cols = filter.dimension(1); local
147 desc.S = filter_cols;
[all...] |
| conv_grad_input_ops.cc | 141 auto filter_cols = kernel.dimension(1); local
152 desc.S = filter_cols;
[all...] |
| conv_ops_fused.cc | 809 const int filter_cols = static_cast<int>(filter.dim_size(1)); variable
[all...] |
| mkl_conv_ops.cc | 157 const int filter_cols = static_cast<int>(filter.dim_size(1)); variable
179 GetWindowedOutputSize(input_cols, filter_cols, stride_cols,
[all...] |
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| utils_test.cc | 63 int filter_cols = 3; local
77 CreateConstOp("filter", {filter_rows, filter_cols, in_depth, out_depth},
92 std::vector<int32>({filter_rows, filter_cols, in_depth, out_depth}),
|
| /external/tensorflow/tensorflow/core/kernels/neon/ |
| neon_depthwise_conv_op.cc | 83 const int32 filter_cols = filter.dim_size(1); variable
95 GetWindowedOutputSize(input_cols, filter_cols, stride,
112 << filter_cols << ", " << in_depth << ", " << depth_multiplier
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| fused_conv2d_bias_activation_op.cc | 194 const int32 filter_cols = GetFilterDim(filter, filter_format_, 'W'); variable
205 OP_REQUIRES_OK(context, GetWindowedOutputSize(conv_input_cols, filter_cols,
216 << ", filter_cols = " << filter_cols
345 const int filter_cols = GetFilterDim(filter_param, filter_format, 'W'); local
363 0, (output_cols - 1) * col_stride + filter_cols - conv_input_cols);
370 AdjustPaddingForCudnn(padding_cols, is_int8x4, filter_cols, &padding_cols,
452 .set_input_filter_width(filter_cols)
530 {{filter_rows, filter_cols}},
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| optimized_ops.h | 588 const int filter_cols = filter_dims.sizes[0]; local
599 filter_data, output_rows, filter_cols, filter_cols);
601 input_data, filter_cols, batches, filter_cols);
881 const int filter_cols = local
1042 const int filter_cols = local
[all...] |
