Searched refs:var_to_shape_map (Results 1 – 4 of 4) sorted by relevance
32 var_to_shape_map = reader.get_variable_to_shape_map()40 for v in var_to_shape_map:43 exclude_num_params += np.prod(var_to_shape_map[v])45 new_var_to_shape_map[v] = var_to_shape_map[v]46 var_to_shape_map = new_var_to_shape_map50 var_sizes = [np.prod(var_to_shape_map[v]) for v in var_to_shape_map]74 var_to_shape_map = reader.get_variable_to_shape_map()76 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]))
161 var_to_shape_map = reader.get_variable_to_shape_map()173 for key in var_to_shape_map:
257 var_to_shape_map = reader.get_variable_to_shape_map()258 self.assertLen(var_to_shape_map, 1)259 self.assertIn('v', var_to_shape_map)280 var_to_shape_map = reader.get_variable_to_shape_map()281 self.assertLen(var_to_shape_map, 2)282 self.assertIn('v/foo', var_to_shape_map)283 self.assertIn('v/bar', 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()