Searched refs:ExtractLayout (Results 1 – 1 of 1) sorted by relevance
190 static xla::Layout ExtractLayout( in ExtractLayout() function1233 ExtractLayout(op, bitcast_proto->shape().dimensions_size(), in ExportXlaOp()1237 ExtractLayout(op, operand_proto->shape().dimensions_size(), in ExportXlaOp()1419 ExtractLayout(inst, shape->dimensions().size()).ToProto(); in Lower()1486 layout = ExtractLayout(inst, const_attr.getType().getRank()); in Lower()