Home
last modified time | relevance | path

Searched refs:dicfo (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/rnn/kernels/
Dlstm_ops.h195 typename TTypes<T>::Matrix dicfo, typename TTypes<T>::Matrix cs_prev_grad,
221 typename TTypes<T>::Matrix dicfo, typename TTypes<T>::Matrix cs_prev_grad, in operator()
249 dicfo.slice(icfo_i_offsets(), cell_extents()).device(d) = di; in operator()
250 dicfo.slice(icfo_c_offsets(), cell_extents()).device(d) = dci; in operator()
251 dicfo.slice(icfo_f_offsets(), cell_extents()).device(d) = df; in operator()
252 dicfo.slice(icfo_o_offsets(), cell_extents()).device(d) = do_; in operator()
263 typename TTypes<T>::ConstMatrix const_dicfo(dicfo.data(), in operator()
264 dicfo.dimensions()); in operator()
282 b_grad.device(d) += dicfo.sum(Eigen::array<int, 1>({0})); in operator()
Dlstm_ops_gpu.cu.cc301 T* dicfo, // [input_size + cell_size, 4 * cell_size] in lstm_gates_bprop() argument
345 dicfo[gid + 0 * cell_size] = di_local; in lstm_gates_bprop()
346 dicfo[gid + 1 * cell_size] = dci_local; in lstm_gates_bprop()
347 dicfo[gid + 2 * cell_size] = df_local; in lstm_gates_bprop()
348 dicfo[gid + 3 * cell_size] = do_local; in lstm_gates_bprop()
370 typename TTypes<T>::Matrix dicfo, typename TTypes<T>::Matrix cs_prev_grad, in LSTMBlockCellBpropWithCUDA() argument
385 dci.data(), df.data(), di.data(), dicfo.data(), cs_prev_grad.data(), in LSTMBlockCellBpropWithCUDA()
444 typename TTypes<T>::Matrix dicfo, \
450 cs_grad, h_grad, do_, dcs, dci, df, di, dicfo, cs_prev_grad, wci_grad, \
Dlstm_ops.cc140 typename TTypes<T>::Matrix dicfo, typename TTypes<T>::Matrix cs_prev_grad, in LSTMBlockCellBpropWithEigen() argument
165 dicfo.slice(cell.icfo_i_offsets(), cell.cell_extents()).device(d) = di; in LSTMBlockCellBpropWithEigen()
166 dicfo.slice(cell.icfo_c_offsets(), cell.cell_extents()).device(d) = dci; in LSTMBlockCellBpropWithEigen()
167 dicfo.slice(cell.icfo_f_offsets(), cell.cell_extents()).device(d) = df; in LSTMBlockCellBpropWithEigen()
168 dicfo.slice(cell.icfo_o_offsets(), cell.cell_extents()).device(d) = do_; in LSTMBlockCellBpropWithEigen()
216 typename TTypes<T>::Matrix dicfo, \
222 i, cs, f, o, ci, co, cs_grad, h_grad, do_, dcs, dci, df, di, dicfo, \
687 typename TTypes<T>::Matrix dicfo, \
1329 typename TTypes<T>::Matrix dicfo, \
/external/tensorflow/tensorflow/contrib/rnn/python/ops/
Dlstm_ops.py255 (cs_prev_grad, dicfo, wci_grad, wcf_grad,
276 xh_grad = math_ops.matmul(dicfo, w, transpose_b=True)
287 w_grad = math_ops.matmul(xh, dicfo, transpose_a=True)
291 b_grad = nn_ops.bias_add_grad(dicfo)