Searched refs:_assert_trans_nhwc_to_nchw (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/grappler/ |
| D | layout_optimizer_test.py | 210 def _assert_trans_nhwc_to_nchw(self, name, nodes): member in LayoutOptimizerTest
282 self._assert_trans_nhwc_to_nchw('Conv2D-0', nodes)
318 self._assert_trans_nhwc_to_nchw('Conv2D-0', nodes)
353 self._assert_trans_nhwc_to_nchw('Conv2D-0', nodes)
388 self._assert_trans_nhwc_to_nchw('Conv2D-0', nodes)
420 self._assert_trans_nhwc_to_nchw('Conv2D-0', nodes)
450 self._assert_trans_nhwc_to_nchw('Conv2D-0', nodes)
482 self._assert_trans_nhwc_to_nchw('Conv2D-0', nodes)
513 self._assert_trans_nhwc_to_nchw('Conv2D-0', nodes)
544 self._assert_trans_nhwc_to_nchw('Conv2D-0', nodes)
[all …]
|