Searched refs:SetPaddings (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/kernels/
H A Dspace_to_batch_nd_test.cc37 void SetPaddings(std::initializer_list<int> data) { function in class:tflite::__anon25954::SpaceToBatchNDOpModel
80 // m.SetPaddings(paddings);
115 m.SetPaddings({0, 0, 0, 0});
135 m.SetPaddings({0, 0, 0, 0});
157 m.SetPaddings({1, 0, 2, 0});
182 m.SetPaddings({1, 1, 2, 4});
H A Dpad_test.cc32 void SetPaddings(std::initializer_list<int> paddings) { function in class:tflite::__anon25947::PadOpModel
71 // m.SetPaddings(paddings_data);
119 m.SetPaddings({0, 0, 1, 1, 1, 1, 0, 0});
139 m.SetPaddings({0, 0, 0, 2, 1, 3, 0, 0});

Completed in 501 milliseconds