Searched refs:layout_string (Results 1 – 7 of 7) sorted by relevance
58 const std::string& layout_string, in ValidateAndAssignResourceInputLayout() argument102 if (layout_string != previous_layout.str()) in ValidateAndAssignResourceInputLayout()108 resource_arg_index, layout_string); in ValidateAndAssignResourceInputLayout()112 resource_arg_index, layout_string); in ValidateAndAssignResourceInputLayout()206 std::string layout_string; in ExpandOp() local208 layout_string = resource_layout->ToString(); in ExpandOp()210 layout_string = value_layout->ToString(); in ExpandOp()212 enclosing_device_cluster, layout_string, block_arg_index, &builder)); in ExpandOp()
60 const std::string layout_string = copy_to_mesh.layout().str(); in WrapDeviceCluster() local61 auto layout_or = Layout::FromString(layout_string); in WrapDeviceCluster()65 layout_string)); in WrapDeviceCluster()
129 const std::string layout_string = in PropagateFunctionDefaultLayoutAttrToLayoutOp() local131 auto result_layout_or_status = Layout::FromString(layout_string); in PropagateFunctionDefaultLayoutAttrToLayoutOp()137 kCustomDefaultLayoutAttr, ret_index, layout_string)); in PropagateFunctionDefaultLayoutAttrToLayoutOp()
260 const std::string layout_string = in ExtractLayoutFromFunctionReturnAttr() local262 auto result_layout_or_status = Layout::FromString(layout_string); in ExtractLayoutFromFunctionReturnAttr()268 layout_string) in ExtractLayoutFromFunctionReturnAttr()
85 const string& layout_string = layouts_flat(i); in Compute() local97 const auto& layout_or = Layout::FromString(layout_string); in Compute()101 " has invalid layout string : ", layout_string))); in Compute()
309 layout_string = _pywrap_dtensor_device.FetchLayout(315 return layout_lib.Layout.from_string(layout_string)
293 std::string layout_string = in PYBIND11_MODULE() local301 PyUnicode_FromString(layout_string.c_str())); in PYBIND11_MODULE()