Searched refs:use_v2 (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| D | cudnn_rnn_test.py | 1205 use_v2, delta, tolerance): argument
1216 if use_v2:
1273 for config, dropout, use_v2 in itertools.product(test_configs, dropouts,
1282 use_v2 = False
1287 dir_count, dropout, dtype, use_v2, delta, tolerance)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_grad.py | 833 def _BaseFusedBatchNormGrad(op, use_v2, *grad): argument
866 if use_v2 else gen_nn_ops.fused_batch_norm_grad)
|