Home
last modified time | relevance | path

Searched refs:param_buffer_table (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/export/
H A D_trace.py233 def _replace_param_buffer_names(param_buffer_table, sig): argument
239 spec.target = param_buffer_table[spec.target]
245 spec.target = param_buffer_table[spec.target]
328 param_buffer_table: Dict[str, str] = {}
332 assert dynamo_name not in param_buffer_table
334 param_buffer_table[dynamo_name] = param_lookup[id(dynamo_param)]
339 assert dynamo_name not in param_buffer_table
341 param_buffer_table[dynamo_name] = buffer_lookup[id(dynamo_buffer)]
343 return param_buffer_table
467 param_buffer_table = _get_param_buffer_mapping(original_module, traced_module)
[all …]