Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dreduce_test.cc58 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 …]
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc3419 class MeanOpConstModel : public BaseReduceOpModel { class
3421 MeanOpConstModel(const TensorData& input, const TensorData& output, in MeanOpConstModel() function in tflite::__anon0c68fcb70111::MeanOpConstModel
3438 MeanOpConstModel m({TensorType_FLOAT32, {4, 3, 2}}, {TensorType_FLOAT32, {2}}, in TEST()
3450 MeanOpConstModel m({TensorType_FLOAT32, {4, 3, 2}}, {TensorType_FLOAT32, {3}}, in TEST()