Searched refs:assert_shapes_equal (Results 1 – 1 of 1) sorted by relevance
200 def assert_shapes_equal(expected, actual): function220 assert_shapes_equal(tensor_shape.TensorShape(expected_output_shape),232 assert_shapes_equal(computed_output_shape, actual_output_shape)233 assert_shapes_equal(computed_output_signature.shape, actual_output_shape)