Searched refs:_replace_tensor_with_spec (Results 1 – 1 of 1) sorted by relevance
453 spec_fields = nest.map_structure(_replace_tensor_with_spec, value_fields)722 def _replace_tensor_with_spec(value): function