Searched refs:_get_composite_tensor_spec (Results 1 – 1 of 1) sorted by relevance
1154 def _get_composite_tensor_spec(x): function1207 arg_value = nest.map_structure(_get_composite_tensor_spec, arg_value)