Searched refs:o_local (Results 1 – 1 of 1) sorted by relevance
180 T o_local; in lstm_gates() local182 o_local = sigmoid_op(icfo[3 * cell_size + gid] + b[3 * cell_size + act_id] + in lstm_gates()185 o_local = sigmoid_op(icfo[3 * cell_size + gid] + b[3 * cell_size + act_id]); in lstm_gates()187 o[cid] = o_local; in lstm_gates()189 h[cid] = o_local * co_local; in lstm_gates()315 const T o_local = o[cid]; in lstm_gates_bprop() local319 const T do_local = o_local * (one - o_local) * h_grad_local * co_local; in lstm_gates_bprop()327 (one - co_local * co_local) * h_grad_local * o_local + cs_grad[cid]; in lstm_gates_bprop()