Home
last modified time | relevance | path

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

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