| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | mesh_propagation.cc | 123 absl::optional<Mesh>& operand_mesh) -> mlir::LogicalResult { in ExtractMeshFromOperand()
148 absl::optional<Mesh> mesh; in ExtractMeshFromOperand() local
162 absl::optional<Mesh> mesh; in ExtractMeshFromOperand() local
190 mlir::tf_device::ClusterOp cluster, absl::optional<Mesh>* mesh, in InferMeshFromInputs()
210 auto mesh = layout_op.layout().mesh(); in InferMeshFromInputs() local
269 mlir::tf_device::ClusterOp cluster, absl::optional<Mesh>* mesh, in InferMeshFromConsumers()
291 auto mesh = status_or_mesh.ValueOrDie(); in InferMeshFromConsumers() local
383 const Mesh& mesh, in AnnotateFunctionArgumentsWithMeshInformation()
515 absl::optional<mlir::StringAttr> mesh; in PropagateDefaultMeshToUnAssignedClusters() local
529 const auto& mesh = mesh_or_status.ValueOrDie(); in PropagateDefaultMeshToUnAssignedClusters() local
|
| D | collectives.cc | 337 int GetMeshDimensionOffsetWithNeighbor(const Mesh& mesh, in GetMeshDimensionOffsetWithNeighbor()
350 StatusOr<int> GetMeshCoordinateIndex(const Mesh& mesh, in GetMeshCoordinateIndex()
361 StatusOr<mlir::Value> CreateConstSrcTargetPair(const Mesh& mesh, in CreateConstSrcTargetPair()
439 const Mesh& mesh = layout.mesh(); in EmitHaloExchange() local
|
| D | collectives_common.cc | 76 StatusOr<std::string> DeviceTypeFromMesh(const Mesh& mesh) { in DeviceTypeFromMesh()
|
| D | undo_merge_const_across_mesh.cc | 53 const Mesh& mesh = layout.mesh(); in runOnOperation() local
|
| D | merge_clusters.cc | 265 void CloneEmptyIfWithPredicate(mlir::TF::IfRegionOp if_region, const Mesh& mesh, in CloneEmptyIfWithPredicate()
490 Mesh mesh; in MergeClusters() local
510 for (const Mesh& mesh : meshes) { in MergeClusters() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/dtensor_dialect/ir/ |
| D | dtensor_attributes.cc | 38 const Mesh& mesh = key; in hashKey() local
46 Mesh mesh; member
50 MeshAttr MeshAttr::get(MLIRContext* context, const Mesh& mesh) { in get()
|
| /external/rust/crates/plotters/src/chart/ |
| D | mod.rs | 19 mod mesh; module
|
| /external/deqp/external/vulkancts/modules/vulkan/mesh_shader/ |
| D | vktMeshShaderBuiltinTestsEXT.cpp | 687 std::ostringstream mesh; in initPrograms() local
845 std::ostringstream mesh; in initPrograms() local
980 std::ostringstream mesh; in initPrograms() local
1124 std::ostringstream mesh; in initPrograms() local
1201 std::ostringstream mesh; in initPrograms() local
1280 std::ostringstream mesh; in initPrograms() local
1389 std::ostringstream mesh; in initPrograms() local
1542 std::ostringstream mesh; in initPrograms() local
1657 std::ostringstream mesh; in initPrograms() local
1806 std::ostringstream mesh; in initPrograms() local
[all …]
|
| D | vktMeshShaderBuiltinTests.cpp | 556 std::ostringstream mesh; in initPrograms() local
712 std::ostringstream mesh; in initPrograms() local
837 std::ostringstream mesh; in initPrograms() local
978 std::ostringstream mesh; in initPrograms() local
1054 std::ostringstream mesh; in initPrograms() local
1128 std::ostringstream mesh; in initPrograms() local
1238 std::ostringstream mesh; in initPrograms() local
1396 std::ostringstream mesh; in initPrograms() local
1507 std::ostringstream mesh; in initPrograms() local
1612 std::ostringstream mesh; in initPrograms() local
[all …]
|
| D | vktMeshShaderSmokeTests.cpp | 158 std::ostringstream mesh; in initPrograms() local
228 std::ostringstream mesh; in initPrograms() local
297 std::ostringstream mesh; in initPrograms() local
566 std::ostringstream mesh; in initGradientPrograms() local
|
| D | vktMeshShaderMiscTestsEXT.cpp | 514 std::ostringstream mesh; in initPrograms() local
650 std::ostringstream mesh; in initPrograms() local
719 std::ostringstream mesh; in initPrograms() local
793 std::ostringstream mesh; in initPrograms() local
867 std::ostringstream mesh; in initPrograms() local
964 std::ostringstream mesh; in initPrograms() local
1081 std::ostringstream mesh; in initPrograms() local
1249 std::ostringstream mesh; in initPrograms() local
1366 std::ostringstream mesh; in initPrograms() local
1449 std::ostringstream mesh; in initPrograms() local
[all …]
|
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | save_restore_util.cc | 56 const auto& mesh = layout.mesh(); in BuildSliceSpecDeviceMap() local
165 const Layout& layout, const Mesh& mesh, const DeviceLocation& device_coords, in SliceSpecOnDevice()
|
| D | dtensor_device_util.cc | 59 const MeshWithParallelDevice& mesh, in BroadcastTensorHandleToParallelTensor()
94 const MeshWithParallelDevice& mesh, const std::string& dtensor_device_name, in BroadcastResourceTensor()
274 const MeshWithParallelDevice& mesh, const std::string& dtensor_device_name, in Broadcast()
333 const MeshWithParallelDevice& mesh, const Layout& layout) { in Wrap()
348 const MeshWithParallelDevice& mesh, const Layout& layout) { in Dummy()
437 const MeshWithParallelDevice& mesh, const Layout& layout, in Wrap()
856 Mesh mesh; in IdentifyAllFunctionsToExecute() local
|
| D | tensor_layout.h | 257 const Mesh& mesh() const { return mesh_; } in mesh() function
262 void set_mesh(Mesh mesh) { mesh_ = mesh; } in set_mesh()
|
| D | tensor_layout.cc | 61 inline std::vector<DeviceLocation> ComputeDeviceLocations(const Mesh* mesh) { in ComputeDeviceLocations()
188 Mesh mesh; in ParseFromProto() local
230 Mesh mesh; in GetAbstractMesh() local
662 const std::vector<std::string>& sharding_spec_strs, const Mesh& mesh) { in GetLayout()
675 const std::vector<ShardingSpec>& sharding_specs, const Mesh& mesh) { in GetLayout()
795 StatusOr<int> IndexOfMeshDimension(const Mesh& mesh, in IndexOfMeshDimension()
1025 Layout Layout::ReplicatedOnMesh(const Mesh& mesh, int rank) { in ReplicatedOnMesh()
1030 Layout Layout::AnyOnMesh(const Mesh& mesh, int rank) { in AnyOnMesh()
|
| D | dtensor_device_util.h | 310 const MeshWithParallelDevice& mesh() const { return mesh_; } in mesh() function
429 const MeshWithParallelDevice& mesh, const Layout& layout, in ResourceHandleWithLayout()
461 const MeshWithParallelDevice& mesh, const Layout& layout) { in Dummy()
506 : TensorWithLayout(nullptr, mesh, layout, local_shape), in TensorWithLayout() argument
|
| D | dtensor_sharded_prefix_kernel.cc | 64 const Mesh& mesh = *mesh_or; in Compute() local
|
| /external/oss-fuzz/projects/alembic/ |
| D | alembic_dump_info_fuzzer.cc | 93 const IPolyMesh mesh(node.getParent(), header.getName()); in dumpPolyMesh() local
102 const ISubD mesh(node.getParent(), header.getName()); in dumpSubD() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | top_k_spmd_expander.cc | 101 const Mesh& mesh = input_layout.mesh(); in ComputeLayoutBackward() local
|
| D | dataparallel_spmd_expander.cc | 96 const llvm::DenseMap<int, int>& batchable_args, const Mesh& mesh) { in MergeBatchLayouts()
152 const llvm::DenseMap<int, int>& batchable_outputs, const Mesh& mesh) { in IntermediateBatchLayout()
|
| D | gather_spmd_expander.cc | 307 int index_dimensions, const Mesh& mesh) { in GatherNdGetOutputLayoutFromInput()
342 const Mesh& mesh) { in GatherNdGetInputLayoutFromOutput()
|
| /external/tensorflow/tensorflow/dtensor/tests/ |
| D | tensor_layout_test.cc | 71 Mesh mesh = Mesh::Empty(); in TEST_F() local
115 Mesh mesh = BatchLayout().mesh(); in TEST_F() local
122 Mesh mesh = BatchLayout().mesh(); in TEST_F() local
540 auto mesh = Mesh::Empty(); in TEST_F() local
545 Mesh mesh = Mesh::FromString("mesh:|x=2,batch=1|*TPU").ValueOrDie(); in TEST_F() local
|
| /external/skia/src/gpu/ganesh/ops/ |
| D | GrMeshDrawOp.h | 67 GrSimpleMesh* mesh() { return fMesh; } in mesh() function
|
| /external/skia/src/gpu/ganesh/geometry/ |
| D | GrAATriangulator.cpp | 112 void GrAATriangulator::connectPartners(VertexList* mesh, const Comparator& c) { in connectPartners()
248 void GrAATriangulator::Event::apply(VertexList* mesh, const Comparator& c, EventList* events, in apply()
311 bool GrAATriangulator::collapseOverlapRegions(VertexList* mesh, const Comparator& c, in collapseOverlapRegions()
622 std::tuple<Poly*, bool> GrAATriangulator::tessellate(const VertexList& mesh, const Comparator& c) { in tessellate()
|
| D | GrTriangulator.cpp | 997 void GrTriangulator::mergeVertices(Vertex* src, Vertex* dst, VertexList* mesh, in mergeVertices()
1015 Vertex* GrTriangulator::makeSortedVertex(const SkPoint& p, uint8_t alpha, VertexList* mesh, in makeSortedVertex()
1086 Vertex** current, VertexList* mesh, in checkForIntersection()
1169 bool GrTriangulator::mergeCoincidentVertices(VertexList* mesh, const Comparator& c) const { in mergeCoincidentVertices()
1190 void GrTriangulator::buildEdges(VertexList* contours, int contourCnt, VertexList* mesh, in buildEdges()
1325 GrTriangulator::SimplifyResult GrTriangulator::simplify(VertexList* mesh, in simplify()
1508 void GrTriangulator::contoursToMesh(VertexList* contours, int contourCnt, VertexList* mesh, in contoursToMesh()
1547 VertexList mesh; in contoursToPolys() local
|