D | reduce_test.cc | 58 class MeanOpConstModel : public BaseOpModel { class 60 MeanOpConstModel(const TensorData& input, const TensorData& output, in MeanOpConstModel() function in tflite::__anon5bc6b6600111::MeanOpConstModel 239 MeanOpConstModel m({TensorType_FLOAT32, {4, 3, 2}}, {TensorType_FLOAT32, {2}}, in TEST() 251 MeanOpConstModel m({TensorType_FLOAT32, {4, 3, 2}}, {TensorType_FLOAT32, {3}}, in TEST() 266 MeanOpConstModel m({TensorType_FLOAT32, {2, 2, 3, 2}}, in TEST() 279 MeanOpConstModel m({TensorType_UINT8, {1, 2, 2, 3}, -1.0, 1.0}, in TEST() 295 MeanOpConstModel m({TensorType_UINT8, {1, 2, 2, 9}, -1.0, 1.0}, in TEST() 310 MeanOpConstModel m({TensorType_UINT8, {1, 2, 3, 2}, 0.0, 1.0}, in TEST() 322 MeanOpConstModel m({TensorType_FLOAT32, {}}, {TensorType_FLOAT32, {}}, {}, in TEST() 377 MeanOpConstModel m({TensorType_UINT8, {1, 3, 2}, -1.0, 1.0}, in TEST() [all …]
|