Searched refs:opaque_param (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_aot_test_helper.cc | 54 auto opaque_param = Parameter(&builder, 0, opaque_shape, "x"); in main() local
56 auto sum = CustomCall(&builder, "SumStructElements", {opaque_param}, r0f32); in main()
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| D | cudnn_rnn_ops_test.py | 1083 def _create_saveable(self, opaque_param, rnn_mode, num_units, input_size, argument
1094 opaque_param, num_layers, num_units, input_size, direction=direction)
1124 opaque_param = self._create_opaque_param(rnn_mode, num_units, input_size,
1126 saveable = self._create_saveable(opaque_param, rnn_mode, num_units,
1136 reset_op = state_ops.assign(opaque_param,
1137 array_ops.zeros_like(opaque_param))
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| D | cudnn_rnn_ops.py | 222 def opaque_to_tf_canonical(self, opaque_param): argument
224 cu_weights, cu_biases = self._opaque_to_cu_canonical(opaque_param)
228 def _opaque_to_cu_canonical(self, opaque_param): argument
242 params=opaque_param,
|