Searched refs:un_op (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | all_reduce.py | 255 gpu_perm, red_op, un_op=None): argument
286 if un_op:
287 chunks_by_dev = _apply_unary_to_chunks(un_op, chunks_by_dev)
426 def build_recursive_hd_all_reduce(input_tensors, red_op, un_op=None): argument
466 if un_op:
467 reduced_shards = [un_op(t) for t in reduced_shards]
552 def build_shuffle_all_reduce(input_tensors, gather_devices, red_op, un_op=None): argument
579 red_op, un_op)
586 def _build_shuffle_gather(input_tensors, gather_devices, red_op, un_op=None): argument
617 if un_op:
[all …]
|
| D | all_reduce_test.py | 147 return lambda x, un_op: ar.build_ring_all_reduce(
148 x, num_workers, subdiv, gpu_perm, math_ops.add, un_op)
157 un_op = lambda x: math_ops.div( function
161 output_tensors = build_f(input_tensors, un_op)
194 return lambda x, un_op: ar.build_shuffle_all_reduce(
195 x, gather_devices, math_ops.add_n, un_op)
218 return lambda x, un_op: ar.build_recursive_hd_all_reduce(
219 x, math_ops.add, un_op)
|
| /external/apache-commons-bcel/src/examples/Mini/ |
| D | Mini.jjt | 96 int un_op=-1;
105 un_op = UnOp() { jjtThis.setUnOp(un_op); } Expr()
|
| D | Mini.jj | 141 int un_op=-1;
152 un_op = UnOp() { jjtn000.setUnOp(un_op); } Expr()/*@bgen(jjtree)*/
|
| D | MiniParser.java | 150 int un_op=-1; in Expr() local
181 un_op = UnOp(); in Expr()
182 jjtn000.setUnOp(un_op); in Expr()
|