Searched refs:get_flat_tensor_specs (Results 1 – 2 of 2) sorted by relevance
271 def get_flat_tensor_specs(element_spec): function297 return [spec.shape for spec in get_flat_tensor_specs(element_spec)]310 return [spec.dtype for spec in get_flat_tensor_specs(element_spec)]
3573 input_signature=structure.get_flat_tensor_specs(