Home
last modified time | relevance | path

Searched refs:StridedSliceOpModel (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dstrided_slice_test.cc28 class StridedSliceOpModel : public SingleOpModel { class
30 StridedSliceOpModel(std::initializer_list<int> input_shape, in StridedSliceOpModel() function in tflite::__anone1578cc60111::StridedSliceOpModel
78 StridedSliceOpModel<>({2, 2, 2, 2, 2}, {5}, {5}, {5}, 0, 0, 0, 0, 0), in TEST()
83 EXPECT_DEATH(StridedSliceOpModel<>({3, 2}, {2}, {2}, {2}, 0, 0, 1, 0, 0), in TEST()
85 EXPECT_DEATH(StridedSliceOpModel<>({3, 2}, {2}, {2}, {2}, 0, 0, 0, 1, 0), in TEST()
91 StridedSliceOpModel<> m({4}, {1}, {1}, {1}, 0, 0, 0, 0, 0); in TEST()
102 StridedSliceOpModel<> m({4}, {1}, {1}, {1}, 0, 0, 0, 0, 0); in TEST()
112 StridedSliceOpModel<> m({4}, {1}, {1}, {1}, 0, 0, 0, 0, 0); in TEST()
123 StridedSliceOpModel<> m({4}, {1}, {1}, {1}, 0, 0, 0, 0, 0); in TEST()
134 StridedSliceOpModel<> m({4}, {1}, {1}, {1}, 0, 0, 0, 0, 0); in TEST()
[all …]
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc1677 class StridedSliceOpModel : public SingleOpModelWithNNAPI { class
1679 StridedSliceOpModel(std::initializer_list<int> input_shape, in StridedSliceOpModel() function in tflite::__anon0c68fcb70111::StridedSliceOpModel
1719 StridedSliceOpModel<> m({4}, {1}, {1}, {1}, {3}, {1}, {1}, 0, 0, 0, 0, 0); in TEST()
1727 StridedSliceOpModel<> m({4}, {1}, {1}, {1}, {3}, {1}, {1}, 1, 0, 0, 0, 0); in TEST()
1735 StridedSliceOpModel<> m({2, 3}, {2}, {0, 0}, {2}, {2, 3}, {2}, {2, 2}, 0, 0, in TEST()
1744 StridedSliceOpModel<> m({2, 3}, {2}, {1, 0}, {2}, {2, 2}, {2}, {1, 1}, 0, 2, in TEST()
1753 StridedSliceOpModel<> m({2, 3, 2}, {3}, {0, 0, 0}, {3}, {2, 3, 1}, {3}, in TEST()