Lines Matching refs:svdf
317 SVDFOpModel svdf(/*batches=*/2, /*units=*/4, /*input_size=*/3, in TEST() local
319 svdf.SetWeightsFeature({-0.31930989, -0.36118156, 0.0079667, 0.37613347, 0.22197971, 0.12416199, in TEST()
323 svdf.SetWeightsTime({-0.31930989, 0.37613347, 0.27901134, -0.36137494, -0.36118156, in TEST()
335 svdf.SetBias({}); in TEST()
337 svdf.ResetState(); in TEST()
338 const int svdf_num_batches = svdf.num_batches(); in TEST()
339 const int svdf_input_size = svdf.input_size(); in TEST()
340 const int svdf_num_units = svdf.num_units(); in TEST()
348 svdf.SetInput(0, batch_start, batch_end); in TEST()
350 svdf.Invoke(); in TEST()
357 EXPECT_THAT(svdf.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()
362 SVDFOpModel svdf(/*batches=*/2, /*units=*/4, /*input_size=*/3, in TEST() local
364 svdf.SetWeightsFeature({-0.31930989, 0.0079667, 0.39296314, 0.37613347, 0.12416199, in TEST()
370 svdf.SetWeightsTime({-0.31930989, 0.37613347, 0.27901134, -0.36137494, -0.36118156, in TEST()
394 svdf.SetBias({}); in TEST()
396 svdf.ResetState(); in TEST()
397 const int svdf_num_batches = svdf.num_batches(); in TEST()
398 const int svdf_input_size = svdf.input_size(); in TEST()
399 const int svdf_num_units = svdf.num_units(); in TEST()
407 svdf.SetInput(0, batch_start, batch_end); in TEST()
409 svdf.Invoke(); in TEST()
416 EXPECT_THAT(svdf.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()