def test_no_convolution_main_and_sub_graph(self): self.assertFalse( graph_or_sub_graph_has_nhwc_ops( self.build_loop_graph(self.build_parameter_result_graph())))
def test_convolution_loop_body_graph(self): self.assertTrue( graph_or_sub_graph_has_nhwc_ops( self.build_loop_graph(self.build_conv_graph())))