Searched refs:dtype_map (Results 1 – 2 of 2) sorted by relevance
222 self.dtype_map = py_checkpoint_reader.NewCheckpointReader(320 def __init__(self, save_path, dtype_map=None): argument322 self.dtype_map = dtype_map379 if spec.name in self.dtype_map:385 dtypes=[self.dtype_map[spec.name]],1295 dtype_map = None1297 dtype_map = reader.get_variable_to_dtype_map()1305 dtype_map=dtype_map)
349 dtype = self._checkpoint.dtype_map[checkpoint_key]