Searched refs:to_ragged_spec (Results 1 – 1 of 1) sorted by relevance
402 def to_ragged_spec(spec): function419 self._structure = nest.map_structure(to_ragged_spec,433 spec = to_ragged_spec(tensor_spec.TensorSpec.from_tensor(value))