HomeSort by relevance Sort by last modified time
    Searched refs:cell_extents (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/contrib/rnn/kernels/
gru_ops.h 58 inline Eigen::array<Eigen::DenseIndex, 2> cell_extents() const { function in struct:tensorflow::functor::GRUCell
99 r.device(d) = (r_u_bar.slice(ru_r_offset(), cell_extents())).sigmoid();
100 u.device(d) = (r_u_bar.slice(ru_u_offset(), cell_extents())).sigmoid();
159 d_r_bar_u_bar.slice(ru_r_offset(), cell_extents()).device(d) = d_r_bar;
160 d_r_bar_u_bar.slice(ru_u_offset(), cell_extents()).device(d) = d_u_bar;
lstm_ops.h 123 inline Eigen::array<Eigen::DenseIndex, 2> cell_extents() const {
247 dicfo.slice(icfo_i_offsets(), cell_extents()).device(d) = di;
248 dicfo.slice(icfo_c_offsets(), cell_extents()).device(d) = dci;
249 dicfo.slice(icfo_f_offsets(), cell_extents()).device(d) = df;
250 dicfo.slice(icfo_o_offsets(), cell_extents()).device(d) = do_;
lstm_ops.cc 76 (icfo.slice(cell.icfo_i_offsets(), cell.cell_extents()) + i_peep)
80 icfo.slice(cell.icfo_i_offsets(), cell.cell_extents()).sigmoid();
84 ci.device(d) = icfo.slice(cell.icfo_c_offsets(), cell.cell_extents()).tanh();
89 f.device(d) = (icfo.slice(cell.icfo_f_offsets(), cell.cell_extents()) +
93 f.device(d) = (icfo.slice(cell.icfo_f_offsets(), cell.cell_extents()) +
113 (icfo.slice(cell.icfo_o_offsets(), cell.cell_extents()) + o_peep)
117 icfo.slice(cell.icfo_o_offsets(), cell.cell_extents()).sigmoid();
164 dicfo.slice(cell.icfo_i_offsets(), cell.cell_extents()).device(d) = di;
165 dicfo.slice(cell.icfo_c_offsets(), cell.cell_extents()).device(d) = dci;
166 dicfo.slice(cell.icfo_f_offsets(), cell.cell_extents()).device(d) = df
    [all...]

Completed in 56 milliseconds