Searched refs:FloatMulOpModel (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | mul_test.cc | 48 class FloatMulOpModel : public BaseMulOpModel { class
89 FloatMulOpModel m({TensorType_FLOAT32, {1, 2, 2, 1}}, in TEST()
100 FloatMulOpModel m( in TEST()
114 FloatMulOpModel m({TensorType_FLOAT32, test_shapes[i]}, in TEST()
131 FloatMulOpModel m({TensorType_FLOAT32, test_shapes[i]}, in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 161 class FloatMulOpModel : public SingleOpModelWithNNAPI { class
163 FloatMulOpModel(const TensorData& input1, const TensorData& input2, in FloatMulOpModel() function in tflite::__anon0c68fcb70111::FloatMulOpModel
186 FloatMulOpModel m({TensorType_FLOAT32, {1, 2, 2, 1}}, in TEST()
|