Searched refs:ConvertTypesToShapesWithLayout (Results 1 – 1 of 1) sorted by relevance
215 static std::vector<xla::Shape> ConvertTypesToShapesWithLayout( in ConvertTypesToShapesWithLayout() function241 if (!tuple_type) return ConvertTypesToShapesWithLayout({type}, layouts)[0]; in GetCustomCallResultShapeWithLayout()244 ConvertTypesToShapesWithLayout(tuple_type.getTypes(), layouts); in GetCustomCallResultShapeWithLayout()1132 auto operand_shapes_with_layout = ConvertTypesToShapesWithLayout( in ExportXlaOp()