HomeSort by relevance Sort by last modified time
    Searched full:h_out (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
dilation_ops_gpu.cu.cc 47 // out_idx = d + depth * (w_out + output_cols * (h_out + output_rows * b))
52 const int h_out = out_idx3 % output_rows; local
54 int h_beg = h_out * stride_rows - pad_top;
86 // out_idx = d + depth * (w_out + output_cols * (h_out + output_rows * b))
91 const int h_out = out_idx3 % output_rows; local
93 int h_beg = h_out * stride_rows - pad_top;
135 // out_idx = d + depth * (w_out + output_cols * (h_out + output_rows * b))
140 const int h_out = out_idx3 % output_rows; local
142 int h_beg = h_out * stride_rows - pad_top;
dilation_ops.cc 182 for (int h_out = 0; h_out < output_rows; ++h_out) {
183 int h_beg = h_out * stride_rows - pad_top;
202 output(b, h_out, w_out, d) = cur_val;
297 for (int h_out = 0; h_out < output_rows; ++h_out) {
298 int h_beg = h_out * stride_rows - pad_top;
322 out_backprop(b, h_out, w_out, d)
    [all...]
  /external/tensorflow/tensorflow/contrib/rnn/kernels/
lstm_ops.cc 935 Tensor* h_out; variable
1106 const Tensor* h_out = nullptr; variable
    [all...]

Completed in 1471 milliseconds