Searched refs:act_id (Results 1 – 2 of 2) sorted by relevance
92 const int act_id = blockIdx.y * blockDim.y + threadIdx.y; in lstm_gates() local97 if (batch_id >= batch_size || act_id >= cell_size) return; in lstm_gates()142 const int gid = batch_id * cell_size * 4 + act_id; in lstm_gates()143 const int cid = batch_id * cell_size + act_id; in lstm_gates()150 i_local = sigmoid_op(icfo[0 * cell_size + gid] + b[0 * cell_size + act_id] + in lstm_gates()151 cs_prev[cid] * wci[act_id]); in lstm_gates()153 i_local = sigmoid_op(icfo[0 * cell_size + gid] + b[0 * cell_size + act_id]); in lstm_gates()158 tanh_op(icfo[1 * cell_size + gid] + b[1 * cell_size + act_id]); in lstm_gates()163 f_local = sigmoid_op(icfo[2 * cell_size + gid] + b[2 * cell_size + act_id] + in lstm_gates()164 forget_bias_t + cs_prev[cid] * wcf[act_id]); in lstm_gates()[all …]
84 local act_id="$1"88 if [ ${act_id} -ne ${exp_id} ]; then