Searched refs:in_cols (Results 1 - 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Ddeep_conv2d.h71 int in_cols; member in struct:tensorflow::Conv2DArgs
86 in_cols(0),
H A Deigen_spatial_convolutions_test.cc669 const int in_cols = 7; local
678 const int out_width = in_cols;
680 Tensor<float, 4> input(in_channels, in_depth, in_rows, in_cols);
710 r - off_r + j < in_rows && c - off_c + k < in_cols) {
730 const int in_cols = 7; local
739 const int out_width = in_cols;
741 Tensor<float, 4, RowMajor> input(in_cols, in_rows, in_depth, in_channels);
772 r - off_r + j < in_rows && c - off_c + k < in_cols) {
792 const int in_cols = 5; local
803 Tensor<float, 4> input(in_channels, in_depth, in_rows, in_cols);
844 const int in_cols = 5; local
898 const int in_cols = 7; local
964 const int in_cols = 7; local
1030 const int in_cols = 5; local
1085 const int in_cols = 5; local
1141 const int in_cols = 5; local
1211 const int in_cols = 5; local
[all...]
H A Ddepthwise_conv_op.h29 int in_cols; member in struct:tensorflow::DepthwiseArgs
46 in_cols(0),
177 // input: [batch, in_rows, in_cols, in_depth]
231 in_c < args.in_cols) {
232 auto* in = input + (in_r * args.in_cols + in_c) * args.in_depth;
H A Dconv_ops_3d.cc178 int64 in_cols = GetTensorDim(input, data_format, '2'); local
197 0, (out_cols - 1) * strides[2] + filter_cols - in_cols);
205 const uint64 m = in_batch * in_planes * in_rows * in_cols;
228 filter_cols == in_cols && padding == Padding::VALID &&
233 const uint64 k = in_planes * in_rows * in_cols * in_depth;
267 const int64 new_in_cols = in_cols + cols_odd;
284 in_cols = new_in_cols;
291 FORMAT_NCHW, in_batch, {{in_planes, in_rows, in_cols}}, in_depth);
315 .set_spatial_dim(DimIndex::X, in_cols)
377 {{in_planes, in_rows, in_cols}},
[all...]
H A Dpooling_ops_3d_sycl.h34 const int in_rows, const int in_cols, const int out_planes,
43 in_cols_(in_cols),
58 const int in_rows, const int in_cols,
63 : SYCL3DPoolParams(depth, batch, in_planes, in_rows, in_cols,
125 const int in_rows, const int in_cols, const int out_planes,
132 : p_(depth, batch, in_planes, in_rows, in_cols, out_planes, out_rows,
192 const int in_cols = GetTensorDim(tensor_in, data_format, '2'); local
207 MaxPool3DSYCL<T> max_pool(depth, batch, in_planes, in_rows, in_cols,
236 const int in_rows, const int in_cols,
245 : p_(depth, batch, in_planes, in_rows, in_cols, output_shap
33 SYCL3DPoolParams(const int depth, const int batch, const int in_planes, const int in_rows, const int in_cols, const int out_planes, const int out_rows, const int out_cols, const std::array<int64, 3>& window, const std::array<int64, 3>& stride, const std::array<int64, 3>& padding) argument
57 SYCL3DPoolParams(const int depth, const int batch, const int in_planes, const int in_rows, const int in_cols, const std::array<int64, 3>& out_shape, const std::array<int64, 3>& window, const std::array<int64, 3>& stride, const std::array<int64, 3>& padding) argument
124 MaxPool3DSYCL(const int depth, const int batch, const int in_planes, const int in_rows, const int in_cols, const int out_planes, const int out_rows, const int out_cols, const std::array<int64, 3>& window, const std::array<int64, 3>& stride, const std::array<int64, 3>& padding, const read_accessor input_accessor, write_accessor output_accessor) argument
235 MaxPool3DGradSYCL(const int depth, const int batch, const int in_planes, const int in_rows, const int in_cols, const std::array<int64, 3>& output_shape, const std::array<int64, 3>& window, const std::array<int64, 3>& stride, const std::array<int64, 3>& padding, const read_accessor input_data_accessor, const read_accessor output_data_accessor, const read_accessor input_backprop_accessor, write_accessor output_backprop_accessor) argument
358 const int in_cols = GetTensorDim(tensor_in, data_format, '2'); local
530 AvgPool3DSYCL(const int depth, const int batch, const int in_planes, const int in_rows, const int in_cols, const int out_planes, const int out_rows, const int out_cols, const std::array<int64, 3>& window, const std::array<int64, 3>& stride, const std::array<int64, 3>& padding, const read_accessor input_accessor, write_accessor output_accessor) argument
598 const int in_cols = GetTensorDim(tensor_in, data_format, '2'); local
641 AvgPool3DGradSYCL(const int depth, const int batch, const int in_planes, const int in_rows, const int in_cols, const std::array<int64, 3>& out_shape, const std::array<int64, 3>& window, const std::array<int64, 3>& stride, const std::array<int64, 3>& padding, const read_accessor input_backprop_accessor, write_accessor output_backprop_accessor) argument
730 const int in_cols = GetTensorDim(tensor_in_shape, data_format, '2'); local
[all...]
H A Davgpooling_op.cc258 const int64 in_cols = output_shape.dim_size(2); variable
284 GetWindowedOutputSize(in_cols, window_cols, col_stride,
292 in_rows, in_cols, window_rows, window_cols, row_stride,
313 GetBroadcastSize(c, in_cols, window_cols, col_stride,
321 int64 input_index = (b * in_rows + r_dst) * in_cols + c_dst;
341 window_rows * window_cols * depth_window * in_rows * in_rows * in_cols;
493 const int64 in_cols = output_shape.dim_size(2); variable
518 GetWindowedOutputSize(in_cols, window_cols, col_stride,
524 in_cols, // width
H A Dextract_image_patches_op.cc63 // [ batch, in_rows, in_cols, channels ]
71 const int in_cols = input.dim_size(2); variable
92 GetWindowedOutputSize(in_cols, ksize_cols_eff, stride_cols,
H A Ddepthwise_conv_op_gpu.cu.cc44 args.in_cols <= 32 && args.in_rows == args.out_rows &&
45 args.in_cols == args.out_cols && args.pad_rows >= 0 &&
49 (args.in_rows + 1) / 2 * args.in_cols;
57 args.in_cols <= 32 && args.in_rows == args.out_rows &&
58 args.in_cols == args.out_cols && args.pad_rows >= 0 &&
61 args.filter_rows * args.filter_cols <= args.in_cols * block_height;
76 const int in_width = args.in_cols;
180 const int in_width = args.in_cols;
190 assert(blockDim.y == args.in_cols);
312 const int in_width = args.in_cols;
[all...]
H A Dfractional_avg_pool_op.cc255 const int64 in_cols = orig_input_tensor_shape_flat(2); variable
274 in_cols * in_rows * in_batch);
281 const int64 in_max_col_index = in_cols - 1;
301 const int64 in_index = (b * in_rows + in_r) * in_cols + in_c;
H A Dconv_ops.cc168 args.in_cols = input_cols;
303 // [ batch, in_rows, in_cols, in_depth ]
562 int64 in_cols = GetTensorDim(input, data_format, 'W'); local
584 (patch_cols - 1) * col_dilation + 1 - in_cols);
590 int64 new_in_cols = in_cols + cols_odd;
605 in_cols = new_in_cols;
612 ShapeFromFormat(FORMAT_NCHW, in_batch, in_rows, in_cols, in_depths);
635 .set_width(in_cols)
694 in_cols}}, // in_cols
[all...]
H A Ddepthwise_conv_grad_op.cc132 args.in_cols = input_cols; \
291 const int64 base_output_index = (in_r * args.in_cols + in_c) * in_depth;
407 args.in_rows * args.in_cols * args.in_depth;
433 for (int64 in_c = 0; in_c < args.in_cols; ++in_c) {
449 const int64 shard_cost = args.in_rows * args.in_cols * args.out_depth;
464 for (int in_c = 0; in_c < args.in_cols; ++in_c) {
504 args.in_depth * (in_c + args.in_cols * (in_r + args.in_rows * b));
769 args.in_rows * args.in_cols * args.in_depth;
864 in_c < args.in_cols) {
872 (in_c + args.in_cols * (in_
[all...]
H A Dpooling_ops_common.h199 const int32 in_cols = params.tensor_in_cols;
220 for (int32 w = 0; w < in_cols; ++w) {
234 const int32 in_offset = (b * in_rows + h) * in_cols + w;
446 const int32 in_cols = params.tensor_in_cols;
467 for (int32 w = 0; w < in_cols; ++w) {
481 const int32 in_offset = (b * in_rows + h) * in_cols + w;
H A Ddepthwise_conv_op.cc190 args.in_rows * args.in_cols * args.in_depth;
294 // [ batch, in_rows, in_cols, in_depth ]
387 args.in_cols = input_cols;
H A Dmaxpooling_op.cc94 const int32 in_cols = params.tensor_in_cols;
118 for (int w = 0; w < in_cols; ++w) {
130 const int64 in_index = (b * in_rows + h) * in_cols + w;
160 const int64 in_size = in_rows * in_cols * depth;
571 const int32 in_cols = params.tensor_in_cols;
598 const int w_end = std::min(w_start + window_cols, in_cols);
608 const int in_index = (b * in_rows + h) * in_cols + w;
H A Ddeep_conv2d.cc622 // [in_rows, in_cols, in_depth]
651 if (in_c < 0 || in_c >= args.in_cols) continue;
653 auto* in = input + (in_r * args.in_cols + in_c) * args.in_depth;
675 // [in_rows, in_cols, in_depth]
1113 const int64 input_image_size = args.in_rows * args.in_cols * in_depth;
H A Dpooling_ops_3d.cc581 const int32 in_cols = params.tensor_in_cols;
616 const int w_end = std::min(w_start + window_cols, in_cols);
630 ((b * in_planes + p) * in_rows + h) * in_cols + w;
/external/tensorflow/tensorflow/python/framework/
H A Dcommon_shapes.py193 in_cols = input_shape[2]
213 out_rows, out_cols = get2d_conv_output_size(in_rows, in_cols, filter_rows,
252 in_cols = input_shape[2]
274 out_rows, out_cols = get2d_conv_output_size(in_rows, in_cols, filter_rows,
317 in_cols = input_shape[2]
337 out_rows, out_cols = get2d_conv_output_size(in_rows, in_cols, filter_rows,
385 in_cols = input_shape[2]
400 out_rows, out_cols = get2d_conv_output_size(in_rows, in_cols, ksize_r,
453 in_cols = input_shape[2]
472 out_rows, out_cols = get2d_conv_output_size(in_rows, in_cols, ksize_
[all...]
/external/trappy/tests/trappy/
H A Dthermal.py188 in_cols = [s for s in dfr.columns if re.match("req_power[0-9]+", s)]
190 plot_dfr = dfr[in_cols]
230 in_cols = [s for s in dfr.columns if re.match(r"req_power\d+", s)]
233 for in_col, (name, weight) in zip(in_cols, actor_weights):
/external/trappy/trappy/
H A Dthermal.py188 in_cols = [s for s in dfr.columns if re.match("req_power[0-9]+", s)]
190 plot_dfr = dfr[in_cols]
230 in_cols = [s for s in dfr.columns if re.match(r"req_power\d+", s)]
233 for in_col, (name, weight) in zip(in_cols, actor_weights):
/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dpad_test.cc343 constexpr int64 in_cols = 35; local
344 auto operand = MakeUnique<Array2D<float>>(in_rows, in_cols);
366 constexpr int64 in_cols = 129; local
370 auto operand = MakeUnique<Array2D<float>>(in_rows, in_cols);
393 constexpr int64 in_cols = 129; local
397 auto operand = MakeUnique<Array2D<float>>(in_rows, in_cols);
421 constexpr int64 in_cols = 11; local
425 auto operand = MakeUnique<Array2D<float>>(in_rows, in_cols);
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dcudnn_convolution_algorithm_picker.cc105 int64 in_cols = local
112 std::max(in_depths, out_depths) * in_cols * in_rows *
/external/tensorflow/tensorflow/core/ops/
H A Dnn_ops.cc906 auto in_cols = c->Value(in_cols_dim);
921 in_cols, filter_cols_eff, stride_cols, padding, &output_cols,
H A Darray_ops.cc2338 auto in_cols = c->Value(in_cols_dim);
2349 in_cols, ksize_cols_eff, stride_cols, padding, &output_cols,

Completed in 662 milliseconds