Home
last modified time | relevance | path

Searched refs:SetBlockShape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dspace_to_batch_nd_test.cc39 void SetBlockShape(std::initializer_list<int> data) { in SetBlockShape() function in tflite::__anonf57345680111::SpaceToBatchNDOpModel
134 m.SetBlockShape({2, 2}); in TEST()
156 m.SetBlockShape({2, 2}); in TEST()
180 m.SetBlockShape({3, 2}); in TEST()
207 m.SetBlockShape({3, 2}); in TEST()
273 m.SetBlockShape({3, 2}); in TEST_F()
289 m.SetBlockShape({3, 2}); in TEST_F()
321 m.SetBlockShape({3, 2}); in TEST_F()
Dbatch_to_space_nd_test.cc34 void SetBlockShape(std::initializer_list<int> data) { in SetBlockShape() function in tflite::__anon9f6216280111::BatchToSpaceNDOpModel
127 m.SetBlockShape({2, 2}); in TEST()
139 m.SetBlockShape({2, 2}); in TEST()
162 m.SetBlockShape({2, 2}); in TEST()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc1599 void SetBlockShape(std::initializer_list<int> data) { in SetBlockShape() function in tflite::__anon0c68fcb70111::SpaceToBatchNDOpModel