Home
last modified time | relevance | path

Searched defs:layout_string (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dresource_spmd_expander.cc58 const std::string& layout_string, in ValidateAndAssignResourceInputLayout()
206 std::string layout_string; in ExpandOp() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dop_to_device_cluster.cc60 const std::string layout_string = copy_to_mesh.layout().str(); in WrapDeviceCluster() local
H A Dpropagate_default_layout.cc129 const std::string layout_string = in PropagateFunctionDefaultLayoutAttrToLayoutOp() local
H A Dlayout_parsing.cc260 const std::string layout_string = in ExtractLayoutFromFunctionReturnAttr() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_sharded_prefix_kernel.cc85 const string& layout_string = layouts_flat(i); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/python/
H A Dpywrap_dtensor_device.cc293 std::string layout_string = in PYBIND11_MODULE() local