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

  /external/tensorflow/tensorflow/contrib/rnn/kernels/
lstm_ops.h 223 typename TTypes<T>::Matrix xh_grad, typename TTypes<T>::Matrix x_grad,
262 // xh_grad.
266 ctx, d, false, true, 1.f, const_dicfo, w, 0.f, xh_grad);
274 x_grad.device(d) = xh_grad.slice(xh_x_offsets(), xh_x_extents());
275 h_prev_grad.device(d) = xh_grad.slice(xh_h_offsets(), xh_h_extents());
lstm_ops.cc     [all...]
  /external/tensorflow/tensorflow/contrib/rnn/python/ops/
lstm_ops.py 276 xh_grad = math_ops.matmul(dicfo, w, transpose_b=True)
278 x_grad = array_ops.slice(xh_grad, (0, 0), (batch_size, input_size))
281 h_prev_grad = array_ops.slice(xh_grad, (0, input_size),

Completed in 97 milliseconds