Searched refs:assert_shape_equal_scripted (Results 1 – 1 of 1) sorted by relevance
478 def assert_shape_equal_scripted(self, script_fn, given_ins): member in TestSymbolicShapeAnalysis512 self.assert_shape_equal_scripted(conv_bwd, (input, weight, out_grad))522 self.assert_shape_equal_scripted(conv_bwd_2, (input, weight, out_grad))