| /external/tensorflow/tensorflow/python/tpu/ |
| topology_test.py | 16 """Tests for topology.py."""
23 from tensorflow.python.tpu import topology
30 original_topology = topology.Topology(
35 new_topology = topology.Topology(serialized=serialized_str)
37 # Make sure the topology recovered from serialized str is same as the
38 # original topology.
|
| device_assignment.py | 25 from tensorflow.python.tpu.topology import Topology
32 def _compute_task_and_cores_to_replicas(core_assignment, topology):
38 task_id = topology.task_ordinal_at_coordinates(coordinates)
59 """Mapping from logical cores in a computation to the physical TPU topology.
66 def __init__(self, topology, core_assignment):
70 topology: A `Topology` object that describes the physical TPU topology.
76 ValueError: If `topology` is not `Topology` object
104 def topology(self): member in class:DeviceAssignment
[all...] |
| tpu_strategy_util.py | 31 from tensorflow.python.tpu import topology
64 The tf.tpu.Topology object for the topology of the TPU cluster.
103 return topology.Topology(serialized=serialized_topology)
|
| topology.py | 15 """Defines the `Topology` class, that describes a TPU fabric topology."""
43 class Topology(object):
51 """Builds a Topology object.
53 If `serialized` is not `None`, the topology is parsed from `serialized` and
54 the other arguments are ignored. Otherwise, the topology is computed from
59 serialized proto is parsed to discover the topology.
61 the shape of the TPU topology, in number of cores. Ignored if
68 ValueError: If `serialized` does not describe a well-formed topology.
129 """Inverts a [task,device,axis] topology to [x,y,z] -> task/device maps.""
[all...] |
| /external/clang/test/SemaTemplate/ |
| nested-template.cpp | 148 template< class Topology, class ctype >
153 template< class Topology, class ctype >
155 class ReferenceElement< Topology, ctype > :: BaryCenterArray // expected-error{{out-of-line definition of 'BaryCenterArray' does not match any declaration in 'ReferenceElement<Topology, ctype>'}}
|
| /device/linaro/bootloader/edk2/OvmfPkg/Include/IndustryStandard/ |
| VirtioBlk.h | 43 VIRTIO_BLK_TOPOLOGY Topology;
|
| /external/guava/guava-tests/benchmark/com/google/common/collect/ |
| BinaryTreeTraverserBenchmark.java | 45 enum Topology {
81 * Generates a tree with topology selected uniformly at random from the topologies of binary
160 Topology topology; field in class:BinaryTreeTraverserBenchmark
177 topology.createTree(size, rng).get(),
|
| /external/skia/src/sfnt/ |
| SkPanose.h | 242 enum class Topology : SK_OT_BYTE {
406 enum class Topology : SK_OT_BYTE {
|
| /external/skqp/src/sfnt/ |
| SkPanose.h | 242 enum class Topology : SK_OT_BYTE {
406 enum class Topology : SK_OT_BYTE {
|
| /device/linaro/bootloader/edk2/OvmfPkg/VirtioBlkDxe/ |
| VirtioBlk.c | 674 Status = VIRTIO_CFG_READ (Dev, Topology.PhysicalBlockExp,
684 Status = VIRTIO_CFG_READ (Dev, Topology.AlignmentOffset, &AlignmentOffset);
689 Status = VIRTIO_CFG_READ (Dev, Topology.OptIoSize, &OptIoSize);
[all...] |
