Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier.cc556 const Shape& operand_shape_with_layout = in HandleCustomCall() local
559 operand_shape_with_layout)) in HandleCustomCall()
561 << operand_shape_with_layout.ToString(); in HandleCustomCall()
562 TF_RET_CHECK(LayoutUtil::HasLayout(operand_shape_with_layout)); in HandleCustomCall()
Dhlo_parser.cc1518 const Shape& operand_shape_with_layout = in ParseInstructionRhs() local
1520 if (!LayoutUtil::HasLayout(operand_shape_with_layout)) { in ParseInstructionRhs()
1524 operand_shape_with_layout), in ParseInstructionRhs()
1527 if (!ShapeUtil::Compatible(operand_shape_with_layout, in ParseInstructionRhs()
1533 ShapeUtil::HumanStringWithLayout(operand_shape_with_layout), in ParseInstructionRhs()
Dlayout_assignment.cc1079 Shape operand_shape_with_layout = ShapeUtil::MakeShapeWithLayout( in ChooseOutputLayoutFromOperandLayout() local
1087 ShapeUtil::AlignLayouts(operand_shape_with_layout, output_shape); in ChooseOutputLayoutFromOperandLayout()