Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_alias_analysis.cpp496 auto isTraining = std::get<2>(param); in TEST_P() local
572 auto isTraining = std::get<2>(param); in TEST_P() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DMHA.cpp19 bool isTraining, in run_cudnn_SDP_fprop()
/aosp_15_r20/external/pytorch/torch/csrc/jit/ir/
H A Dalias_analysis.cpp1132 auto isTraining = getConstantBooleanInput(node, "training"); in analyzeBatchNorm() local