Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/
DContext.cpp215 Float32MatmulPrecision Context::float32MatmulPrecision() const { in float32MatmulPrecision() function in at::Context
DContext.h311 Float32MatmulPrecision float32MatmulPrecision() const;
/external/pytorch/aten/src/ATen/native/mkldnn/
DMatmul.cpp108 …return use_mkldnn_bf16_matmul() && at::globalContext().float32MatmulPrecision() == at::Float32Matm… in use_mkldnn_bf32_matmul()
/external/pytorch/torch/csrc/
DModule.cpp681 auto p = at::globalContext().float32MatmulPrecision(); in THPModule_float32MatmulPrecision()