/external/tensorflow/tensorflow/core/kernels/ |
H A D | deep_conv2d.h | 71 int in_cols; member in struct:tensorflow::Conv2DArgs 86 in_cols(0),
|
H A D | eigen_spatial_convolutions_test.cc | 669 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 D | depthwise_conv_op.h | 29 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 D | conv_ops_3d.cc | 178 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 D | pooling_ops_3d_sycl.h | 34 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 D | avgpooling_op.cc | 258 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 D | extract_image_patches_op.cc | 63 // [ 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 D | depthwise_conv_op_gpu.cu.cc | 44 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 D | fractional_avg_pool_op.cc | 255 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 D | conv_ops.cc | 168 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 D | depthwise_conv_grad_op.cc | 132 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 D | pooling_ops_common.h | 199 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 D | depthwise_conv_op.cc | 190 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 D | maxpooling_op.cc | 94 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 D | deep_conv2d.cc | 622 // [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 D | pooling_ops_3d.cc | 581 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 D | common_shapes.py | 193 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 D | thermal.py | 188 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 D | thermal.py | 188 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 D | pad_test.cc | 343 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 D | cudnn_convolution_algorithm_picker.cc | 105 int64 in_cols = local 112 std::max(in_depths, out_depths) * in_cols * in_rows *
|
/external/tensorflow/tensorflow/core/ops/ |
H A D | nn_ops.cc | 906 auto in_cols = c->Value(in_cols_dim); 921 in_cols, filter_cols_eff, stride_cols, padding, &output_cols,
|
H A D | array_ops.cc | 2338 auto in_cols = c->Value(in_cols_dim); 2349 in_cols, ksize_cols_eff, stride_cols, padding, &output_cols,
|