Searched defs:DropoutDescriptorParams (Results 1 – 1 of 1) sorted by relevance
124 struct DropoutDescriptorParams { struct125 bool train;126 double dropout;127 Tensor dropout_state;129 void set(bool train_, double dropout_, Tensor dropout_state_) { in set()134 DropoutDescriptor descriptor(cudnnHandle_t handle) const { in descriptor()