Searched refs:dtype_map (Results 1 – 3 of 3) sorted by relevance
96 self.dtype_map = pywrap_tensorflow.NewCheckpointReader(174 def __init__(self, save_path, dtype_map=None): argument176 self.dtype_map = dtype_map232 if spec.name in self.dtype_map:238 dtypes=[self.dtype_map[spec.name]],1054 dtype_map = None1056 dtype_map = reader.get_variable_to_dtype_map()1064 save_path=save_path, dtype_map=dtype_map)
292 dtype = self._checkpoint.dtype_map[checkpoint_key]
56 dtype_map = reader.get_variable_to_dtype_map()81 dtype_map[attribute.checkpoint_key].name)