| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | hierarchical_tree_broadcaster.cc | 43 string BroadcastBufKey(const string& exec_key, int subdiv, int src_rank, in BroadcastBufKey() argument
46 return strings::StrCat("broadcast(", exec_key, "):subdiv(", subdiv, in BroadcastBufKey()
50 return strings::StrCat(exec_key, ":", subdiv, ":", src_rank, ":", dst_rank); in BroadcastBufKey()
218 int subdiv) { in TreeRecvFrom() argument
219 DCHECK_LT(subdiv, static_cast<int>(cp.subdiv_rank.size())); in TreeRecvFrom()
220 int my_rank = cp.subdiv_rank[subdiv]; in TreeRecvFrom()
224 DCHECK_LT(subdiv, static_cast<int>(impl.subdiv_source_rank.size())); in TreeRecvFrom()
225 int source_rank = impl.subdiv_source_rank[subdiv]; in TreeRecvFrom()
237 int subdiv, in TreeSendTo() argument
239 DCHECK_LT(subdiv, static_cast<int>(cp.subdiv_rank.size())); in TreeSendTo()
[all …]
|
| D | hierarchical_tree_broadcaster.h | 57 static int TreeRecvFrom(const CollectiveParams& cp, int subdiv);
61 static void TreeSendTo(const CollectiveParams& cp, int subdiv,
70 void DispatchSend(int subdiv, int dst_rank, int src_rank,
76 void DispatchRecv(int subdiv, int src_rank, int dst_rank, Tensor* dst_tensor,
|
| /external/deqp/framework/randomshaders/ |
| D | rsgUtils.hpp | 61 const float subdiv = 8; in quantizeFloatRange() local
63 float newMin = deFloatCeil(min*subdiv)/subdiv; in quantizeFloatRange()
69 float newMax = deFloatFloor(max*subdiv)/subdiv; in quantizeFloatRange()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | all_reduce_test.py | 145 def _buildRing(self, num_workers, num_gpus, subdiv): argument
148 x, num_workers, subdiv, gpu_perm, math_ops.add, un_op)
166 def _testRingAllReduce(self, num_workers, num_gpus, shape, subdiv): argument
168 build_f = self._buildRing(num_workers, num_gpus, subdiv)
173 (num_workers, num_gpus, shape, subdiv, elapsed))
|
| D | all_reduce.py | 773 def build_nccl_then_ring(input_tensors, subdiv, red_op, un_op=None): argument
776 return build_ring_all_reduce(y, len(y), subdiv, [0], red_op, un_op)
839 def build_shuffle_then_ring(input_tensors, gather_devices, subdiv, argument
843 return build_ring_all_reduce(tensors, len(tensors), subdiv, [0],
|
| /external/skqp/src/gpu/ops/ |
| D | GrAAHairLinePathRenderer.cpp | 281 int subdiv = num_quad_subdivs(devPts); in gather_lines_and_quads() local
282 SkASSERT(subdiv >= -1); in gather_lines_and_quads()
283 if (-1 == subdiv) { in gather_lines_and_quads()
300 quadSubdivCnts->push_back() = subdiv; in gather_lines_and_quads()
301 totalQuadCount += 1 << subdiv; in gather_lines_and_quads()
632 int subdiv, in add_quads() argument
636 SkASSERT(subdiv >= 0); in add_quads()
637 if (subdiv) { in add_quads()
640 add_quads(newP + 0, subdiv-1, toDevice, toSrc, vert); in add_quads()
641 add_quads(newP + 2, subdiv-1, toDevice, toSrc, vert); in add_quads()
|
| /external/skia/src/gpu/ops/ |
| D | GrAAHairLinePathRenderer.cpp | 280 int subdiv = num_quad_subdivs(devPts); in gather_lines_and_quads() local
281 SkASSERT(subdiv >= -1); in gather_lines_and_quads()
282 if (-1 == subdiv) { in gather_lines_and_quads()
299 quadSubdivCnts->push_back() = subdiv; in gather_lines_and_quads()
300 totalQuadCount += 1 << subdiv; in gather_lines_and_quads()
631 int subdiv, in add_quads() argument
635 SkASSERT(subdiv >= 0); in add_quads()
636 if (subdiv) { in add_quads()
639 add_quads(newP + 0, subdiv-1, toDevice, toSrc, vert); in add_quads()
640 add_quads(newP + 2, subdiv-1, toDevice, toSrc, vert); in add_quads()
|