Home
last modified time | relevance | path

Searched refs:setAllowFP16ReductionCPU (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/aten/src/ATen/
DContext.cpp529 void Context::setAllowFP16ReductionCPU(bool b) { in setAllowFP16ReductionCPU() function in at::Context
DContext.h335 void setAllowFP16ReductionCPU(bool);
/external/pytorch/torch/csrc/
DModule.cpp1073 at::globalContext().setAllowFP16ReductionCPU(arg == Py_True); in THPModule_setAllowFP16ReductionCPU()