Searched refs:GetCustomCallResultShapeWithLayout (Results 1 – 1 of 1) sorted by relevance
238 static xla::Shape GetCustomCallResultShapeWithLayout(mlir::Type type, in GetCustomCallResultShapeWithLayout() function1134 xla::Shape result_shape_with_layout = GetCustomCallResultShapeWithLayout( in ExportXlaOp()