Searched refs:input_component_shape (Results 1 – 1 of 1) sorted by relevance
2790 def _is_padded_shape_compatible_with(padded_shape, input_component_shape): argument2802 if padded_shape.dims is None or input_component_shape.dims is None:2804 if len(padded_shape.dims) != len(input_component_shape.dims):2807 padded_shape.dims, input_component_shape.dims):2814 def _padded_shape_to_tensor(padded_shape, input_component_shape): argument2855 input_component_shape):2858 % (padded_shape_as_shape, input_component_shape))2927 for input_component_shape, padded_shape in zip(2930 _padded_shape_to_tensor(padded_shape, input_component_shape))