Searched refs:svdf (Results 1 – 4 of 4) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | SVDFTest.cpp | 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()
[all …]
|
| /packages/modules/NeuralNetworks/common/types/operations/src/ |
| D | SVDF.cpp | 24 namespace svdf { namespace
50 NN_DEFINE_VALIDATION_FUNCTION(SVDF, svdf::validate);
|
| /packages/modules/NeuralNetworks/runtime/test/generated/spec_V1_0/ |
| D | svdf.example.cpp | 7 namespace generated_tests::svdf { namespace
124 namespace generated_tests::svdf { namespace
|
| /packages/modules/NeuralNetworks/common/ |
| D | CpuExecutor.cpp | 1090 SVDF svdf(operation, operands); in executeOperation() local
1094 setInfoAndAllocateIfNeeded(&output, outputShape, &result) && svdf.Eval(); in executeOperation()
|