Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/distributed/algorithms/_quantization/
H A Dquantization.py31 def _quantize_tensor(tensor, qtype): function
51 quantized_tensor_list = [_quantize_tensor(t, qtype) for t in tensor_list]
115 input_tensors = _quantize_tensor(args[1], qtype)
138 input_tensors = _quantize_tensor(args[1], qtype)
139 out_tensors = _quantize_tensor(tensors, qtype)