Searched refs:SetPaddings (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| D | space_to_batch_nd_test.cc | 37 void SetPaddings(std::initializer_list<int> data) { in SetPaddings() function in tflite::__anon09a4ca450111::SpaceToBatchNDOpModel
115 m.SetPaddings({0, 0, 0, 0}); in TEST()
135 m.SetPaddings({0, 0, 0, 0}); in TEST()
157 m.SetPaddings({1, 0, 2, 0}); in TEST()
182 m.SetPaddings({1, 1, 2, 4}); in TEST()
|
| D | pad_test.cc | 32 void SetPaddings(std::initializer_list<int> paddings) { in SetPaddings() function in tflite::__anond79f939a0111::PadOpModel
119 m.SetPaddings({0, 0, 1, 1, 1, 1, 0, 0}); in TEST()
139 m.SetPaddings({0, 0, 0, 2, 1, 3, 0, 0}); in TEST()
|