Home
last modified time | relevance | path

Searched refs:opaque_params (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/
Dcudnn_rnn_ops_test.py141 opaque_params = format_converter.tf_canonical_to_opaque([w, b])
151 opaque_params,
170 [inputs, cu_initial_h_op, cu_initial_c_op, opaque_params])
640 opaque_params = format_converter.tf_canonical_to_opaque(ws + bs)
649 opaque_params,
662 cu_outputs_op, [inputs, cu_initial_h_op, opaque_params])
945 opaque_params = format_converter.tf_canonical_to_opaque(ws + bs)
953 ws_r, bs_r = format_converter.opaque_to_tf_canonical(opaque_params)
1019 opaque_params = format_converter.tf_canonical_to_opaque(ws + bs)
1027 ws_r, bs_r = format_converter.opaque_to_tf_canonical(opaque_params)
[all …]
Dcudnn_rnn_test.py542 opaque_params = rnn.trainable_variables[0]
546 opaque_params,
547 array_ops.zeros(array_ops.shape(opaque_params), dtype=dtype))
586 opaque_params = (model1.rnn.trainable_variables[0],
597 for params in opaque_params
1051 opaque_params = rnn.trainable_variables[0]
1054 opaque_params_size_v = opaque_params.eval().size
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/layers/
Dcudnn_rnn.py498 def _forward(self, inputs, h, c, opaque_params, sequence_lengths, time_major, argument
504 opaque_params,
528 opaque_params=self.trainable_variables[0],
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/
Dcudnn_rnn_ops.py219 opaque_params = self._cu_canonical_to_opaque(cu_weights, cu_biases)
220 return opaque_params
633 opaque_params, argument
698 self._variables = opaque_params
735 opaque_params = self.format_converter.tf_canonical_to_opaque(
738 self._variables, opaque_params, validate_shape=False)