Home
last modified time | relevance | path

Searched refs:operand_shape_with_layout (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_verifier.cc1174 const Shape& operand_shape_with_layout = in HandleCustomCall() local
1177 operand_shape_with_layout)) in HandleCustomCall()
1179 << operand_shape_with_layout.ToString(); in HandleCustomCall()
1180 TF_RET_CHECK(LayoutUtil::HasLayout(operand_shape_with_layout)); in HandleCustomCall()
H A Dhlo_parser.cc2662 const Shape& operand_shape_with_layout = in CreateInstruction() local
2664 if (!LayoutUtil::HasLayout(operand_shape_with_layout)) { in CreateInstruction()
2667 ShapeUtil::HumanStringWithLayout(operand_shape_with_layout), in CreateInstruction()
2671 if (!ShapeUtil::Compatible(operand_shape_with_layout, in CreateInstruction()
2675 ShapeUtil::HumanStringWithLayout(operand_shape_with_layout), in CreateInstruction()
H A Dlayout_assignment.cc1367 Shape operand_shape_with_layout = ShapeUtil::MakeShapeWithLayout( in ChooseOutputLayoutFromOperandLayout() local
1374 ShapeUtil::AlignLayouts(operand_shape_with_layout, output_shape); in ChooseOutputLayoutFromOperandLayout()