| 120 | |
| 121 | template <typename T> |
| 122 | void CalculateTrueResults(const std::vector<T>& input_data, T exponent, |
| 123 | int flat_size, std::vector<T>* output_data) { |
| 124 | for (int i = 0; i < flat_size; ++i) { |
| 125 | output_data->at(i) = std::pow(input_data[i], exponent); |
| 126 | } |
| 127 | } |
| 128 | |
| 129 | TEST(PowOpModel, FloatSingleIntegerExponentTest) { |
| 130 | PowOpModel<float> model({TensorType_FLOAT32, {1, 2, 2, 1}}, |