Searched defs:Dropout (Results 1 – 12 of 12) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/ |
| H A D | dropout.h | 14 struct Dropout { struct
19 … __forceinline__ __device__ Dropout(const unsigned long long seed, const unsigned long long offset, in Dropout() argument
28 __forceinline__ __device__ void apply_dropout(Tensor<Engine, Layout> &tensor_, in apply_dropout()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Lstm.h | 33 static constexpr uint32_t Dropout = 3u; member
48 static constexpr uint32_t Dropout = 3u; member
|
| H A D | Gru.h | 33 static constexpr uint32_t Dropout = 3u; member
48 static constexpr uint32_t Dropout = 3u; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/layers/ |
| H A D | core.py | 25 Dropout = core.Dropout variable
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/ |
| H A D | dropout.py | 8 class Dropout(torch.nn.Dropout): class
|
| /aosp_15_r20/external/pytorch/torch/nn/modules/ |
| H A D | dropout.py | 35 class Dropout(_DropoutNd): class
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/ |
| H A D | core.py | 192 class Dropout(keras_layers.Dropout, base.Layer): class
|
| /aosp_15_r20/external/libaom/test/ |
| H A D | encodemb_test.cc | 46 void Dropout(TX_SIZE tx_size, TX_TYPE tx_type, int dropout_num_before, in Dropout() function
|
| /aosp_15_r20/external/pytorch/test/cpp/jit/ |
| H A D | test_schema_info.cpp | 110 TEST(SchemaInfoIsNonDeterministicTest, Dropout) { in TEST() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/layers/ |
| H A D | core.py | 137 class Dropout(Layer): class
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | functional.cpp | 2853 TEST_F(FunctionalTest, Dropout) { in TEST_F() argument
|
| H A D | modules.cpp | 1352 TEST_F(ModulesTest, Dropout) { in TEST_F() argument
|