Searched refs:bfloat16_output (Results 1 – 2 of 2) sorted by relevance
1420 bool bfloat16_output = output_dtype.has_value() && (output_dtype.value() == c10::kBFloat16); in _quantized_convolution_onednn() local1421 if (fp32_output || bfloat16_output) { in _quantized_convolution_onednn()1448 if (fp32_output || bfloat16_output) { in _quantized_convolution_onednn()1597 .dtype(fp32_output ? c10::kFloat : (bfloat16_output ? c10::kBFloat16 : c10::kByte)) in _quantized_convolution_onednn()
6662 bfloat16_output = True if qconv_output_dtype is torch.bfloat16 else False6663 if fp32_output or bfloat16_output:6827 if fp32_output or bfloat16_output: