| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | tensor_layout.h | 89 class Mesh {
95 static Mesh Empty();
97 Mesh() = default;
100 static StatusOr<Mesh> ParseFromProto(const MeshProto& proto);
108 static StatusOr<Mesh> FromString(const std::string& str);
114 static StatusOr<Mesh> GetAbstractMesh(
117 static StatusOr<Mesh> GetMesh(
146 StatusOr<Mesh> ToDeviceType(const std::string& device_type) const;
190 bool operator==(const Mesh& b) const;
191 bool operator!=(const Mesh& b) const { return !((*this) == b); }
[all …]
|
| D | tensor_layout.cc | 52 constexpr const char* Mesh::kEmptyMeshString;
61 inline std::vector<DeviceLocation> ComputeDeviceLocations(const Mesh* mesh) { in ComputeDeviceLocations()
175 std::map<std::string, std::vector<int>>& Mesh::tpu_core_ids() { in tpu_core_ids()
181 std::string& Mesh::tpu_host_mesh() { in tpu_host_mesh()
187 StatusOr<Mesh> Mesh::ParseFromProto(const MeshProto& proto) { in ParseFromProto()
188 Mesh mesh; in ParseFromProto()
228 StatusOr<Mesh> Mesh::GetAbstractMesh( in GetAbstractMesh()
230 Mesh mesh; in GetAbstractMesh()
251 StatusOr<Mesh> Mesh::GetMesh(const std::string& name, in GetMesh()
257 TF_ASSIGN_OR_RETURN(Mesh mesh, GetAbstractMesh(name, mesh_dims)); in GetMesh()
[all …]
|
| /external/tensorflow/tensorflow/dtensor/mlir/dtensor_dialect/ir/ |
| D | dtensor_attributes.cc | 30 using Mesh = tensorflow::dtensor::Mesh; typedef
31 using KeyTy = Mesh;
33 explicit MeshAttrStorage(Mesh mesh) : mesh(std::move(mesh)) {} in MeshAttrStorage()
38 const Mesh& mesh = key; in hashKey()
46 Mesh mesh;
50 MeshAttr MeshAttr::get(MLIRContext* context, const Mesh& mesh) { in get()
54 const MeshAttr::Mesh& MeshAttr::getValue() const { return getImpl()->mesh; } in getValue()
|
| D | dtensor_attributes.h | 38 using Mesh = tensorflow::dtensor::Mesh; variable
41 static MeshAttr get(MLIRContext* context, const Mesh& mesh);
44 const Mesh& getValue() const;
53 using Mesh = tensorflow::dtensor::Mesh; variable
|
| D | ops.cc | 70 using Mesh = tensorflow::dtensor::Mesh; in ParseMeshAttr() typedef
71 using MeshOr = tensorflow::dtensor::StatusOr<Mesh>; in ParseMeshAttr()
72 MeshOr mesh_or = Mesh::FromString(mesh_str); in ParseMeshAttr()
|
| /external/tensorflow/tensorflow/dtensor/tests/ |
| D | tensor_layout_test.cc | 71 Mesh mesh = Mesh::Empty(); in TEST_F()
73 EXPECT_EQ(mesh_str, Mesh::kEmptyMeshString); in TEST_F()
115 Mesh mesh = BatchLayout().mesh(); in TEST_F()
118 EqualsProto(Mesh::FromString(mesh_str).ValueOrDie().ToProto())); in TEST_F()
122 Mesh mesh = BatchLayout().mesh(); in TEST_F()
220 Mesh reduced_mesh = replicated_layout.ReducedMesh(); in TEST_F()
227 Mesh original_mesh = layout.mesh(); in TEST_F()
228 Mesh reduced_mesh = layout.ReducedMesh(); in TEST_F()
241 Mesh reduced_mesh = layout->ReducedMesh(); in TEST_F()
256 Mesh reduced_mesh = layout->ReducedMesh(); in TEST_F()
[all …]
|
| /external/exoplayer/tree_15dc86382f17a24a3e881e52e31a810c1ea44b49/library/core/src/main/java/com/google/android/exoplayer2/video/spherical/ |
| D | Projection.java | 148 return new Projection(new Mesh(subMesh), stereoMode); in createEquirectangular()
152 public final Mesh leftMesh;
157 public final Mesh rightMesh;
169 public Projection(Mesh mesh, int stereoMode) { in Projection()
181 public Projection(Mesh leftMesh, Mesh rightMesh, int stereoMode) { in Projection()
219 public static final class Mesh { class in Projection
222 public Mesh(SubMesh... subMeshes) { in Mesh() method in Projection.Mesh
|
| D | ProjectionDecoder.java | 23 import com.google.android.exoplayer2.video.spherical.Projection.Mesh;
65 ArrayList<Mesh> meshes = null; in decode()
94 private static @Nullable ArrayList<Mesh> parseProj(ParsableByteArray input) { in parseProj()
115 private static @Nullable ArrayList<Mesh> parseMshp(ParsableByteArray input) { in parseMshp()
140 private static @Nullable ArrayList<Mesh> parseRawMshpData(ParsableByteArray input) { in parseRawMshpData()
141 ArrayList<Mesh> meshes = new ArrayList<>(); in parseRawMshpData()
151 Mesh mesh = parseMesh(input); in parseRawMshpData()
163 private static @Nullable Mesh parseMesh(ParsableByteArray input) { in parseMesh()
228 return new Mesh(subMeshes); in parseMesh()
|
| /external/exoplayer/tree_8e57d3715f9092d5ec54ebe2e538f34bfcc34479/library/core/src/main/java/com/google/android/exoplayer2/video/spherical/ |
| D | Projection.java | 152 return new Projection(new Mesh(subMesh), stereoMode); in createEquirectangular()
156 public final Mesh leftMesh;
161 public final Mesh rightMesh;
173 public Projection(Mesh mesh, int stereoMode) { in Projection()
185 public Projection(Mesh leftMesh, Mesh rightMesh, int stereoMode) { in Projection()
223 public static final class Mesh { class in Projection
226 public Mesh(SubMesh... subMeshes) { in Mesh() method in Projection.Mesh
|
| D | ProjectionDecoder.java | 23 import com.google.android.exoplayer2.video.spherical.Projection.Mesh;
66 ArrayList<Mesh> meshes = null; in decode()
96 private static ArrayList<Mesh> parseProj(ParsableByteArray input) { in parseProj()
118 private static ArrayList<Mesh> parseMshp(ParsableByteArray input) { in parseMshp()
144 private static ArrayList<Mesh> parseRawMshpData(ParsableByteArray input) { in parseRawMshpData()
145 ArrayList<Mesh> meshes = new ArrayList<>(); in parseRawMshpData()
155 Mesh mesh = parseMesh(input); in parseRawMshpData()
168 private static Mesh parseMesh(ParsableByteArray input) { in parseMesh()
233 return new Mesh(subMeshes); in parseMesh()
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | mesh_propagation.cc | 56 absl::optional<Mesh>* out) { in ExtractMeshFromBlockArgument()
77 Mesh::FromString(mesh_attr.getValue().str()); in ExtractMeshFromBlockArgument()
90 absl::optional<Mesh>* out) { in ExtractMeshFromOpOutput()
118 mlir::OpOperand* operand, absl::optional<Mesh>* out) { in ExtractMeshFromOperand()
122 [](mlir::Location loc, absl::optional<Mesh>& mesh, in ExtractMeshFromOperand()
123 absl::optional<Mesh>& operand_mesh) -> mlir::LogicalResult { in ExtractMeshFromOperand()
145 absl::optional<Mesh> operand_mesh; in ExtractMeshFromOperand()
148 absl::optional<Mesh> mesh; in ExtractMeshFromOperand()
162 absl::optional<Mesh> mesh; in ExtractMeshFromOperand()
190 mlir::tf_device::ClusterOp cluster, absl::optional<Mesh>* mesh, in InferMeshFromInputs()
[all …]
|
| D | merge_clusters.cc | 64 Mesh* mesh_output) { in ExtractMeshFromCluster()
68 const absl::optional<Mesh>& mesh_or_null = *mesh_or_status; in ExtractMeshFromCluster()
230 Mesh cluster_mesh; in InlineNestedDeviceClusters()
234 Mesh parent_cluster_mesh; in InlineNestedDeviceClusters()
265 void CloneEmptyIfWithPredicate(mlir::TF::IfRegionOp if_region, const Mesh& mesh, in CloneEmptyIfWithPredicate()
366 Mesh nested_mesh; in DecomposeIf()
487 std::map<Mesh, llvm::SmallVector<mlir::tf_device::ClusterOp, 4>> cluster_map; in MergeClusters()
488 std::vector<Mesh> meshes; in MergeClusters()
490 Mesh mesh; in MergeClusters()
504 std::sort(meshes.begin(), meshes.end(), [](const Mesh& a, const Mesh& b) { in MergeClusters()
[all …]
|
| D | lower_send_recv.cc | 50 Mesh* mesh_output) { in ExtractMeshFromCluster()
54 const absl::optional<Mesh>& mesh_or_null = *mesh_or_status; in ExtractMeshFromCluster()
81 const Mesh recv_mesh = dtensor_recv.layout().mesh(); in LowerDTensorSendRecvsOps()
83 Mesh send_mesh; in LowerDTensorSendRecvsOps()
|
| D | tpu_integration.cc | 56 void AddMetadataToTPUCluster(const Mesh& mesh_config, in AddMetadataToTPUCluster()
75 mlir::ModuleOp module, llvm::SmallVectorImpl<Mesh>* tpu_meshs, in IdentifyTPUFunctions()
81 auto mesh_or_status = Mesh::FromString(string(call.config())); in IdentifyTPUFunctions()
141 llvm::SmallVector<Mesh, 4> tpu_meshes; in runOnOperation()
|
| D | move_compilation_to_host.cc | 54 std::map<Mesh, mlir::TF::StatefulPartitionedCallOp>* function_map) { in IdentifyAndValidateMeshComputations() argument
62 const Mesh& computation_mesh = mesh_or->value(); in IdentifyAndValidateMeshComputations()
87 const Mesh& mesh, mlir::ModuleOp module, mlir::func::FuncOp function, in CreateSendRecvOpsToTransferProgramKey()
178 const Mesh& receiving_function_mesh;
188 const Mesh& host_mesh, mlir::ModuleOp module, in SendRecvCompilationKey()
212 std::map<Mesh, mlir::TF::StatefulPartitionedCallOp>& computation_map, in HandleCompilationOps() argument
225 Mesh host_mesh; in HandleCompilationOps()
382 std::map<Mesh, mlir::TF::StatefulPartitionedCallOp> computation_map; in runOnOperation()
|
| D | layout_parsing.cc | 136 StatusOr<Mesh> ExtractDeviceMeshEnclosingCluster(mlir::Operation* op) { in ExtractDeviceMeshEnclosingCluster()
149 StatusOr<absl::optional<Mesh>> ExtractDeviceMeshFromOp(mlir::Operation* op) { in ExtractDeviceMeshFromOp()
150 absl::optional<Mesh> extracted_mesh; in ExtractDeviceMeshFromOp()
156 TF_ASSIGN_OR_RETURN(Mesh mesh, in ExtractDeviceMeshFromOp()
157 Mesh::FromString(mesh_str_attr.getValue().str())); in ExtractDeviceMeshFromOp()
|
| D | handle_cross_cluster_dependencies.cc | 56 Mesh* mesh_output) { in ExtractMeshFromCluster()
141 Mesh mesh; in CloneConstantsAcrossMesh()
157 Mesh input_mesh; in CloneConstantsAcrossMesh()
269 Mesh current_mesh; in ReplaceCopyToMeshWithVirtualSendRecv()
288 Mesh input_mesh; in ReplaceCopyToMeshWithVirtualSendRecv()
|
| D | undo_merge_const_across_mesh.cc | 45 llvm::SmallVector<Mesh> known_meshes; in runOnOperation()
53 const Mesh& mesh = layout.mesh(); in runOnOperation()
|
| /external/skia/src/gpu/ganesh/ops/ |
| D | DrawMeshOp.cpp | 509 class Mesh { class in __anon19f680500111::MeshOp
511 Mesh() = delete;
512 explicit Mesh(const SkMesh& mesh);
513 Mesh(sk_sp<SkVertices>, const SkMatrix& viewMatrix);
514 Mesh(const Mesh&) = delete;
515 Mesh(Mesh&& m);
517 Mesh& operator=(const Mesh&) = delete;
518 Mesh& operator=(Mesh&&) = delete; // not used by SkSTArray but could be implemented.
520 ~Mesh();
603 SkSTArray<1, Mesh> fMeshes;
[all …]
|
| /external/tensorflow/tensorflow/dtensor/mlir/utils/ |
| D | update_tpu_metadata.cc | 76 mlir::LogicalResult UpdateTPUCompileMetadata(const Mesh& mesh_config, in UpdateTPUCompileMetadata()
112 if (!Mesh::tpu_core_ids().empty() && in UpdateTPUCompileMetadata()
115 if (Mesh::tpu_core_ids().count(mesh_name) == 0) { in UpdateTPUCompileMetadata()
121 const std::vector<int>& tpu_core_ids = Mesh::tpu_core_ids()[mesh_name]; in UpdateTPUCompileMetadata()
158 auto mesh_or_status = Mesh::FromString(call_config.str()); in runOnOperation()
|
| /external/tensorflow/tensorflow/dtensor/python/ |
| D | mesh_util.py | 34 device_type: str, mesh: layout.Mesh) -> None: argument
74 device_type: Optional[str] = None) -> layout.Mesh:
114 mesh = layout.Mesh(
133 device_type: Optional[str] = None) -> layout.Mesh:
199 mesh = layout.Mesh(
274 def barrier(mesh: layout.Mesh, barrier_name: Optional[str] = None): argument
|
| D | layout.py | 50 class Mesh(object): class
300 return Mesh(dims, global_device_ids, local_device_ids, local_devices, name,
342 return Mesh.from_proto(mesh_proto)
363 h_mesh = Mesh(
527 def __init__(self, sharding_specs: List[str], mesh: Mesh): argument
540 if not isinstance(mesh, Mesh):
587 def batch_sharded(mesh: Mesh, batch_dim: str, rank: int) -> 'Layout': argument
609 mesh = Mesh.from_proto(layout_proto.mesh_config)
626 mesh = Mesh.from_string(mesh_str)
631 def inner_sharded(mesh: Mesh, inner_dim: str, rank: int) -> 'Layout': argument
[all …]
|
| D | dtensor_device.py | 51 def __init__(self, meshes: List[layout_lib.Mesh], is_async=True): argument
61 if any(not isinstance(mesh, layout_lib.Mesh) for mesh in meshes):
120 def _create_embedding_host_mesh(self, tpu_mesh: layout_lib.Mesh): argument
147 return layout_lib.Mesh(
153 def _register_mesh(self, mesh: layout_lib.Mesh): argument
179 def meshes(self) -> Set[layout_lib.Mesh]:
382 def _experimental_default_mesh(self, mesh: layout_lib.Mesh): argument
|
| D | save_restore.py | 35 mesh: layout_lib.Mesh, argument
84 mesh: layout_lib.Mesh, argument
154 mesh: layout_lib.Mesh, argument
226 def name_based_save(mesh: layout_lib.Mesh, checkpoint_prefix: Union[str, argument
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | dtensor_op_spmd_expander.cc | 75 const Mesh& send_mesh = send_layout.mesh(); in ValidateSendRecvLayoutConfiguration()
76 const Mesh& recv_mesh = recv_layout.mesh(); in ValidateSendRecvLayoutConfiguration()
123 bool SendRecvOpUsesXla(const Mesh& send_mesh, const Mesh& recv_mesh) { in SendRecvOpUsesXla()
240 const Mesh& send_mesh = send_layout.mesh(); in IsOneToOneHostMeshTransfer()
241 const Mesh& recv_mesh = recv_layout.mesh(); in IsOneToOneHostMeshTransfer()
307 TF_ASSIGN_OR_RETURN(absl::optional<Mesh> mesh, in ExpandOp()
346 TF_ASSIGN_OR_RETURN(const Mesh recv_mesh, in ExpandOp()
403 TF_ASSIGN_OR_RETURN(const Mesh send_mesh, in ExpandOp()
411 const Mesh& recv_mesh = recv_layout.mesh(); in ExpandOp()
|