Home
last modified time | relevance | path

Searched refs:seq_length_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc157 seq_length_(max_seq_length), in CudnnRnnParameters()
186 std::to_string(seq_length_), in ToString()
202 return std::make_tuple(num_layers_, input_size_, num_units_, seq_length_, in get_data_as_tuple()
210 const int seq_length_; member in tensorflow::__anonebbbcdba0111::CudnnRnnParameters
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1785 : seq_length_(seq_length), in MIOpenRnnSequenceTensorDescriptor()
1821 int seq_length() const { return seq_length_; } in seq_length()
1826 int seq_length_; member in stream_executor::gpu::MIOpenRnnSequenceTensorDescriptor