Home
last modified time | relevance | path

Searched defs:layout_attr (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dhandle_cross_cluster_dependencies.cc89 const std::string layout_attr = copy_to_mesh.layout().str(); in CloneOpToCluster() local
196 const std::string layout_attr = copy_to_mesh.layout().str(); in LowerToSendRecv() local
H A Dpropagate_default_layout.cc92 auto layout_attr = function.getArgAttrOfType<mlir::StringAttr>( in PropagateFunctionArgAttrToLayoutOp() local
H A Dlayout_parsing.cc199 auto layout_attr = func_op.getArgAttrOfType<mlir::StringAttr>( in ExtractLayoutFromOperand() local
H A Dmerge_clusters.cc88 mlir::ArrayAttr layout_attr, mlir::DenseIntElementsAttr indices_attr2, in MergeAttributes()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Ddtensor_op_spmd_expander.cc148 mlir::StringRef layout_attr = relayout.layout(); in ComputeRelayoutLayout() local
176 mlir::StringRef layout_attr = relayout.layout(); in ExpandOp() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/dtensor_dialect/ir/
H A Dops.cc149 if (auto layout_attr = attr.dyn_cast<LayoutAttr>()) in printAttribute() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device_util.cc191 absl::string_view layout_attr, in ParseAttrMap()