Home
last modified time | relevance | path

Searched defs:bias_mode (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/native/miopen/
DRNN_miopen.cpp80 miopenRNNBiasMode_t bias_mode = miopenRNNNoBias; member
340 auto bias_mode = rnn.bias_mode; in get_parameters() local
462 miopenRNNBiasMode_t bias_mode = (weight_stride0 == 4) ? miopenRNNwithBias : miopenRNNNoBias; in miopen_rnn() local
688 miopenRNNBiasMode_t bias_mode = (weight_stride0 == 4) ? miopenRNNwithBias : miopenRNNNoBias; in miopen_rnn_backward_weight() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.cc1310 cudnnRNNBiasMode_t bias_mode = CUDNN_RNN_DOUBLE_BIAS; in Create() local