Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Ddistribute_utils.py289 def create_mirrored_variable(strategy, real_mirrored_creator, class_mapping, argument
327 var_cls = class_mapping.get("VariableClass")
330 var_cls = class_mapping.get(synchronization)
/aosp_15_r20/external/pytorch/torch/ao/quantization/
H A Dutils.py242 class_mapping = custom_module_class_mapping.get(quant_type, {})
243 assert type(custom_module) in class_mapping, (
247 return class_mapping[type(custom_module)]