Searched refs:assert_shape_match (Results 1 – 2 of 2) sorted by relevance
1499 def assert_shape_match(inp, out): function1503 nest.map_structure(assert_shape_match, inputs, outputs)
1334 def assert_shape_match(inp, out): function1337 nest.map_structure(assert_shape_match, inputs, outputs)