Searched refs:w_ru (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_GRUBlockCellGrad.pbtxt | 9 w_ru: Weight matrix for the reset and update gate.
35 `w_ru` can be segmented into 4 different matrices.
37 w_ru = [w_r_x w_u_x
67 [d_x_component_1 d_h_prev_component_1] = d_r_bar_u_bar * w_ru^T
|
| D | api_def_GRUBlockCell.pbtxt | 9 w_ru: Weight matrix for the reset and update gate.
36 [r_bar u_bar] = x_h_prev * w_ru + b_ru
|
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | gru_ops.h | 77 typename TTypes<T>::ConstMatrix w_ru, typename TTypes<T>::ConstMatrix w_c, in operator()
92 const_x_h_prev, w_ru, typename gemm_compute_type<T>::type(0.f), in operator()
133 typename TTypes<T>::ConstMatrix w_ru, typename TTypes<T>::ConstMatrix w_c, in operator()
172 const_d_r_bar_u_bar, w_ru, typename gemm_compute_type<T>::type(0.f), in operator()
|
| D | gru_ops.cc | 394 typename TTypes<T>::ConstMatrix w_ru, \
423 typename TTypes<T>::ConstMatrix w_ru, \
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | GRUBlockCell.pbtxt | 12 name: "w_ru"
|
| D | GRUBlockCellGrad.pbtxt | 12 name: "w_ru"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | GRUBlockCell.pbtxt | 12 name: "w_ru"
|
| D | GRUBlockCellGrad.pbtxt | 12 name: "w_ru"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | rnn_ops.cc | 68 ShapeHandle x, h_prev, w_ru; in __anon652ace880202() local
71 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 2, &w_ru)); in __anon652ace880202()
75 DimensionHandle twice_cell_size = c->Dim(w_ru, 1); in __anon652ace880202()
|
| D | ops.pbtxt | 18064 name: "w_ru"
18116 name: "w_ru"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 1749 …argspec: "args=[\'x\', \'h_prev\', \'w_ru\', \'w_c\', \'b_ru\', \'b_c\', \'name\'], varargs=None, …
1753 …argspec: "args=[\'x\', \'h_prev\', \'w_ru\', \'w_c\', \'b_ru\', \'b_c\', \'r\', \'u\', \'c\', \'d_…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 1749 …argspec: "args=[\'x\', \'h_prev\', \'w_ru\', \'w_c\', \'b_ru\', \'b_c\', \'name\'], varargs=None, …
1753 …argspec: "args=[\'x\', \'h_prev\', \'w_ru\', \'w_c\', \'b_ru\', \'b_c\', \'r\', \'u\', \'c\', \'d_…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 30562 func GRUBlockCellGrad(scope *Scope, x tf.Output, h_prev tf.Output, w_ru tf.Output, w_c tf.Output, b…
30569 x, h_prev, w_ru, w_c, b_ru, b_c, r, u, c, d_h,
37003 func GRUBlockCell(scope *Scope, x tf.Output, h_prev tf.Output, w_ru tf.Output, w_c tf.Output, b_ru …
37010 x, h_prev, w_ru, w_c, b_ru, b_c,
|