Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_graph_dumper.cc1180 string instr_shape; in GetInstructionNodeExtraInfo() local
1182 instr_shape = ShapeUtil::HumanStringWithLayout(instr->shape()); in GetInstructionNodeExtraInfo()
1184 instr_shape = ShapeUtil::HumanString(instr->shape()); in GetInstructionNodeExtraInfo()
1190 if (instr_shape.length() > kMaxShapeLen) { in GetInstructionNodeExtraInfo()
1191 instr_shape = StrCat( in GetInstructionNodeExtraInfo()
1192 absl::string_view(instr_shape).substr(0, kMaxShapeLen - 3), "..."); in GetInstructionNodeExtraInfo()
1194 lines.push_back(HtmlLikeStringSanitize(instr_shape)); in GetInstructionNodeExtraInfo()
Dlayout_assignment.cc814 const Shape& instr_shape = in CreateCopyWithNewLayout() local
817 HloInstruction::CreateGetTupleElement(instr_shape, instruction, i)); in CreateCopyWithNewLayout()
820 instr_shape)) { in CreateCopyWithNewLayout()