Searched defs:shapes_with_layouts (Results 1 – 2 of 2) sorted by relevance
161 const absl::Span<const Shape> shapes_with_layouts, mlir::Builder* builder) { in ExtractLayoutsFromShapes()
243 std::vector<xla::Shape> shapes_with_layouts = in GetCustomCallResultShapeWithLayout() local