Home | History | Annotate | Download | only in kernels

Lines Matching defs:output_cols

43                                int output_cols, int stride_rows,
47 // out_idx = d + depth * (w_out + output_cols * (h_out + output_rows * b))
50 const int w_out = out_idx2 % output_cols;
51 const int out_idx3 = out_idx2 / output_cols;
83 int output_cols, int stride_rows, int stride_cols, int rate_rows,
86 // out_idx = d + depth * (w_out + output_cols * (h_out + output_rows * b))
89 const int w_out = out_idx2 % output_cols;
90 const int out_idx3 = out_idx2 / output_cols;
132 int output_cols, int stride_rows, int stride_cols, int rate_rows,
135 // out_idx = d + depth * (w_out + output_cols * (h_out + output_rows * b))
138 const int w_out = out_idx2 % output_cols;
139 const int out_idx3 = out_idx2 / output_cols;
194 const int output_cols = output.dimension(2);
196 const int total_count = batch * output_rows * output_cols * depth;
203 output_cols, stride_rows, stride_cols, rate_rows, rate_cols, pad_top,
225 const int output_cols = out_backprop.dimension(2);
237 total_count = batch * output_rows * output_cols * depth;
243 filter_cols, output_rows, output_cols, stride_rows, stride_cols,
265 const int output_cols = out_backprop.dimension(2);
277 total_count = batch * output_rows * output_cols * depth;
283 filter_cols, output_rows, output_cols, stride_rows, stride_cols,