Searched refs:_assert_trans_nchw_to_nhwc (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/grappler/ |
| D | layout_optimizer_test.py | 207 def _assert_trans_nchw_to_nhwc(self, name, nodes): member in LayoutOptimizerTest
283 self._assert_trans_nchw_to_nhwc('Relu_1-0-0', nodes)
319 self._assert_trans_nchw_to_nhwc('add_2-0-0', nodes)
354 self._assert_trans_nchw_to_nhwc('SplitV-0-0', nodes)
389 self._assert_trans_nchw_to_nhwc('Pad-0-0', nodes)
451 self._assert_trans_nchw_to_nhwc('Cast-0-0', nodes)
665 self._assert_trans_nchw_to_nhwc('Sum-0-0', nodes)
760 self._assert_trans_nchw_to_nhwc('concat-0-0', nodes)
805 self._assert_trans_nchw_to_nhwc('Fill-0-0', nodes)
841 self._assert_trans_nchw_to_nhwc('Tile-0-0', nodes)
[all …]
|