Home
last modified time | relevance | path

Searched refs:kKernelMap (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dbatch_matmul_test.cc81 const auto kKernelMap = new std::map<string, TfLiteRegistration*>({ variable
90 return *kKernelMap; in GetKernelMap()
319 ::testing::ValuesIn(SingleOpTest::GetKernelTags(*kKernelMap)));
446 return *kKernelMap; in GetKernelMap()
773 ::testing::ValuesIn(SingleOpTest::GetKernelTags(*kKernelMap)));
778 return *kKernelMap; in GetKernelMap()
915 ::testing::ValuesIn(SingleOpTest::GetKernelTags(*kKernelMap)));
978 return *kKernelMap; in GetKernelMap()
1036 ::testing::ValuesIn(SingleOpTest::GetKernelTags(*kKernelMap)));
H A Ddepthwise_conv_hybrid_test.cc157 const auto kKernelMap = new std::map<string, TfLiteRegistration*>({ variable
168 return *kKernelMap; in GetKernelMap()
175 return *kKernelMap; in GetKernelMap()
455 ::testing::ValuesIn(SingleOpTest::GetKernelTags(*kKernelMap)));
H A Ddepthwise_conv_test.cc148 const auto kKernelMap = new std::map<string, TfLiteRegistration*>({ variable
158 return *kKernelMap; in GetKernelMap()
1627 return *kKernelMap; in GetKernelMap()
1899 ::testing::ValuesIn(SingleOpTest::GetKernelTags(*kKernelMap)));
1908 ::testing::ValuesIn(SingleOpTest::GetKernelTags(*kKernelMap)));
H A Dtranspose_conv_test.cc124 const auto kKernelMap = new std::map<string, TfLiteRegistration*>({ variable
133 return kKernelMap->at(std::get<0>(GetParam())); in GetRegistration()
804 ::testing::ValuesIn(SingleOpTest::GetKernelTags(*kKernelMap)),
H A Dconv_test.cc148 const auto kKernelMap = new std::map<string, TfLiteRegistration*>({ variable
161 return *kKernelMap; in GetKernelMap()
2024 ::testing::ValuesIn(SingleOpTest::GetKernelTags(*kKernelMap)));
H A Dfully_connected_test.cc466 const auto kKernelMap = new std::map<string, TfLiteRegistration*>({ variable
475 return *kKernelMap; in GetKernelMap()
1477 ::testing::ValuesIn(SingleOpTest::GetKernelTags(*kKernelMap)));