Esempio n. 1
0
 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())))
Esempio n. 2
0
 def test_convolution_loop_body_graph(self):
     self.assertTrue(
         graph_or_sub_graph_has_nhwc_ops(
             self.build_loop_graph(self.build_conv_graph())))