Home
last modified time | relevance | path

Searched refs:layout_string (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_tfgraph_builder.cc162 string layout_string; in SetNodeAttrs() local
166 layout_string = ShapeUtil::HumanStringWithLayout(instruction->shape()); in SetNodeAttrs()
168 layout_string = StrCat( in SetNodeAttrs()
171 attrs["layout"].set_s(layout_string); in SetNodeAttrs()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.cc526 string layout_string; in ParseShapeStringInternal() local
535 &layout_string)) { in ParseShapeStringInternal()
573 if (format_string.empty() && layout_string.empty()) { in ParseShapeStringInternal()
577 TF_ASSIGN_OR_RETURN(int64 max_elements, string_to_int64(layout_string)); in ParseShapeStringInternal()
583 comma_list_to_int64s(layout_string)); in ParseShapeStringInternal()
589 << format_string << "\", layout: \"" << layout_string << "\""; in ParseShapeStringInternal()