Home | Sort by relevance Sort by last modified time |
/external/tensorflow/tensorflow/contrib/all_reduce/python/ | |
all_reduce.py | 776 def upper_builder(y): function in function:build_nccl_then_ring 779 return _reduce_non_singleton(x, upper_builder, un_op) 842 def upper_builder(tensors): function in function:build_shuffle_then_ring 846 return _reduce_non_singleton(tensors, upper_builder, un_op) 854 def upper_builder(tensors): function in function:build_shuffle_then_shuffle 858 return _reduce_non_singleton(tensors, upper_builder, un_op) |