Searched defs:layout_str (Results 1 – 8 of 8) sorted by relevance
78 std::string layout_str = layout.ToString(); in TEST_F() local86 std::string layout_str = layout.ToString(); in TEST_F() local93 std::string layout_str = "sharding_specs:x," + string(Layout::kUnshardedDim) + in TEST_F() local99 std::string layout_str = in TEST_F() local105 std::string layout_str = "sharding_specs:x, mesh:|x=2,y=2|*CPU"; in TEST_F() local
62 const std::string layout_str = relayout.layout().str(); in PropagateDTensorLayoutForRelayout() local95 const auto layout_str = layout_attr.getValue().str(); in PropagateFunctionArgAttrToLayoutOp() local
99 std::string layout_str = spec.str(); in ParseLayoutAttr() local
1045 StatusOr<Layout> Layout::FromString(std::string layout_str) { in FromString()1085 std::string layout_str = "sharding_specs:"; in ToString() local
577 absl::string_view layout_str = in FetchLayoutFromAttributes() local
340 const std::string layout_str = "layout"; in iterate() local
694 std::string layout_str = LayoutUtil::HumanString(shape.layout()); in HumanStringWithLayout() local
524 const char *layout_str; in drmGetFormatModifierNameFromAmlogic() local