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
101 r.device(d) = (r_u_bar.slice(ru_r_offset(), cell_extents())).sigmoid();
102 u.device(d) = (r_u_bar.slice(ru_u_offset(), cell_extents())).sigmoid();
163 d_r_bar_u_bar.slice(ru_r_offset(), cell_extents()).device(d) = d_r_bar;
164 d_r_bar_u_bar.slice(ru_u_offset(), cell_extents()).device(d) = d_u_bar;
lstm_ops.h 122 inline Eigen::array<Eigen::DenseIndex, 2> cell_extents() const {
249 dicfo.slice(icfo_i_offsets(), cell_extents()).device(d) = di;
250 dicfo.slice(icfo_c_offsets(), cell_extents()).device(d) = dci;
251 dicfo.slice(icfo_f_offsets(), cell_extents()).device(d) = df;
252 dicfo.slice(icfo_o_offsets(), cell_extents()).device(d) = do_;
lstm_ops.cc 77 (icfo.slice(cell.icfo_i_offsets(), cell.cell_extents()) + i_peep)
81 icfo.slice(cell.icfo_i_offsets(), cell.cell_extents()).sigmoid();
85 ci.device(d) = icfo.slice(cell.icfo_c_offsets(), cell.cell_extents()).tanh();
90 f.device(d) = (icfo.slice(cell.icfo_f_offsets(), cell.cell_extents()) +
94 f.device(d) = (icfo.slice(cell.icfo_f_offsets(), cell.cell_extents()) +
114 (icfo.slice(cell.icfo_o_offsets(), cell.cell_extents()) + o_peep)
118 icfo.slice(cell.icfo_o_offsets(), cell.cell_extents()).sigmoid();
165 dicfo.slice(cell.icfo_i_offsets(), cell.cell_extents()).device(d) = di;
166 dicfo.slice(cell.icfo_c_offsets(), cell.cell_extents()).device(d) = dci;
167 dicfo.slice(cell.icfo_f_offsets(), cell.cell_extents()).device(d) = df
    [all...]

Completed in 97 milliseconds