Searched refs:quantize_tensor (Results 1 – 4 of 4) sorted by relevance
73 void quantize_tensor(const ITensor *input, ITensor *output) in quantize_tensor() function254 quantize_tensor(&_scores_out_f32, _scores_out); in run()255 quantize_tensor(&_boxes_out_f32, _boxes_out); in run()256 quantize_tensor(&_classes_f32, _classes); in run()259 quantize_tensor(&_batch_splits_out_f32, _batch_splits_out); in run()263 quantize_tensor(&_keeps_f32, _keeps); in run()
117 TORCH_API Tensor quantize_tensor(
105 def quantize_tensor(self, tensor: torch.Tensor) -> torch.Tensor: member in QuantParams
536 const_val = quant_params.quantize_tensor(const_val).contiguous()