Home
last modified time | relevance | path

Searched refs:ExtractLayoutFromFunctionReturnAttr (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dlayout_parsing.h86 StatusOr<absl::optional<Layout>> ExtractLayoutFromFunctionReturnAttr(
H A Dlayout_parsing.cc249 StatusOr<absl::optional<Layout>> ExtractLayoutFromFunctionReturnAttr( in ExtractLayoutFromFunctionReturnAttr() function
H A Dmesh_propagation.cc248 TF_ASSIGN_OR_RETURN(auto layout, ExtractLayoutFromFunctionReturnAttr( in ExtractMeshFromFuctionOutput()