Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/quantization/
H A Dquantization.cpp63 at::Tensor _bfloat16_to_float_cpu(const at::Tensor& input) { in _bfloat16_to_float_cpu() function
88 m.impl("_Bfloat16QuantizedToFloat", _bfloat16_to_float_cpu); in TORCH_LIBRARY_IMPL()
H A Dquantization.h14 at::Tensor _bfloat16_to_float_cpu(const at::Tensor& input);