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

  /external/tensorflow/tensorflow/contrib/rnn/kernels/
gru_ops.h 34 cell_size_(cell_size) {}
47 return {batch_size_, cell_size_};
55 return {0, cell_size_};
59 return {batch_size_, cell_size_};
65 const int cell_size_; member in struct:tensorflow::functor::GRUCell
lstm_ops.h 99 cell_size_(cell_size) {}
105 int cell_size() const { return cell_size_; }
112 return {0, cell_size_};
116 return {0, cell_size_ * 2};
120 return {0, cell_size_ * 3};
124 return {batch_size_, cell_size_};
140 return {batch_size_, cell_size_};
146 const int cell_size_;
231 Eigen::array<Eigen::DenseIndex, 2> p_shape({1, cell_size_});
lstm_ops_gpu.cu.cc 374 cell_size_, input_size_); \
399 wcf_grad, wco_grad, batch_size_, cell_size_, use_peephole); \

Completed in 633 milliseconds