Searched refs:compute_rnn_shape (Results 1 – 3 of 3) sorted by relevance
63 …auto shape_info = TensorInfo(misc::shape_calculator::compute_rnn_shape(recurrent_weights, hidden_s… in validate()80 …TensorShape shape = misc::shape_calculator::compute_rnn_shape(recurrent_weights->info(), hidd… in configure()
66 …auto shape_info = TensorInfo(compute_rnn_shape(recurrent_weights, hidden_state->dimension(idx_heig… in validate()91 …TensorShape shape = compute_rnn_shape(recurrent_weights->info(), hidden_state->info()->dimens… in configure()
888 inline TensorShape compute_rnn_shape(const ITensorInfo *input, const unsigned int batch_size) in compute_rnn_shape() function