Searched refs:out_col (Results 1 – 2 of 2) sorted by relevance
71 Eigen::Tensor<float, 1, Eigen::RowMajor> out_col(depth); in Compare() local79 out_col(d) = in(i, d) / denom; in Compare()81 out.chip<0>(i) = out_col; in Compare()
97 const int out_col = (thread_id / out_depth) % out_width; in DepthwiseConv2dGPUKernelNHWC() local107 const int input_col_start = out_col * stride - pad_width; in DepthwiseConv2dGPUKernelNHWC()342 const int out_col = thread_id % out_width; in DepthwiseConv2dGPUKernelNCHW() local387 const int input_col_start = out_col * stride - pad_width; in DepthwiseConv2dGPUKernelNCHW()860 NOUNROLL for (int out_col = out_col_start; out_col <= out_col_end;861 ++out_col) {862 const int filter_col = in_col + pad_width - out_col * stride;867 out_depth * out_col + temp_out_backprop_offset;936 for (int out_col = out_col_start; out_col <= out_col_end; ++out_col) {937 const int filter_col = in_col + pad_width - out_col * stride;[all …]