Searched defs:hidden_bias (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | RNN.cu | 34 const TensorArg& input_bias, const TensorArg& hidden_bias, in checkSizes()
371 const Tensor& input_bias, const Tensor& hidden_bias, in lstm_forward_impl()
444 const Tensor& input_bias, const Tensor& hidden_bias, in gru_forward_impl()
523 const Tensor& hidden_bias = c10::value_or_else(hidden_bias_opt, [] {return Tensor();}); in _thnn_fused_lstm_cell_cuda() local
609 const Tensor& hidden_bias = c10::value_or_else(hidden_bias_opt, [] {return Tensor();}); in _thnn_fused_gru_cell_cuda() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradient_checker_v2_test.py | 332 def f(inp, hidden_weight, hidden_bias, softmax_weight, softmax_bias): argument
|
| /external/pytorch/aten/src/ATen/native/ |
| D | RNN.cpp | 1554 const Tensor& hidden_bias = c10::value_or_else(hidden_bias_opt, [] {return Tensor();}); in _thnn_differentiable_lstm_cell_backward() local
1606 const Tensor& hidden_bias = c10::value_or_else(hidden_bias_opt, [] {return Tensor();}); in _thnn_differentiable_gru_cell_backward() local
|
| /external/pytorch/torch/ |
| D | _meta_registrations.py | 5866 hidden_bias, argument
5907 hidden_bias=None, argument
|