Home
last modified time | relevance | path

Searched refs:kShapeOpInputLayout (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dmerge_clusters.cc178 src_cluster->getAttrOfType<mlir::ArrayAttr>(kShapeOpInputLayout); in MergeClusterMetadata()
183 target_cluster->getAttrOfType<mlir::ArrayAttr>(kShapeOpInputLayout); in MergeClusterMetadata()
196 target_cluster->setAttr(kShapeOpInputLayout, src_shape_layouts); in MergeClusterMetadata()
209 kShapeOpInputLayout, in MergeClusterMetadata()
H A Dspmd_expander_common.cc564 auto shape_layouts = op->getAttrOfType<mlir::ArrayAttr>(kShapeOpInputLayout); in ValidateMetadataAttributes()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Dconstants.h58 static constexpr char kShapeOpInputLayout[] = "_shape_input_layout"; variable
H A Ddtensor_device_util.h252 AddNodeAttr(kShapeOpInputLayout, {shape_metadata_layout()->ToString()}, in set_const_value()
H A Ddtensor_device_util.cc225 return ParseAttrMap(node, kShapeOpInputLayoutIndices, kShapeOpInputLayout, in ParseShapeInputLayouts()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dmeta_spmd_expander.cc1222 kShapeOpInputLayout, in ExpandOp()