Lines Matching defs:direction
196 direction=CUDNN_RNN_UNIDIRECTION,
205 tensors followed by bias tensors, and forward direction followed by
206 backward direction (if applicable). When restoring, a user could name
239 direction: the direction model that the model operates. Could be either
250 self._direction = direction
315 direction=self._direction)
336 direction=self._direction)
771 def check_direction(direction):
772 """Check validity of direction."""
773 if direction not in (CUDNN_RNN_UNIDIRECTION, CUDNN_RNN_BIDIRECTION):
774 raise ValueError("Invalid direction: %s, expecting %s or %s" %
775 (direction, CUDNN_RNN_UNIDIRECTION, CUDNN_RNN_BIDIRECTION))
786 def _get_num_params(rnn_mode, num_layers, direction):
799 if direction != CUDNN_RNN_UNIDIRECTION:
811 direction=CUDNN_RNN_UNIDIRECTION,
835 direction: the direction model that the model operates. Could be either
845 check_direction(direction)
856 direction=direction,
870 direction=CUDNN_RNN_UNIDIRECTION,
893 direction: the direction model that the model operates. Could be either
903 input_mode, direction, dropout, seed, name)
912 direction=CUDNN_RNN_UNIDIRECTION,
934 direction: the direction model that the model operates. Could be either
946 direction, dropout, seed, name)
955 direction=CUDNN_RNN_UNIDIRECTION,
976 direction: the direction model that the model operates. Could be either
986 input_mode, direction, dropout, seed, name)
994 direction=CUDNN_RNN_UNIDIRECTION,
1015 direction: the direction model that the model operates. Could be either
1025 CUDNN_RNN_RELU, input_mode, direction, dropout,
1034 direction=CUDNN_RNN_UNIDIRECTION,
1055 direction: the direction model that the model operates. Could be either
1065 CUDNN_RNN_TANH, input_mode, direction, dropout,
1075 direction=CUDNN_RNN_UNIDIRECTION,
1097 direction: the direction model that the model operates. Could be either
1106 ValueError: if rnn_mode or direction is invalid.
1110 check_direction(direction)
1112 num_params = _get_num_params(rnn_mode, num_layers, direction)
1121 direction=direction,
1137 direction=CUDNN_RNN_UNIDIRECTION,
1160 direction: the direction
1169 ValueError: if rnn_mode or direction is invalid.
1172 check_direction(direction)
1183 direction=direction,
1195 direction=CUDNN_RNN_UNIDIRECTION,
1217 direction: the direction model that the model operates. Could be either
1227 ValueError: if rnn_mode or direction is invalid.
1230 check_direction(direction)
1244 direction=direction,
1263 direction=CUDNN_RNN_UNIDIRECTION,
1284 direction: the direction model that the model operates. Could be either
1291 ValueError: if direction is invalid.
1298 self._direction = direction
1324 def direction(self):
1342 direction=self._direction)
1369 direction=self._direction,
1389 direction=self._direction,
1411 direction=self._direction,
1428 direction=CUDNN_RNN_UNIDIRECTION,
1445 direction: the direction model that the model operates. Could be either
1457 direction=direction,
1493 direction=CUDNN_RNN_UNIDIRECTION,
1510 direction: the direction model that the model operates. Could be either
1517 ValueError: if direction is not 'unidirectional' or 'bidirectional'.
1520 if direction not in (CUDNN_RNN_UNIDIRECTION, CUDNN_RNN_BIDIRECTION):
1521 raise ValueError("Invalid direction: %s", direction)
1529 direction=direction,
1555 direction=self._direction,
1609 direction=op.get_attr("direction"))
