Searched refs:to_tensor_spec (Results 1 – 2 of 2) sorted by relevance
281 def to_tensor_spec(input_spec, default_dtype=None): function
385 … spec = input_spec.to_tensor_spec(x, layer._compute_dtype) # pylint: disable=protected-access