Searched refs:instr_shape (Results 1 – 2 of 2) sorted by relevance
1142 string instr_shape; in GetInstructionNodeExtraInfo() local1144 instr_shape = ShapeUtil::HumanStringWithLayout(instr->shape()); in GetInstructionNodeExtraInfo()1146 instr_shape = ShapeUtil::HumanString(instr->shape()); in GetInstructionNodeExtraInfo()1152 if (instr_shape.length() > kMaxShapeLen) { in GetInstructionNodeExtraInfo()1153 instr_shape = StrCat( in GetInstructionNodeExtraInfo()1154 absl::string_view(instr_shape).substr(0, kMaxShapeLen - 3), "..."); in GetInstructionNodeExtraInfo()1156 lines.push_back(instr_shape); in GetInstructionNodeExtraInfo()
787 const Shape& instr_shape = in CreateCopyWithNewLayout() local790 HloInstruction::CreateGetTupleElement(instr_shape, instruction, i)); in CreateCopyWithNewLayout()792 if (ShapeUtil::Equal(target_shape, instr_shape)) { in CreateCopyWithNewLayout()