Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Ddropout.h195 inline Tensor feature_alpha_dropout( in feature_alpha_dropout() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DDropout.cpp167 Tensor feature_alpha_dropout(const Tensor& input, double p, bool train) { in feature_alpha_dropout() function
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DPyTorchOperatorHacks.cpp247 Tensor feature_alpha_dropout(const Tensor& input, double p, bool train) { in feature_alpha_dropout() function
/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.py1614 def feature_alpha_dropout( function
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dmodules.cpp1334 FeatureAlphaDropout feature_alpha_dropout(0.5); in TEST_F() local