| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | hierarchical_tree_broadcaster.cc | 44 string BroadcastBufKey(const string& exec_key, int subdiv, int src_rank, in BroadcastBufKey() argument
47 return strings::StrCat("broadcast(", exec_key, "):subdiv(", subdiv, in BroadcastBufKey()
51 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 | 59 static int TreeRecvFrom(const CollectiveParams& cp, int subdiv);
63 static void TreeSendTo(const CollectiveParams& cp, int subdiv,
72 void DispatchSend(int subdiv, int dst_rank, int src_rank,
78 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/v1/ |
| D | all_reduce_test.py | 144 def _buildRing(self, num_workers, num_gpus, subdiv): argument
147 x, num_workers, subdiv, gpu_perm, math_ops.add, un_op)
165 def _testRingAllReduce(self, num_workers, num_gpus, shape, subdiv): argument
167 build_f = self._buildRing(num_workers, num_gpus, subdiv)
172 (num_workers, num_gpus, shape, subdiv, elapsed))
|
| D | all_reduce.py | 779 def build_nccl_then_ring(input_tensors, subdiv, red_op, un_op=None): argument
782 return build_ring_all_reduce(y, len(y), subdiv, [0], red_op, un_op)
845 def build_shuffle_then_ring(input_tensors, gather_devices, subdiv, argument
849 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 | 282 int subdiv = num_quad_subdivs(devPts); in gather_lines_and_quads() local
283 SkASSERT(subdiv >= -1); in gather_lines_and_quads()
284 if (-1 == subdiv) { in gather_lines_and_quads()
301 quadSubdivCnts->push_back() = subdiv; in gather_lines_and_quads()
302 totalQuadCount += 1 << subdiv; in gather_lines_and_quads()
634 int subdiv, in add_quads() argument
638 SkASSERT(subdiv >= 0); in add_quads()
648 int stepCount = 1 << subdiv; in add_quads()
|