Searched refs:lstm_gates (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| D | lstm_ops_gpu.cu.cc | 86 __global__ void lstm_gates(const T* icfo, const T* b, const T* cs_prev, in lstm_gates() function
266 lstm_gates<T, true>, grid_dim_2d, block_dim_2d, 0, cu_stream, in LSTMBlockCellFpropWithCUDA()
272 lstm_gates<T, false>, grid_dim_2d, block_dim_2d, 0, cu_stream, in LSTMBlockCellFpropWithCUDA()
|