Searched refs:var_to_shape_map (Results 1 – 3 of 3) sorted by relevance
135 std::unique_ptr<TensorSliceReader::VarToShapeMap> var_to_shape_map( in BuildV2VarMaps() local146 (*var_to_shape_map)[key] = TensorShape(entry.shape()); in BuildV2VarMaps()150 return std::make_pair(std::move(var_to_shape_map), in BuildV2VarMaps()
50 var_to_shape_map = reader.get_variable_to_shape_map()51 for key in sorted(var_to_shape_map):
165 var_to_shape_map = reader.get_variable_to_shape_map()177 for key in var_to_shape_map: