Home
last modified time | relevance | path

Searched defs:bf32_usable (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/
H A DMatmul.cpp128 bool bf32_usable = std::is_same_v<scalar_t, float> && use_mkldnn_bf32_matmul(); in mkldnn_gemm() local
273 bool bf32_usable = mat1.scalar_type() == at::kFloat && use_mkldnn_bf32_matmul(); in mkldnn_matmul() local