Searched refs:cell_size_ (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | lstm_ops.h | 108 cell_size_(cell_size) {} in LSTMBlockCell()
114 int cell_size() const { return cell_size_; } in cell_size()
122 return {0, gate_c_offset(gate_layout, cell_size_)}; in gates_c_offsets()
127 return {0, gate_f_offset(gate_layout, cell_size_)}; in gates_f_offsets()
131 return {0, cell_size_ * 3}; in gates_o_offsets()
135 return {batch_size_, cell_size_}; in cell_extents()
151 return {batch_size_, cell_size_}; in xh_h_extents()
157 const int cell_size_; member
246 Eigen::array<Eigen::DenseIndex, 2> p_shape({1, cell_size_}); in operator()
|
| D | gru_ops.h | 34 cell_size_(cell_size) {} in GRUCell()
47 return {batch_size_, cell_size_}; in h_extends()
55 return {0, cell_size_}; in ru_u_offset()
59 return {batch_size_, cell_size_}; in cell_extents()
65 const int cell_size_; member
|
| D | lstm_ops_gpu.cu.cc | 428 cell_size_, input_size_); \
454 wci_grad, wcf_grad, wco_grad, batch_size_, cell_size_, use_peephole); \
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval_test.cc | 313 std::vector<int32_t> cell_size_ = {n_batch_, n_cell_}; member in tflite::__anonb312a86f0111::BaseLstmParam
508 PackWeightToTensor(&cell_tensor_, cell_, cell_size_); in GetCell()
694 PackWeightToTensor(&cell_quantized_tensor_, cell_quantized_, cell_size_); in GetCellStateQuantized()
733 PackWeightToTensor(&cell_tensor_, cell_state_, cell_size_); in GetCell()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1073 cell_size_(cell_size), in CudnnRnnDescriptor()
1205 int cell_size() const { return cell_size_; } in cell_size()
1236 int cell_size_; member in stream_executor::gpu::CudnnRnnDescriptor
|