Searched refs:ExtractLayoutsFromShapes (Results 1 – 4 of 4) sorted by relevance
59 StatusOr<mlir::ArrayAttr> ExtractLayoutsFromShapes(
160 StatusOr<mlir::ArrayAttr> ExtractLayoutsFromShapes( in ExtractLayoutsFromShapes() function202 return ExtractLayoutsFromShapes(shape.tuple_shapes(), builder); in ExtractLayoutsFromTuple()
632 ExtractLayoutsFromShapes(custom_call->operand_shapes_with_layout(), in ImportInstructionImpl()644 ExtractLayoutsFromShapes({custom_call->shape()}, builder_)); in ImportInstructionImpl()
171 ExtractLayoutsFromShapes( in CustomCallInternal()182 ExtractLayoutsFromShapes({shape}, &builder_)); in CustomCallInternal()