This commit is contained in:
louisfd 2024-01-25 10:07:00 -05:00
parent cf24be3028
commit cf7059aab8
1 changed files with 2 additions and 2 deletions

View File

@ -224,9 +224,9 @@ impl<B: Backend> Lstm<B> {
mod tests {
use super::*;
use crate::{
module::Param, nn::LinearRecord, record::Record, TestAutodiffBackend, TestBackend,
module::Param, nn::LinearRecord, TestAutodiffBackend, TestBackend,
};
use burn_tensor::{ops::BoolTensor, Bool, Data, Distribution, Shape};
use burn_tensor::{Data, Distribution, Shape};
#[test]
fn test_with_uniform_initializer() {