| /external/tensorflow/tensorflow/dtensor/python/ |
| D | save_restore.py | 35 mesh: layout_lib.Mesh,
84 mesh: layout_lib.Mesh,
154 mesh: layout_lib.Mesh,
226 def name_based_save(mesh: layout_lib.Mesh, checkpoint_prefix: Union[str,
|
| D | d_checkpoint.py | 48 def __init__(self, mesh: layout.Mesh,
169 def __init__(self, mesh: layout.Mesh, **kwargs):
228 def __init__(self, mesh: layout.Mesh, graph_view):
373 def __init__(self, mesh: layout.Mesh, root=None, **kwargs):
|
| D | dtensor_device.py | 51 def __init__(self, meshes: List[layout_lib.Mesh], is_async=True):
120 def _create_embedding_host_mesh(self, tpu_mesh: layout_lib.Mesh):
153 def _register_mesh(self, mesh: layout_lib.Mesh):
382 def _experimental_default_mesh(self, mesh: layout_lib.Mesh):
|
| D | layout.py | 50 class Mesh(object): class
527 def __init__(self, sharding_specs: List[str], mesh: Mesh):
587 def batch_sharded(mesh: Mesh, batch_dim: str, rank: int) -> 'Layout':
631 def inner_sharded(mesh: Mesh, inner_dim: str, rank: int) -> 'Layout':
677 def replicated(mesh: Mesh, rank: int) -> 'Layout':
|
| D | mesh_util.py | 34 device_type: str, mesh: layout.Mesh) -> None:
274 def barrier(mesh: layout.Mesh, barrier_name: Optional[str] = None):
|
| D | tpu_util.py | 748 def get_device_ids(mesh: layout_lib.Mesh,
777 mesh: layout_lib.Mesh,
|
| D | api.py | 86 def run_on(mesh: layout_lib.Mesh):
|
| D | input_util.py | 340 mesh: layout_lib.Mesh,
|
| /external/tensorflow/tensorflow/dtensor/mlir/dtensor_dialect/ir/ |
| D | dtensor_attributes.h | 38 using Mesh = tensorflow::dtensor::Mesh; variable
53 using Mesh = tensorflow::dtensor::Mesh; variable
|
| D | dtensor_attributes.cc | 30 using Mesh = tensorflow::dtensor::Mesh; typedef
|
| D | ops.cc | 70 using Mesh = tensorflow::dtensor::Mesh; in ParseMeshAttr() typedef
|
| /external/exoplayer/tree_15dc86382f17a24a3e881e52e31a810c1ea44b49/library/core/src/main/java/com/google/android/exoplayer2/video/spherical/ |
| D | Projection.java | 219 public static final class Mesh { class in Projection
222 public Mesh(SubMesh... subMeshes) { in Mesh() method in Projection.Mesh
|
| /external/exoplayer/tree_8e57d3715f9092d5ec54ebe2e538f34bfcc34479/library/core/src/main/java/com/google/android/exoplayer2/video/spherical/ |
| D | Projection.java | 223 public static final class Mesh { class in Projection
226 public Mesh(SubMesh... subMeshes) { in Mesh() method in Projection.Mesh
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | move_compilation_to_host.cc | 54 std::map<Mesh, mlir::TF::StatefulPartitionedCallOp>* function_map) { in IdentifyAndValidateMeshComputations()
212 std::map<Mesh, mlir::TF::StatefulPartitionedCallOp>& computation_map, in HandleCompilationOps()
|
| /external/skia/src/gpu/ganesh/ops/ |
| D | DrawMeshOp.cpp | 509 class Mesh { class in __anon19f680500111::MeshOp
616 MeshOp::Mesh::Mesh(const SkMesh& mesh) { in Mesh() function in __anon19f680500111::MeshOp::Mesh
643 MeshOp::Mesh::Mesh(sk_sp<SkVertices> vertices, const SkMatrix& viewMatrix) in Mesh() function in __anon19f680500111::MeshOp::Mesh
648 MeshOp::Mesh::Mesh(Mesh&& that) { in Mesh() function in __anon19f680500111::MeshOp::Mesh
|