Searched refs:var_to_shape_map (Results 1 – 3 of 3) sorted by relevance
35 var_to_shape_map = reader.get_variable_to_shape_map()43 for v in var_to_shape_map:46 exclude_num_params += np.prod(var_to_shape_map[v])48 new_var_to_shape_map[v] = var_to_shape_map[v]49 var_to_shape_map = new_var_to_shape_map53 var_sizes = [np.prod(var_to_shape_map[v]) for v in var_to_shape_map]77 var_to_shape_map = reader.get_variable_to_shape_map()79 for key, value in sorted(var_to_shape_map.items()):90 var_to_shape_map = reader.get_variable_to_shape_map()94 var_to_shape_map[tensor_name]))
165 var_to_shape_map = reader.get_variable_to_shape_map()177 for key in var_to_shape_map:
134 std::unique_ptr<TensorSliceReader::VarToShapeMap> var_to_shape_map( in BuildV2VarMaps() local145 (*var_to_shape_map)[key] = TensorShape(entry.shape()); in BuildV2VarMaps()149 return std::make_pair(std::move(var_to_shape_map), in BuildV2VarMaps()