Searched refs:param_specs (Results 1 – 1 of 1) sorted by relevance
1243 def __init__(self, param_specs, non_tensor_params, prefer_static_fields): argument1257 self._param_specs = param_specs1279 param_specs = {}1285 param_specs[k] = type_spec_or_v1293 param_specs=param_specs,1325 param_specs=nest.map_structure(1332 param_specs=nest.map_structure(