Searched refs:get_flat_tensor_specs (Results 1 – 2 of 2) sorted by relevance
196 input_signature=structure.get_flat_tensor_specs(221 input_signature=structure.get_flat_tensor_specs(235 input_signature=structure.get_flat_tensor_specs(
275 def get_flat_tensor_specs(element_spec): function302 return [spec.shape for spec in get_flat_tensor_specs(element_spec)]315 return [spec.dtype for spec in get_flat_tensor_specs(element_spec)]