Searched refs:block_dim_2d (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| D | lstm_ops_gpu.cu.cc | 260 dim3 block_dim_2d(std::min(batch_size, 8), 32); in LSTMBlockCellFpropWithCUDA() local
261 dim3 grid_dim_2d(Eigen::divup(batch_size, static_cast<int>(block_dim_2d.x)), in LSTMBlockCellFpropWithCUDA()
262 Eigen::divup(cell_size, static_cast<int>(block_dim_2d.y))); in LSTMBlockCellFpropWithCUDA()
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()
376 dim3 block_dim_2d(std::min(batch_size, 8), 32); in LSTMBlockCellBpropWithCUDA() local
377 dim3 grid_dim_2d(Eigen::divup(batch_size, static_cast<int>(block_dim_2d.x)), in LSTMBlockCellBpropWithCUDA()
378 Eigen::divup(cell_size, static_cast<int>(block_dim_2d.y))); in LSTMBlockCellBpropWithCUDA()
381 lstm_gates_bprop<T>, grid_dim_2d, block_dim_2d, 0, cu_stream, in LSTMBlockCellBpropWithCUDA()
|