Searched refs:_AssertShapesMatch (Results 1 – 1 of 1) sorted by relevance
791 _AssertShapesMatch('mul_fold', mul_fold.inputs[0], mul_fold.outputs[0])814 _AssertShapesMatch('add_fold', add_fold.inputs[0], add_fold.outputs[0])940 def _AssertShapesMatch(op_name, in_tensor, out_tensor): function