Searched refs:ConcatenationOpModel (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | concatenation_test.cc | 52 class ConcatenationOpModel : public BaseConcatenationOpModel { class
97 ConcatenationOpModel m0({TensorType_FLOAT32, {2, 1, 2}}, /*axis=*/1, in TEST()
105 ConcatenationOpModel m0({TensorType_FLOAT32, {1}}, /*axis=*/0, in TEST()
113 ConcatenationOpModel m0({TensorType_FLOAT32, {2, 3}}, /*axis=*/0, in TEST()
125 ConcatenationOpModel m0({TensorType_FLOAT32, {2, 3}}, /*axis=*/0, in TEST()
133 ConcatenationOpModel m0_negative({TensorType_FLOAT32, {2, 3}}, /*axis=*/-2, in TEST()
141 ConcatenationOpModel m1({TensorType_FLOAT32, {2, 3}}, /*axis=*/1, in TEST()
149 ConcatenationOpModel m1_negative({TensorType_FLOAT32, {2, 3}}, /*axis=*/-1, in TEST()
159 ConcatenationOpModel m0({TensorType_FLOAT32, {2, 1, 2}}, /*axis=*/2, in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 927 class ConcatenationOpModel : public BaseConcatenationOpModel { class
937 ConcatenationOpModel m0({TensorType_FLOAT32, {2, 1, 2}}, /*axis=*/1, in TEST()
945 ConcatenationOpModel m0({TensorType_FLOAT32, {2, 1, 2}}, /*axis=*/2, in TEST()
|