Searched refs:output_weights_shape (Results 1 – 1 of 1) sorted by relevance
1124 const auto& output_weights_shape = recurrent_to_output_weights_array.shape(); in ProcessUnidirectionalSequenceLstmOperator() local1125 const int output_size = output_weights_shape.dims(1); in ProcessUnidirectionalSequenceLstmOperator()1218 const auto& output_weights_shape = recurrent_to_output_weights_array.shape(); in ProcessBidirectionalSequenceLstmOperator() local1219 const int output_size = output_weights_shape.dims(1); in ProcessBidirectionalSequenceLstmOperator()1270 const auto& output_weights_shape = forward_weights_array.shape(); in ProcessBidirectionalSequenceRnnOperator() local1271 const int output_size = output_weights_shape.dims(0); in ProcessBidirectionalSequenceRnnOperator()