Searched refs:svdf (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | SVDFTest.cpp | 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()
[all …]
|
| /frameworks/ml/nn/runtime/test/generated/spec_V1_0/ |
| D | svdf.example.cpp | 7 namespace generated_tests::svdf { namespace
124 namespace generated_tests::svdf { namespace
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 1145 SVDF svdf(operation, operands); in executeOperation() local
1149 setInfoAndAllocateIfNeeded(&output, outputShape, &result) && svdf.Eval(); in executeOperation()
|