(dali_res, np_res)
| 193 | |
| 194 | |
| 195 | def compare(dali_res, np_res): |
| 196 | for dali_sample, np_sample in zip(dali_res, np_res): |
| 197 | assert dali_sample.shape == np_sample.shape |
| 198 | if dali_res[0].dtype == np.float32: |
| 199 | assert np.allclose(dali_sample, np_sample) |
| 200 | else: |
| 201 | if not np.array_equal(dali_sample, np_sample): |
| 202 | print(dali_sample) |
| 203 | print(np_sample) |
| 204 | assert np.array_equal(dali_sample, np_sample) |
| 205 | |
| 206 | |
| 207 | def np_mean_square(input, keepdims=False, axis=None, dtype=None): |
no test coverage detected