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.cc804 const Shape& operand_shape_with_layout = in HandleCustomCall() local
807 operand_shape_with_layout)) in HandleCustomCall()
809 << operand_shape_with_layout.ToString(); in HandleCustomCall()
810 TF_RET_CHECK(LayoutUtil::HasLayout(operand_shape_with_layout)); in HandleCustomCall()
Dhlo_parser.cc2299 const Shape& operand_shape_with_layout = in ParseInstructionRhs() local
2301 if (!LayoutUtil::HasLayout(operand_shape_with_layout)) { in ParseInstructionRhs()
2305 operand_shape_with_layout), in ParseInstructionRhs()
2308 if (!ShapeUtil::Compatible(operand_shape_with_layout, in ParseInstructionRhs()
2314 ShapeUtil::HumanStringWithLayout(operand_shape_with_layout), in ParseInstructionRhs()
Dlayout_assignment.cc1137 Shape operand_shape_with_layout = ShapeUtil::MakeShapeWithLayout( in ChooseOutputLayoutFromOperandLayout() local
1145 ShapeUtil::AlignLayouts(operand_shape_with_layout, output_shape); in ChooseOutputLayoutFromOperandLayout()