Lines Matching refs:svdf
315 SVDFOpModel svdf(/*batches=*/2, /*units=*/4, /*input_size=*/3, in TEST() local
317 svdf.SetWeightsFeature({-0.31930989, -0.36118156, 0.0079667, 0.37613347, 0.22197971, 0.12416199, in TEST()
321 svdf.SetWeightsTime({-0.31930989, 0.37613347, 0.27901134, -0.36137494, -0.36118156, in TEST()
333 svdf.SetBias({}); in TEST()
335 svdf.ResetState(); in TEST()
336 const int svdf_num_batches = svdf.num_batches(); in TEST()
337 const int svdf_input_size = svdf.input_size(); in TEST()
338 const int svdf_num_units = svdf.num_units(); in TEST()
346 svdf.SetInput(0, batch_start, batch_end); in TEST()
348 svdf.Invoke(); in TEST()
355 EXPECT_THAT(svdf.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
360 SVDFOpModel svdf(/*batches=*/2, /*units=*/4, /*input_size=*/3, in TEST() local
362 svdf.SetWeightsFeature({-0.31930989, 0.0079667, 0.39296314, 0.37613347, 0.12416199, in TEST()
368 svdf.SetWeightsTime({-0.31930989, 0.37613347, 0.27901134, -0.36137494, -0.36118156, in TEST()
392 svdf.SetBias({}); in TEST()
394 svdf.ResetState(); in TEST()
395 const int svdf_num_batches = svdf.num_batches(); in TEST()
396 const int svdf_input_size = svdf.input_size(); in TEST()
397 const int svdf_num_units = svdf.num_units(); in TEST()
405 svdf.SetInput(0, batch_start, batch_end); in TEST()
407 svdf.Invoke(); in TEST()
414 EXPECT_THAT(svdf.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()