| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | PyTorchOperatorHacks.cpp | 164 template<bool feature_dropout, bool alpha_dropout, bool inplace, typename T>
179 if (feature_dropout) { in _dropout_impl()
231 Tensor feature_dropout(const Tensor& input, double p, bool train) { in feature_dropout() function
265 m.impl("feature_dropout", dropout_hack::feature_dropout); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Dropout.cpp | 61 template<bool feature_dropout, bool alpha_dropout, bool inplace, typename T>
73 auto noise = feature_dropout ? make_feature_noise(input) : at::empty_like(input); in _dropout_impl()
151 Tensor feature_dropout(const Tensor& input, double p, bool train) { in feature_dropout() function
|
| H A D | native_functions.yaml | 317 - func: feature_dropout(Tensor input, float p, bool train) -> Tensor
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | remove_dropout.cpp | 30 node->kind() == c10::Symbol::fromQualString("aten::feature_dropout") || in removeDropoutImpl()
|
| H A D | shape_analysis.cpp | 854 "aten::feature_dropout(Tensor input, float p, bool train) -> Tensor", in PropagateTensorShapeOnNode()
|
| /aosp_15_r20/external/pytorch/functorch/op_analysis/ |
| H A D | public_api | 414 feature_dropout
|
| H A D | annotated_ops | 8 feature_dropout, composite pointwise
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | coverage.yaml | 228 - aten::feature_dropout
812 aten::feature_dropout: 1
|
| H A D | model_ops.yaml | 143 aten::feature_dropout: 1
|
| /aosp_15_r20/external/pytorch/functorch/dim/ |
| H A D | op_properties.py | 142 torch.feature_dropout,
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_vmap_registrations.py | 71 "aten::feature_dropout",
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | dropout.h | 77 result = torch::feature_dropout(input, p, training); in _dropoutNd_helper()
|
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | functional_tensor.py | 110 torch.ops.aten.feature_dropout.default, # type: ignore[has-type]
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset9.py | 3259 "aten::feature_dropout",
3260 decorate=[symbolic_helper._apply_params("aten::feature_dropout")],
3264 def feature_dropout(g, input, p, train): function
3270 return feature_dropout
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_freezing.py | 2738 FileCheck().check("aten::feature_dropout").run(mod.graph)
2740 FileCheck().check_not("aten::feature_dropout").run(frozen_mod.graph)
|
| /aosp_15_r20/external/pytorch/torch/nn/ |
| H A D | functional.py | 1494 else _VF.feature_dropout(input, p, training)
1556 else _VF.feature_dropout(input, p, training)
1606 else _VF.feature_dropout(input, p, training)
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | trace_rules.py | 1817 "torch.feature_dropout",
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | overrides.py | 617 torch.feature_dropout: lambda input, p, train: -1,
|