Searched refs:get_flat_tensor_shapes (Results 1 – 9 of 9) sorted by relevance
113 self._flat_tensor_shapes = structure.get_flat_tensor_shapes(212 output_shapes=structure.get_flat_tensor_shapes(284 output_shapes=structure.get_flat_tensor_shapes(output_structure))430 output_shapes=structure.get_flat_tensor_shapes(676 self._flat_output_shapes = structure.get_flat_tensor_shapes(698 self._flat_output_shapes = structure.get_flat_tensor_shapes(810 output_shapes=structure.get_flat_tensor_shapes(
191 output_shapes=structure.get_flat_tensor_shapes(self._element_spec))
78 output_shapes=structure.get_flat_tensor_shapes(85 output_shapes=structure.get_flat_tensor_shapes(self._element_spec))
545 return structure.get_flat_tensor_shapes(self.element_spec)2218 output_shapes=structure.get_flat_tensor_shapes(state_structure),3197 output_shapes=structure.get_flat_tensor_shapes(self._structure))3224 output_shapes=structure.get_flat_tensor_shapes(self._structure))4276 output_shapes=structure.get_flat_tensor_shapes(self._structure))4287 output_shapes=structure.get_flat_tensor_shapes(self._structure))
52 flat_shapes = structure.get_flat_tensor_shapes(output_spec)
287 def get_flat_tensor_shapes(element_spec): function
92 flat_shapes = structure.get_flat_tensor_shapes(s)
342 structure.get_flat_tensor_shapes(opt_structure))
289 structure.get_flat_tensor_shapes(dataset_structure))