Searched refs:TensorShapeMap (Results 1 – 2 of 2) sorted by relevance
43 using TensorShapeMap = absl::flat_hash_map<std::string, TensorShape>; variable49 const TensorShapeMap& GetTensorShapeMap() const { return shape_map_; } in GetTensorShapeMap()56 DataTypeMap data_type_map, TensorShapeMap shape_map);61 TensorShapeMap shape_map_;
62 TensorShapeMap shape_map; in Create()76 DataTypeMap data_type_map, TensorShapeMap shape_map) in CheckpointReader()