Searched defs:un_op (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/v1/ |
| H A D | all_reduce.py | 251 gpu_perm, red_op, un_op=None): argument
422 def build_recursive_hd_all_reduce(input_tensors, red_op, un_op=None): argument
554 def build_shuffle_all_reduce(input_tensors, gather_devices, red_op, un_op=None): argument
588 def _build_shuffle_gather(input_tensors, gather_devices, red_op, un_op=None): argument
681 def build_nccl_all_reduce(input_tensors, red_op, un_op=None): argument
761 def _reduce_non_singleton(input_tensors, red_f, un_op): argument
775 def build_nccl_then_ring(input_tensors, subdiv, red_op, un_op=None): argument
784 def build_nccl_then_recursive_hd(input_tensors, red_op, un_op=None): argument
791 shuffle_red_op, un_op=None): argument
842 red_n_op, red_op, un_op=None): argument
[all …]
|
| H A D | all_reduce_test.py | 152 un_op = lambda x: math_ops.div( function
|
| /aosp_15_r20/external/apache-commons-bcel/src/examples/Mini/ |
| H A D | MiniParser.java | 150 int un_op=-1; in Expr() local
|