Searched refs:svdf (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | svdf_test.cc | 235 int golden_size, BaseSVDFOpModel* svdf, in VerifyGoldens() argument
237 const int svdf_num_batches = svdf->num_batches(); in VerifyGoldens()
238 const int svdf_input_size = svdf->input_size(); in VerifyGoldens()
239 const int svdf_num_units = svdf->num_units(); in VerifyGoldens()
248 svdf->SetInput(0, batch_start, batch_end); in VerifyGoldens()
250 svdf->Invoke(); in VerifyGoldens()
259 EXPECT_THAT(svdf->GetOutput(), in VerifyGoldens()
266 SVDFOpModel svdf(/*batches=*/2, /*units=*/4, /*input_size=*/3, in TEST_F() local
268 svdf.SetWeightsFeature({-0.31930989, -0.36118156, 0.0079667, 0.37613347, in TEST_F()
272 svdf.SetWeightsTime( in TEST_F()
[all …]
|
| D | svdf.cc | 36 namespace svdf { namespace
430 static TfLiteRegistration r = {svdf::Init, svdf::Free, svdf::Prepare, in Register_SVDF()
431 svdf::Eval}; in Register_SVDF()
|
| D | Android.bp | 110 "svdf.cc",
|
| D | BUILD | 222 "svdf.cc",
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 2164 int golden_size, BaseSVDFOpModel* svdf, in VerifyGoldens() argument
2166 const int svdf_num_batches = svdf->num_batches(); in VerifyGoldens()
2167 const int svdf_input_size = svdf->input_size(); in VerifyGoldens()
2168 const int svdf_num_units = svdf->num_units(); in VerifyGoldens()
2177 svdf->SetInput(0, batch_start, batch_end); in VerifyGoldens()
2179 svdf->Invoke(); in VerifyGoldens()
2188 EXPECT_THAT(svdf->GetOutput(), in VerifyGoldens()
2195 SVDFOpModel svdf(/*batches=*/2, /*units=*/4, /*input_size=*/3, in TEST_F() local
2197 svdf.SetWeightsFeature({-0.31930989, -0.36118156, 0.0079667, 0.37613347, in TEST_F()
2201 svdf.SetWeightsTime( in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/lite/ |
| D | tflite_static.bp | 116 "kernels/svdf.cc",
|