| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | strided_slice_test.cc | 32 class StridedSliceOpModel : public SingleOpModel { class
34 StridedSliceOpModel(std::initializer_list<int> input_shape, in StridedSliceOpModel() function in tflite::__anon7e1c7ebc0111::StridedSliceOpModel
96 EXPECT_DEATH(StridedSliceOpModel<TypeParam>({2, 2, 2, 2, 2, 2}, {5}, {5}, {5}, in TYPED_TEST()
103 StridedSliceOpModel<TypeParam> m({0}, {1}, {1}, {1}, 0, 0, 0, 0, 0); in TYPED_TEST()
112 StridedSliceOpModel<TypeParam> m({4}, {1}, {1}, {1}, 0, 0, 0, 0, 0); in TYPED_TEST()
123 StridedSliceOpModel<TypeParam> m({32768}, {1}, {1}, {1}, 0, 0, 0, 0, 0); in TYPED_TEST()
138 StridedSliceOpModel<TypeParam> m({4}, {1}, {1}, {1}, 0, 0, 0, 0, 0); in TYPED_TEST()
148 StridedSliceOpModel<TypeParam> m({4}, {1}, {1}, {1}, 0, 0, 0, 0, 0); in TYPED_TEST()
159 StridedSliceOpModel<TypeParam> m({4}, {1}, {1}, {1}, 0, 0, 0, 0, 0); in TYPED_TEST()
170 StridedSliceOpModel<TypeParam> m({4}, {1}, {1}, {1}, 0, 0, 0, 0, 0); in TYPED_TEST()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| H A D | strided_slice_test.cc | 24 class StridedSliceOpModel : public SingleOpModelWithHexagon { class
26 StridedSliceOpModel(const TensorData& input, const TensorData& output, in StridedSliceOpModel() function in tflite::StridedSliceOpModel
66 TEST(StridedSliceOpModel, In1D_UInt8) { in TEST() argument
67 StridedSliceOpModel<uint8_t> m( in TEST()
83 TEST(StridedSliceOpModel, In1D_NegativeBegin_Int8) { in TEST() argument
84 StridedSliceOpModel<int8_t> m( in TEST()
100 TEST(StridedSliceOpModel, In1D_NegativeEnd_Int8) { in TEST() argument
101 StridedSliceOpModel<int8_t> m( in TEST()
117 TEST(StridedSliceOpModel, In2D_MultipleStrides_UInt8) { in TEST() argument
118 StridedSliceOpModel<uint8_t> m( in TEST()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| H A D | nnapi_delegate_test.cc | 2585 class StridedSliceOpModel : public SingleOpModelWithNNAPI { class
2587 StridedSliceOpModel(std::initializer_list<int> input_shape, in StridedSliceOpModel() function in tflite::__anon1a74836d0111::StridedSliceOpModel
2628 StridedSliceOpModel<> m({4}, {1}, {1}, {1}, {3}, {1}, {1}, 0, 0, 0, 0, 0); in TEST()
2636 StridedSliceOpModel<> m({4}, {1}, {1}, {1}, {3}, {1}, {1}, 1, 0, 0, 0, 0); in TEST()
2644 StridedSliceOpModel<> m({2, 3}, {2}, {0, 0}, {2}, {2, 3}, {2}, {2, 2}, 0, 0, in TEST()
2653 StridedSliceOpModel<> m({2, 3}, {2}, {1, 0}, {2}, {2, 2}, {2}, {1, 1}, 0, 2, in TEST()
2662 StridedSliceOpModel<> m({2, 3, 2}, {3}, {0, 0, 0}, {3}, {2, 3, 1}, {3}, in TEST()
|