Searched defs:un_op (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | all_reduce.py | 255 gpu_perm, red_op, un_op=None): argument
426 def build_recursive_hd_all_reduce(input_tensors, red_op, un_op=None): argument
558 def build_shuffle_all_reduce(input_tensors, gather_devices, red_op, un_op=None): argument
592 def _build_shuffle_gather(input_tensors, gather_devices, red_op, un_op=None): argument
685 def build_nccl_all_reduce(input_tensors, red_op, un_op=None): argument
765 def _reduce_non_singleton(input_tensors, red_f, un_op): argument
779 def build_nccl_then_ring(input_tensors, subdiv, red_op, un_op=None): argument
788 def build_nccl_then_recursive_hd(input_tensors, red_op, un_op=None): argument
795 shuffle_red_op, un_op=None): argument
846 red_n_op, red_op, un_op=None): argument
[all …]
|
| D | all_reduce_test.py | 156 un_op = lambda x: math_ops.div( function
|
| /external/apache-commons-bcel/src/examples/Mini/ |
| D | MiniParser.java | 150 int un_op=-1; in Expr() local
|
| /external/llvm-project/polly/lib/External/isl/ |
| D | isl_union_map.c | 1902 static __isl_give isl_union_map *un_op(__isl_take isl_union_map *umap, in un_op() function
|