Searched refs:operand_shape_with_layout (Results 1 – 3 of 3) sorted by relevance
804 const Shape& operand_shape_with_layout = in HandleCustomCall() local807 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()
2299 const Shape& operand_shape_with_layout = in ParseInstructionRhs() local2301 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()
1137 Shape operand_shape_with_layout = ShapeUtil::MakeShapeWithLayout( in ChooseOutputLayoutFromOperandLayout() local1145 ShapeUtil::AlignLayouts(operand_shape_with_layout, output_shape); in ChooseOutputLayoutFromOperandLayout()